加入课程风格选择

app.py

@@ -134,7 +134,6 @@ def constructVDB(file_paths):
     #从embeddings到向量数据库
     # Load the embeddings
     embeddings = paraphrase_embeddings_df.iloc[:, 1:].values  # All columns except the first (chunk text)
-
     # Ensure that the array is C-contiguous
     embeddings = np.ascontiguousarray(embeddings, dtype=np.float32)
     # Preparation for Faiss
@@ -171,7 +170,7 @@ def searchVDB(search_sentence, paraphrase_embeddings_df, index):
         
     return retrieved_chunks_list
 
-def generateCourse(topic, materials, language):
+def generateCourse(topic, materials, language, style_options):
     #调用gpt4 API生成一节课的内容
     system_message = 'You are a great AI teacher and linguist, skilled at writing informative and easy-to-understand course script based on given lesson topic and knowledge materials.'
 
@@ -189,6 +188,7 @@ def generateCourse(topic, materials, language):
             lesson topic and abstract: 「{topic}」,
             knowledge materials related to this lesson：【{materials} 】
             the script should be witten in {language}, and mathematical symbols should be written in markdown form.
+            {style_options}
             Start writting the script of this lesson now.
             """
 
@@ -289,6 +289,22 @@ def regenerate_content(course_content_list):
     except Exception:
         pass
 
+def add_prompt_course_style(selected_style_list):
+    initiate_prompt = 'Please be siginificantly aware that this course is requested to: \n'
+    if len(selected_style_list) != 0:
+        customize_prompt += initiate_prompt
+        for style in selected_style_list:
+            if style == "More examples":
+                customize_prompt += '- **contain more examples**. You should use your own knowledge to vividly exemplify key concepts occured in this course.\n'
+            elif style == "More excercises":
+                customize_prompt += '- **contain more excercises**. So last part of this lesson should be excercises.\n'
+            elif style == "Easier to learn":
+                customize_prompt += '- **Be easier to learn**. So you should use plain language to write the lesson script, and apply some metaphors & analogys wherever appropriate.\n'
+    else:
+        customize_prompt = ''       
+    
+    return customize_prompt
+
 def app():
     st.title("OmniTutor v0.1.0")
     st.markdown("""
@@ -308,13 +324,20 @@ def app():
     """, unsafe_allow_html=True)
     with st.sidebar:
         st.image("https://siyuan-harry.oss-cn-beijing.aliyuncs.com/oss://siyuan-harry/20231021212525.png")
-        added_files = st.file_uploader('Upload .md and .pdf files, simultaneous mixed upload these types is supported.', type=['.md','.pdf'], accept_multiple_files=True)
+        added_files = st.file_uploader('Upload .md or .pdf files, simultaneous mixed upload these types is supported.', type=['.md','.pdf'], accept_multiple_files=True)
         with st.expander('Customize my course'):
             num_lessons = st.slider('How many lessons do you want this course to have?', min_value=2, max_value=15, value=5, step=1)
             language = 'English'
             Chinese = st.checkbox('Output in Chinese')
-        if Chinese:
-            language = 'Chinese'
+            if Chinese:
+                language = 'Chinese'
+            custom_options = st.multiselect(
+                'Customize ',
+                ['More examples', 'More excercises', 'Easier to learn'],
+                max_selections = 2
+            )
+            style_options = add_prompt_course_style(custom_options)
+            
         btn = st.button('Generate my course!')
     
     if "description1" not in st.session_state:
@@ -360,11 +383,11 @@ def app():
         #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_outline_list = initialize_outline(temp_file_paths, num_lessons, language, style_options)
         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)
 
         st.markdown('''
-                    > 🤔 <font color = 'grey'> Not satisfied with this course? Simply click "Generate my course!" button to regenerate a new one! </font>
+                    > 🤔 <font color = 'grey'> **Not satisfied with this course?** Simply click "Generate my course!" button to regenerate a new one! </font>
                     >
                     > 😁 <font color = 'grey'> If the course is good enough for you, learn and enter questions related in the input box below 👇... </font>
 
@@ -406,7 +429,7 @@ def app():
             retrieved_chunks_for_user = searchVDB(user_question, st.session_state.embeddings_df, st.session_state.faiss_index)
             prompt = decorate_user_question(user_question, retrieved_chunks_for_user)
             st.session_state.messages.append({"role": "user", "content": [user_question, prompt]})
-            
+
             # Display assistant response in chat message container
             with st.chat_message("assistant"):
                 message_placeholder = st.empty()
@@ -422,6 +445,5 @@ def app():
             st.session_state.messages.append({"role": "assistant", "content": [full_response,1]})
     
     
-    
 if __name__ == "__main__":
     app()