优化加载图标

app.py

@@ -209,24 +209,21 @@ def decorate_user_question(user_question, retrieved_chunks_for_user):
 @st.cache_data
 def initialize_app(added_files, num_lessons, language):
     temp_file_paths = []
-    file_proc_state = st.empty()
-    file_proc_state.text("Processing file...")
-    for added_file in added_files:
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".md") as tmp:
-            tmp.write(added_file.getvalue())
-            tmp_path = tmp.name
-            temp_file_paths.append(tmp_path)
-    file_proc_state.text("Processing file...Done")
-
-    vdb_state = st.empty()
-    vdb_state.text("Constructing vector database from provided materials...")
-    embeddings_df, faiss_index = constructVDB(temp_file_paths)
-    vdb_state.text("Constructing vector database from provided materials...Done")
+    with st.spinner('Processing file...'):
+        for added_file in added_files:
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".md") as tmp:
+                tmp.write(added_file.getvalue())
+                tmp_path = tmp.name
+                temp_file_paths.append(tmp_path)
+    st.success('Processing file...Done!')
+
+    with st.spinner('Constructing vector database from provided materials...'):
+        embeddings_df, faiss_index = constructVDB(temp_file_paths)
+    st.success("Constructing vector database from provided materials...Done")
     
-    outline_generating_state = st.empty()
-    outline_generating_state.text("Generating Course Outline...")
-    course_outline_list = courseOutlineGenerating(temp_file_paths, num_lessons, language)
-    outline_generating_state.text("Generating Course Outline...Done")
+    with st.spinner('Generating Course Outline...'):
+        course_outline_list = courseOutlineGenerating(temp_file_paths, num_lessons, language)
+    st.success("Generating Course Outline...Done")
 
     outline_presenting = st.empty()
     course_outline_string = ''
@@ -239,25 +236,20 @@ def initialize_app(added_files, num_lessons, language):
         st.write(course_outline_string)
 
     content_presenting = st.empty()
-    content_generating_state = st.empty()
     count_generating_content = 0
     course_content_list = []
     for lesson in course_outline_list:
         count_generating_content += 1
-        content_generating_state = st.text(f"Writing content for lesson {count_generating_content}...")
-        retrievedChunksList = searchVDB(lesson, st.session_state.embeddings_df, st.session_state.faiss_index)
-        courseContent = generateCourse(lesson, retrievedChunksList, language)
-        course_content_list.append(courseContent)
-        content_generating_state.text(f"Writing content for lesson {count_generating_content}...Done")
+        with st.spinner(f"Writing content for lesson {count_generating_content}..."):
+            retrievedChunksList = searchVDB(lesson, st.session_state.embeddings_df, st.session_state.faiss_index)
+            courseContent = generateCourse(lesson, retrievedChunksList, language)
+            course_content_list.append(courseContent)
+        st.success(f"Writing content for lesson {count_generating_content}...Done")
         with content_presenting.expander(f"Learn the lesson {count_generating_content} ", expanded=False):
             st.markdown(courseContent)
 
-    file_proc_state.empty()
-    vdb_state.empty()
-    outline_generating_state.empty()
     outline_presenting.empty()
     content_presenting.empty()
-    content_generating_state.empty()
 
     return embeddings_df, faiss_index, course_outline_list, course_content_list