Upload app.py

app.py

@@ -3,32 +3,45 @@ import gradio as gr
 import openai
 import os
 import sys
+import traceback
 # import markdown
 
-my_api_key = ""    # 在这里输入你的 API 密钥
+my_api_key = "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"    # 在这里输入你的 API 密钥
 initial_prompt = "You are a helpful assistant."
 
-if my_api_key == "":
-    my_api_key = os.environ.get('my_api_key')
-
 if my_api_key == "empty":
     print("Please give a api key!")
     sys.exit(1)
 
+if my_api_key == "":
+    initial_keytxt = None
+elif len(str(my_api_key)) == 51:
+    # initial_keytxt = "默认api-key（未验证）：" + str(my_api_key[:4] + "..." + my_api_key[-4:])
+    initial_keytxt = "Zhou Yang Assistant ！"
+else:
+    initial_keytxt = "默认api-key无效，请重新输入"
+
 def parse_text(text):
     lines = text.split("\n")
+    count = 0
     for i,line in enumerate(lines):
         if "```" in line:
+            count += 1
             items = line.split('`')
-            if items[-1]:
+            if count % 2 == 1:
                 lines[i] = f'<pre><code class="{items[-1]}">'
             else:
                 lines[i] = f'</code></pre>'
         else:
-            if i>0:
-                line = line.replace("<", "&lt;")
-                line = line.replace(">", "&gt;")
-                lines[i] = '<br/>'+line.replace(" ", "&nbsp;")
+            if i > 0:
+                if count % 2 == 1:
+                    line = line.replace("&", "&amp;")
+                    line = line.replace("\"", "&quot;")
+                    line = line.replace("\'", "&apos;")
+                    line = line.replace("<", "&lt;")
+                    line = line.replace(">", "&gt;")
+                    line = line.replace(" ", "&nbsp;")
+                lines[i] = '<br/>'+line
     return "".join(lines)
 
 def get_response(system, context, myKey, raw = False):
@@ -56,9 +69,21 @@ def predict(chatbot, input_sentence, system, context, myKey):
 
     try:
         message, message_with_stats = get_response(system, context, myKey)
-    except:
+    except openai.error.AuthenticationError:
         chatbot.append((input_sentence, "请求失败，请检查API-key是否正确。"))
         return chatbot, context
+    except openai.error.Timeout:
+        chatbot.append((input_sentence, "请求超时，请检查网络连接。"))
+        return chatbot, context
+    except openai.error.APIConnectionError:
+        chatbot.append((input_sentence, "连接失败，请检查网络连接。"))
+        return chatbot, context
+    except openai.error.RateLimitError:
+        chatbot.append((input_sentence, "请求过于频繁，请5s后再试。"))
+        return chatbot, context
+    except:
+        chatbot.append((input_sentence, "发生了未知错误Orz"))
+        return chatbot, context
 
     context.append({"role": "assistant", "content": message})
 
@@ -69,11 +94,25 @@ def predict(chatbot, input_sentence, system, context, myKey):
 def retry(chatbot, system, context, myKey):
     if len(context) == 0:
         return [], []
+    
     try:
         message, message_with_stats = get_response(system, context[:-1], myKey)
-    except:
+    except openai.error.AuthenticationError:
         chatbot.append(("重试请求", "请求失败，请检查API-key是否正确。"))
         return chatbot, context
+    except openai.error.Timeout:
+        chatbot.append(("重试请求", "请求超时，请检查网络连接。"))
+        return chatbot, context
+    except openai.error.APIConnectionError:
+        chatbot.append(("重试请求", "连接失败，请检查网络连接。"))
+        return chatbot, context
+    except openai.error.RateLimitError:
+        chatbot.append(("重试请求", "请求过于频繁，请5s后再试。"))
+        return chatbot, context
+    except:
+        chatbot.append(("重试请求", "发生了未知错误Orz"))
+        return chatbot, context
+    
     context[-1] = {"role": "assistant", "content": message}
 
     chatbot[-1] = (context[-2]["content"], message_with_stats)
@@ -130,16 +169,24 @@ def update_system(new_system_prompt):
 
 def set_apikey(new_api_key, myKey):
     old_api_key = myKey
+    
     try:
         get_response(update_system(initial_prompt), [{"role": "user", "content": "test"}], new_api_key)
-    except:
+    except openai.error.AuthenticationError:
         return "无效的api-key", myKey
+    except openai.error.Timeout:
+        return "请求超时，请检查网络设置", myKey
+    except openai.error.APIConnectionError:
+        return "网络错误", myKey
+    except:
+        return "发生了未知错误Orz", myKey
+    
     encryption_str = "验证成功，api-key已做遮挡处理：" + new_api_key[:4] + "..." + new_api_key[-4:]
     return encryption_str, new_api_key
 
 
 with gr.Blocks() as demo:
-    keyTxt = gr.Textbox(show_label=True, placeholder=f"在这里输入你的API-key...", value=my_api_key, label="API Key").style(container=True)
+    keyTxt = gr.Button(initial_keytxt)
     chatbot = gr.Chatbot().style(color_map=("#1D51EE", "#585A5B"))
     context = gr.State([])
     systemPrompt = gr.State(update_system(initial_prompt))
@@ -158,14 +205,14 @@ with gr.Blocks() as demo:
         reduceTokenBtn = gr.Button("♻️ 优化Tokens")
     newSystemPrompt = gr.Textbox(show_label=True, placeholder=f"在这里输入新的System Prompt...", label="更改 System prompt").style(container=True)
     systemPromptDisplay = gr.Textbox(show_label=True, value=initial_prompt, interactive=False, label="目前的 System prompt").style(container=True)
-    # with gr.Accordion(label="保存/加载对话历史记录(在文本框中输入文件名，点击“保存对话”按钮，历史记录文件会被存储到本地)", open=False):
-    #     with gr.Column():
-    #         with gr.Row():
-    #             with gr.Column(scale=6):
-    #                 saveFileName = gr.Textbox(show_label=True, placeholder=f"在这里输入保存的文件名...", label="保存对话", value="对话历史记录").style(container=True)
-    #             with gr.Column(scale=1):
-    #                 saveBtn = gr.Button("💾 保存对话")
-    #                 uploadBtn = gr.UploadButton("📂 读取对话", file_count="single", file_types=["json"])
+    with gr.Accordion(label="保存/加载对话历史记录(在文本框中输入文件名，点击“保存对话”按钮，历史记录文件会被存储到本地)", open=False):
+        with gr.Column():
+            with gr.Row():
+                with gr.Column(scale=6):
+                    saveFileName = gr.Textbox(show_label=True, placeholder=f"在这里输入保存的文件名...", label="保存对话", value="对话历史记录").style(container=True)
+                with gr.Column(scale=1):
+                    saveBtn = gr.Button("💾 保存对话")
+                    uploadBtn = gr.UploadButton("📂 读取对话", file_count="single", file_types=["json"])
 
     txt.submit(predict, [chatbot, txt, systemPrompt, context, myKey], [chatbot, context], show_progress=True)
     txt.submit(lambda :"", None, txt)
@@ -178,9 +225,10 @@ with gr.Blocks() as demo:
     retryBtn.click(retry, [chatbot, systemPrompt, context, myKey], [chatbot, context], show_progress=True)
     delLastBtn.click(delete_last_conversation, [chatbot, context], [chatbot, context], show_progress=True)
     reduceTokenBtn.click(reduce_token, [chatbot, systemPrompt, context, myKey], [chatbot, context], show_progress=True)
-    keyTxt.submit(set_apikey, [keyTxt, myKey], [keyTxt, myKey], show_progress=True)
-    # uploadBtn.upload(load_chat_history, uploadBtn, [chatbot, systemPrompt, context, systemPromptDisplay], show_progress=True)
-    # saveBtn.click(save_chat_history, [saveFileName, systemPrompt, context], None, show_progress=True)
+    # keyTxt.submit(set_apikey, [keyTxt, myKey], [keyTxt, myKey], show_progress=False)
+    uploadBtn.upload(load_chat_history, uploadBtn, [chatbot, systemPrompt, context, systemPromptDisplay], show_progress=True)
+    saveBtn.click(save_chat_history, [saveFileName, systemPrompt, context], None, show_progress=True)
 
 
-demo.launch()
+# demo.launch(server_name="0.0.0.0", server_port=9000)
+demo.launch()