Update app.py

app.py

@@ -136,10 +136,10 @@ async def text_mode():
         st.session_state["openai_model"] = "gpt-3.5-turbo-16k"
     if radio_1 == 'GPT-3.5':
         # print('----------'*5)
-        print('radio_1: GPT-3.5 starts!')
+        # print('radio_1: GPT-3.5 starts!')
         st.session_state["openai_model"] = "gpt-3.5-turbo-16k"
     else:
-        print('radio_1: GPT-4.0 starts!')
+        # print('radio_1: GPT-4.0 starts!')
         st.session_state["openai_model"] = "gpt-4"
 
     # Initialize chat history
@@ -154,8 +154,8 @@ async def text_mode():
     # Display assistant response in chat message container
     # if prompt := st.chat_input("Say something"):
     prompt = st.chat_input("Say something")
-    print('prompt now:', prompt)
-    print('----------'*5)
+    # print('prompt now:', prompt)
+    # print('----------'*5)
     # if prompt:
     if prompt:
         st.session_state.messages.append({"role": "user", "content": prompt})
@@ -167,7 +167,7 @@ async def text_mode():
             full_response = ""
 
             if radio_2 == '联网模式':
-                print('联网模式入口，prompt:', prompt)
+                # print('联网模式入口，prompt:', prompt)
                 input_message = prompt
                 internet_search_result = search(input_message)
                 search_prompt = [
@@ -197,8 +197,8 @@ async def text_mode():
                 st.session_state.messages = []
 
             if radio_2 == '核心模式':
-                print('GPT only starts!!!')
-                print('messages:', st.session_state['messages'])
+                # print('GPT only starts!!!')
+                # print('messages:', st.session_state['messages'])
                 for response in openai.ChatCompletion.create(
                     model=st.session_state["openai_model"],
                     # messages=[
@@ -212,14 +212,14 @@ async def text_mode():
                     full_response += response.choices[0].delta.get(
                         "content", "")
                     message_placeholder.markdown(full_response + "▌")
-                    print('session completed!')
+                    # print('session completed!')
                 message_placeholder.markdown(full_response)
             st.session_state.messages.append(
                 {"role": "assistant", "content": full_response})
 
 
 async def data_mode():
-    print('数据分析模式启动！')
+    # print('数据分析模式启动！')
     # uploaded_file_path = './upload.csv'
     uploaded_file_path = f'./{name}_upload.csv'
     # # st.write(f"passed file path in data_mode: {uploaded_file_path}")
@@ -238,8 +238,8 @@ async def data_mode():
     # Display assistant response in chat message container
     # if prompt := st.chat_input("Say something"):
     prompt = st.chat_input("Say something")
-    print('prompt now:', prompt)
-    print('----------'*5)
+    # print('prompt now:', prompt)
+    # print('----------'*5)
     # if prompt:
     if prompt:
         st.session_state.messages.append({"role": "user", "content": prompt})
@@ -269,7 +269,7 @@ async def data_mode():
                 user_request = environ_settings + "\n\n" + \
                     "你需要完成以下任务：\n\n" + prompt + "\n\n" \
                     f"注：文件位置在{uploaded_file_path}"
-                print('user_request: \n', user_request)
+                # print('user_request: \n', user_request)
 
                 # 加载上传的文件，主要路径在上面代码中。
                 files = [File.from_path(str(uploaded_file_path))]
@@ -281,7 +281,7 @@ async def data_mode():
                     )
 
                     # output to the user
-                    print("AI: ", response.content)
+                    # print("AI: ", response.content)
                     full_response = response.content
                     ### full_response = "this is full response"
 
@@ -473,11 +473,11 @@ if __name__ == "__main__":
     import asyncio
     try:
         if radio_2 == "核心模式":
-            print(f'radio 选择了 {radio_2}')
+            # print(f'radio 选择了 {radio_2}')
             # * 也可以用命令执行这个python文件。’streamlit run frontend/app.py‘
             asyncio.run(text_mode())
         if radio_2 == "联网模式":
-            print(f'radio 选择了 {radio_2}')
+            # print(f'radio 选择了 {radio_2}')
             # * 也可以用命令执行这个python文件。’streamlit run frontend/app.py‘
             asyncio.run(text_mode())
         if radio_2 == "数据模式":