加入regenerate

app.py

@@ -348,25 +348,32 @@ def app():
     2. Touch "submit" button 
                                
     🎉 Get ready to see what happens..
-
     '''
     )
 
 
     if btn:
-        st.session_state.description1.empty()
-        st.session_state.divider.empty()
-        st.session_state.description2.empty()
+        while True:
+            st.session_state.description1.empty()
+            st.session_state.divider.empty()
+            st.session_state.description2.empty()
+
+            #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)
+
+            regenerate_button = st.button("Regenerate this course")
+            user_question = st.chat_input("Enter your questions when learning...")
+
+            if regenerate_button:
+                continue
+            elif user_question:
+                break
 
-        #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])
-        
-    user_question = st.chat_input("Enter your questions when learning...")
 
     if user_question:
         st.session_state.description1.empty()