大修改

app.py

@@ -205,7 +205,7 @@ def decorate_user_question(user_question, retrieved_chunks_for_user):
     '''
     return decorated_prompt
 
-def initialize_app(added_files, num_lessons, language):
+def initialize_file(added_files):
     temp_file_paths = []
     with st.spinner('Processing file...'):
         for added_file in added_files:
@@ -214,26 +214,30 @@ def initialize_app(added_files, num_lessons, language):
                 tmp_path = tmp.name
                 temp_file_paths.append(tmp_path)
     st.success('Processing file...Done')
+    return temp_file_paths
 
+def initialize_vdb(temp_file_paths):
     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")
-    
+    return embeddings_df, faiss_index
+
+def initialize_outline(temp_file_paths, num_lessons, language):
     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 = ''
     lessons_count = 0
     for outline in course_outline_list:
         lessons_count += 1
         course_outline_string += f"{lessons_count}." + outline[0]
         course_outline_string += '\n\n' + outline[1] + '\n\n'
-    with outline_presenting.expander("Check the course outline", expanded=False):
+    with st.expander("Check the course outline", expanded=False):
         st.write(course_outline_string)
+    
+    return course_outline_list
 
-    content_presenting = st.empty()
+def initialize_content(course_outline_list, embeddings_df, faiss_index, language):
     count_generating_content = 0
     course_content_list = []
     for lesson in course_outline_list:
@@ -243,26 +247,23 @@ def initialize_app(added_files, num_lessons, language):
             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):
+        with st.expander(f"Learn the lesson {count_generating_content} ", expanded=False):
             st.markdown(courseContent)
 
-    outline_presenting.empty()
-    content_presenting.empty()
-
-    return embeddings_df, faiss_index, course_outline_list, course_content_list
+    return course_content_list
 
 def regenerate_outline(course_outline_list):
-    #try:
-    course_outline_string = ''
-    lessons_count = 0
-    for outline in course_outline_list:
-        lessons_count += 1
-        course_outline_string += f"{lessons_count}." + outline[0]
-        course_outline_string += '\n\n' + outline[1] + '\n\n'
-    with st.expander("Check the course outline", expanded=False):
-        st.write(course_outline_string)
-    #except Exception:
-    #    pass
+    try:
+        course_outline_string = ''
+        lessons_count = 0
+        for outline in course_outline_list:
+            lessons_count += 1
+            course_outline_string += f"{lessons_count}." + outline[0]
+            course_outline_string += '\n\n' + outline[1] + '\n\n'
+        with st.expander("Check the course outline", expanded=False):
+            st.write(course_outline_string)
+    except Exception:
+        pass
 
 def regenerate_content(course_content_list):
     try:
@@ -302,24 +303,25 @@ def app():
         st.session_state.messages = []
     
     if btn:
-        st.session_state.embeddings_df, 
-        st.session_state.faiss_index, 
-        st.session_state.course_outline_list,
-        st.session_state.course_content_list = initialize_app(added_files, num_lessons, language)
-        #embeddings_df, faiss_index, course_outline_list = initialize_app(added_files, num_lessons, language)
+        #initialize app
+        temp_file_paths = initialize_file(added_files)
+        st.session_state.embeddings_df, st.session_state.faiss_index = initialize_vdb(temp_file_paths)
+        st.session_state.course_outline_list = initialize_outline(temp_file_paths, num_lessons, language)
+        st.session_state.course_content_list = initialize_content(st.session_state.course_outline_list, st.session_state.embeddings_df, st.session_state.faiss_index, language)
 
     col1, col2 = st.columns([0.6,0.4])
     
     # 这里还可以设置col1 和 col2 不同时出现 - 需要拆分initialize_app()函数
-    if st.session_state.course_content_list != 1:
         
-        user_question = st.chat_input("Enter your questions when learning...")
+    user_question = st.chat_input("Enter your questions when learning...")
 
+    if st.session_state.course_content_list != 1:
+        
         with col1:
             #把课程大纲打印出来
             regenerate_outline(st.session_state.course_outline_list)
             #把课程内容打印出来
-            #regenerate_content(st.session_state.course_content_list)
+            regenerate_content(st.session_state.course_content_list)
 
         with col2:
             st.caption(''':blue[AI Assistant]: Ask this TA any questions related to this course and get direct answers. :sunglasses:''')
@@ -354,7 +356,7 @@ def app():
                         message_placeholder.markdown(full_response + "▌")
                     message_placeholder.markdown(full_response)
                 st.session_state.messages.append({"role": "assistant", "content": [full_response,1]})
-        
+    
     
     
 if __name__ == "__main__":