111

app.py

@@ -11,9 +11,12 @@ from nltk.tokenize import word_tokenize
 from nltk.corpus import stopwords
 from nltk.stem import WordNetLemmatizer
 from collections import Counter
+#from langdetect import detect
+#import jieba
+#import jieba.analyse
 import nltk
 
-openai.api_key = st.secrets["OPENAI_API_KEY"]
+
 
 @st.cache_data
 def download_nltk():
@@ -24,7 +27,6 @@ def download_nltk():
 def chunkstring(string, length):
         return (string[0+i:length+i] for i in range(0, len(string), length))
 
-
 def pdf_parser(input_pdf):
     pdf = PdfReader(input_pdf)
     pdf_content = ""
@@ -62,7 +64,8 @@ def get_keywords(file_paths): #这里的重点是，对每一个file做尽可能
 
     return keywords_list
 
-def get_completion_from_messages(messages, model="gpt-4-1106-preview", temperature=0):
+
+def get_completion_from_messages(messages, model="gpt-4", temperature=0):
         response = openai.ChatCompletion.create(
             model=model,
             messages=messages,
@@ -302,6 +305,12 @@ def add_prompt_course_style(selected_style_list):
 
 def app():
     st.title("OmniTutor v0.1.0")
+    announce = st.caption('''
+    :blue[⚠️Key Announcement:] The free service has been disrupted due to high costs. In order to use OmniTutor you need to type your own OPENAI API key into the sidebar.
+
+    :blue[⚠️关键公告：] 免费服务现在已经因为过高的支出而中断。为了使用OmniTutor，你需要在边栏顶部输入自己的Openai API key。
+    ''')
+    divider = st.divider()
     st.markdown("""
         <style>
             .footer {
@@ -318,6 +327,7 @@ def app():
         <div class="footer">Made with 🧡 by Siyuan</div>
     """, unsafe_allow_html=True)
     with st.sidebar:
+        api_key = st.text_input('Your OpenAI API key:', 'sk-...')
         st.image("https://siyuan-harry.oss-cn-beijing.aliyuncs.com/oss://siyuan-harry/20231021212525.png")
         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'):
@@ -353,22 +363,21 @@ def app():
     if "course_content_list" not in st.session_state:
         st.session_state.course_content_list = ''
     
+    if "OPENAI_API_KEY" not in st.session_state:
+        st.session_state["OPENAI_API_KEY"] = ''
     if "openai_model" not in st.session_state:
         st.session_state["openai_model"] = "gpt-3.5-turbo"
         # Initialize chat history
     if "messages" not in st.session_state:
         st.session_state.messages = []
-    
 
     st.session_state.start_col1, st.session_state.start_col2 = st.columns(2)
 
     with st.session_state.start_col1:
         st.session_state.description = st.markdown('''
-        > <font color = 'grey'> An all-round teacher. A teaching assistant who really knows the subject </font>
-        > 
-        > <font color = 'grey'> Anything. Anywhere. All at once. </font> :100:
-        > 
-        > Github Repo: https://github.com/Siyuan-Harry/OmniTutor 
+        <font color = 'grey'> An all-round teacher. A teaching assistant who really knows the subject. **Anything. Anywhere. All at once.** </font> :100:
+        
+        Github Repo: https://github.com/Siyuan-Harry/OmniTutor 
 
         ### ✨ Key features                                           
                                                    
@@ -379,12 +388,14 @@ def app():
         
         ### 🏃‍♂️ Get started!
                                                     
-        1. **Upload learning materials**: <font color = 'grey'>The upload widget in the sidebar supports PDF and .md files simutaenously.</font>
-        2. **Customize your course**: <font color = 'grey'>By few clicks and swipes, adjusting teaching style, lesson count and language for your course.</font>
-        3. **Start course generating**: <font color = 'grey'>Touch "Generate my course!" button in the sidebar, then watch how OmniTutor creates personal-customized course for you.</font>
-        4. **Interactive learning**: <font color = 'grey'>Learn the course, and ask OmniTutor any questions related to this course whenever you encountered them.</font>
+        1. **Input Your OpenAI API Key**: <font color = 'grey'>Give OmniTutor your own OpenAI API key (On top of the **sidebar**) to get started.</font>
+        2. **Upload learning materials**: <font color = 'grey'>The upload widget in the sidebar supports PDF and .md files simutaenously.</font>
+        3. **Customize your course**: <font color = 'grey'>By few clicks and swipes, adjusting teaching style, lesson count and language for your course.</font>
+        4. **Start course generating**: <font color = 'grey'>Touch "Generate my course!" button in the sidebar, then watch how OmniTutor creates personal-customized course for you.</font>
+        5. **Interactive learning**: <font color = 'grey'>Learn the course, and ask OmniTutor any questions related to this course whenever you encountered them.</font>
                                 
         🎉 Have fun playing with Omnitutor!
+                                                                                                                                       
         ''', unsafe_allow_html=True
         )
     with st.session_state.start_col2:
@@ -393,21 +404,21 @@ def app():
                                                 
         > <font color = 'grey'> 用OmniTutor阅读专业的论文! </font>                                        
 
-        <img src="https://siyuan-harry.oss-cn-beijing.aliyuncs.com/oss://siyuan-harry/WX20231104-122256@2x%2016.03.49.png" height = "300" />
+        <img src="https://siyuan-harry.oss-cn-beijing.aliyuncs.com/oss://siyuan-harry/WX20231104-122256@2x%2016.03.49.png" height = "200" />
                                                 
-        <img src="https://siyuan-harry.oss-cn-beijing.aliyuncs.com/oss://siyuan-harry/WX20231104-113553@2x%202.png" height = "300" />
+        <img src="https://siyuan-harry.oss-cn-beijing.aliyuncs.com/oss://siyuan-harry/WX20231104-113553@2x%202.png" height = "200" />
 
         > <font color = 'grey'> 用OmniTutor学习Python! </font>
                                                 
-        <img src="https://siyuan-harry.oss-cn-beijing.aliyuncs.com/oss://siyuan-harry/WX20231103-215001@2x.png" height = "300" />
+        <img src="https://siyuan-harry.oss-cn-beijing.aliyuncs.com/oss://siyuan-harry/WX20231103-215001@2x.png" height = "200" />
                                                 
-        <img src="https://siyuan-harry.oss-cn-beijing.aliyuncs.com/oss://siyuan-harry/WX20231103-214916@2x.png" height = "300" />
+        <img src="https://siyuan-harry.oss-cn-beijing.aliyuncs.com/oss://siyuan-harry/WX20231103-214916@2x.png" height = "200" />
         
         ---
 
         ### 💰 Support by donation 捐款支持
         
-        - 请捐款支持。因为每位用户的每次课程生成，都会产生一定的费用。个人开发者维护不易
+        - 如果OmniTutor对你有所帮助，请捐款赞助。因为每位用户的每次课程生成，都会产生一定的费用。个人开发者维护不易
         - Please donate to support. Because each course generated by each user will incur a certain cost for me. 
 
         <img src="https://siyuan-harry.oss-cn-beijing.aliyuncs.com/oss://siyuan-harry/20231110145229.png" height = "260" />
@@ -426,24 +437,37 @@ def app():
     
 
     if btn:
-        st.session_state.start_col1.empty()
-        st.session_state.start_col2.empty()
-        st.session_state.description.empty()
-        st.session_state.case_pay.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, style_options)
-
-        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'> If the course is good enough for you, learn and enter questions related in the input box below 👇... </font>
-
-                    :blue[Wish you all the best in your learning journey :)]
-                    ''', unsafe_allow_html=True)
+        if api_key != "sk-..." and api_key !="" and api_key.startswith("sk-"):
+            st.session_state.start_col1.empty()
+            st.session_state.start_col2.empty()
+            st.session_state.description.empty()
+            st.session_state.case_pay.empty()
+            announce.empty()
+            divider.empty()
+
+            #initialize app
+            temp_file_paths = initialize_file(added_files)
+            st.session_state["OPENAI_API_KEY"] = api_key
+            openai.api_key = st.session_state["OPENAI_API_KEY"]
+            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, style_options)
+
+            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'> If the course is good enough for you, learn and enter questions related in the input box below 👇... </font>
+
+                        :blue[Wish you all the best in your learning journey :)]
+                        ''', unsafe_allow_html=True)
+        else:
+            st.session_state.start_col1.empty()
+            st.session_state.start_col2.empty()
+            st.session_state.description.empty()
+            st.session_state.case_pay.empty()
+            announce.empty()
+            divider.empty()
+            warning = st.write("请输入正确的API Key令牌")
                     
 
     col1, col2 = st.columns([0.6,0.4])
@@ -454,6 +478,8 @@ def app():
         st.session_state.start_col2.empty()
         st.session_state.description.empty()
         st.session_state.case_pay.empty()
+        announce.empty()
+        divider.empty()
         
         with col1:
             #把课程大纲打印出来