update password

app.py

@@ -1,52 +1,60 @@
+import os
+import time
+
 import gradio as gr
 import openai
-import random
-import time
-import os
 
 openai.api_key = os.environ["OPENAI_API_KEY"]
 openai.organization = os.environ["Organization_KEY"]
-model = os.environ["Model_Type"]
+zhanghao = os.environ["zhanghao"]
+mima = os.environ["mimamimamima"]
 
-openai.organization = "org-Gj9KBS6PafLRg7PzdOO6nZ2W"
 system_message = {"role": "system", "content": "You are a helpful assistant."}
 
-with gr.Blocks() as demo:
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox()
-    sub = gr.Button("发送")
-    clear = gr.Button("清除")
-    state = gr.State([])
-
-    def user(user_message, history):
-        return "", history + [[user_message, None]]
-
-    def bot(history, messages_history):
-        user_message = history[-1][0]
-        bot_message, messages_history = ask_gpt(user_message, messages_history)
-        messages_history += [{"role": "assistant", "content": bot_message}]
-        history[-1][1] = bot_message
-        time.sleep(1)
-        return history, messages_history
-
-    def ask_gpt(message, messages_history):
-        messages_history += [{"role": "user", "content": message}]
-        response = openai.ChatCompletion.create(
-            model=model,
-            messages=messages_history
-        )
-        return response['choices'][0]['message']['content'], messages_history
 
-    def init_history(messages_history):
-        messages_history = []
-        messages_history += [system_message]
-        return messages_history
+def user(user_message, history):
+    return "", history + [[user_message, None]]
+
 
-    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-        bot, [chatbot, state], [chatbot, state]
+def bot_3(history, messages_history):
+    model = "gpt-3.5-turbo"
+    user_message = history[-1][0]
+    bot_message, messages_history = ask_gpt(user_message, messages_history, model)
+    messages_history += [{"role": "assistant", "content": bot_message}]
+    history[-1][1] = bot_message
+    time.sleep(1)
+    return history, messages_history
+
+
+def ask_gpt(message, messages_history, model):
+    messages_history += [{"role": "user", "content": message}]
+    response = openai.ChatCompletion.create(
+        model=model,
+        messages=messages_history
     )
-    sub.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-        bot, [chatbot, state], [chatbot, state])
-    clear.click(lambda: None, None, chatbot, queue=False).success(init_history, [state], [state])
+    return response['choices'][0]['message']['content'], messages_history
+
+
+def init_history(messages_history):
+    messages_history = []
+    messages_history += [system_message]
+    return messages_history
+
+
+with gr.Blocks() as demo:
+    gr.Markdown("ChatGPT非科学接入")
+    with gr.Tab("GPT-3.5"):
+        chatbot = gr.Chatbot()
+        msg = gr.Textbox('输入栏')
+        sub = gr.Button("发送")
+        clear = gr.Button("清除")
+        state = gr.State([])
+
+        msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+            bot_3, [chatbot, state], [chatbot, state]
+        )
+        sub.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+            bot_3, [chatbot, state], [chatbot, state])
+        clear.click(lambda: None, None, chatbot, queue=False).success(init_history, [state], [state])
 
-demo.launch()
+demo.launch(auth=(zhanghao, mima))