save

app.py

@@ -1,107 +1,83 @@
-# import gradio as gr
-# import os, openai
-#
-#
-# conversation = []
-#
-# class ChatGPT:
-#
-#
-#     def __init__(self):
-#         self.api_key = ""
-#         self.messages = conversation
-#         self.model = os.getenv("OPENAI_MODEL", default = "gpt-3.5-turbo")
-#
-#     def save_api_key(self, user_input0):
-#         self.api_key = user_input0
-#
-#     def get_response(self, user_input):
-#         openai.api_key = self.api_key
-#         conversation.append({"role": "user", "content": user_input})
-#
-#
-#         response = openai.ChatCompletion.create(
-# 	            model=self.model,
-#                 messages = self.messages
-#
-#                 )
-#
-#         conversation.append({"role": "assistant", "content": response['choices'][0]['message']['content']})
-#
-#         print("AI回答內容：")
-#         print(response['choices'][0]['message']['content'].strip())
-#
-#
-#
-#         return response['choices'][0]['message']['content'].strip()
-#
-#
-# chatgpt = ChatGPT()
-#
-#
-# def greet(prompt, api_key):
-#     chatgpt.save_api_key(api_key)
-#
-#     reply_text = chatgpt.get_response(prompt)
-#
-#     greeting = f"{reply_text}"
-#
-#     return greeting
-#
-# demo = gr.Interface(
-#     fn=greet,
-#     inputs=["text", "text"],
-#     outputs=["text"],
-# )
-#
-# demo.launch()
+import gradio as gr
+import openai
+import time
 
-import argparse
+with gr.Blocks() as demo:
+    with gr.Row():
+        key = gr.Textbox(placeholder="API_KEY")
+    with gr.Row():
+        with gr.Column():
+            msg = gr.Textbox(placeholder="Question")
+            submit = gr.Button("Submit")
+            clear = gr.Button("Clear")
+        with gr.Column():
+            chatbot = gr.Chatbot()
 
-import gradio as gr
-from loguru import logger
 
-from chat_completion import ChatCompletion
+    # state = gr.State([])
+
+    def user(user_message, history):
+        return "", history + [[user_message, None]]
 
-parser = argparse.ArgumentParser()
-parser.add_argument('--api_key_path', type=str, default='./openai_api_key')
-parser.add_argument('--log_path', type=str, default='./log.txt')
-parser.add_argument('--share', action='store_true', default=False)
-parser.add_argument('--welcome', type=str, default='Say something to ChatGPT here ...')
-parser.add_argument('--title', type=str, default='ChatGPT')
-parser.add_argument('--setting', type=str, default=None)
-args = parser.parse_args()
 
-bot = ChatCompletion(api_key_path=args.api_key_path)
-logger.add(args.log_path)
+    def bot(history, key):
+        openai.api_key = key
+        bot_message = ask_gpt(history)
+        print(history)
+        history[-1][1] = bot_message
+        time.sleep(1)
+        return history
 
-with gr.Blocks(title=args.title) as demo:
-    chatbot = gr.Chatbot(show_label=False)
-    msg = gr.TextArea(show_label=False, placeholder=args.welcome)
-    send_btn = gr.Button('Send')
-    retry_btn = gr.Button('Retry')
-    reset_btn = gr.Button('Reset')
 
-    def send(user_message, history):
-        if not user_message:
-            return '', history
+    def ask_gpt(history):
+        messages = []
+        for i in range(len(history) - 1):
+            messages.append({"role": "user", "content": history[i][0]})
+            messages.append({"role": "assistant", "content": history[i][1]})
+        messages.append({"role": "user", "content": history[-1][0]})
+        try:
+            response = openai.ChatCompletion.create(
+                model="gpt-3.5-turbo",
+                messages=messages
+            )
+            return response['choices'][0]['message']['content'].replace("```", "")
+        except Exception as e:
+            print(e)
+            return e
 
-        logger.info(f'[MSG] {user_message}')
-        response = bot(user_message, setting=args.setting) if user_message != 'retry' else bot.retry()
-        logger.info(f'[ANS] {response}')
-        return '', history + [[user_message, response]]
 
-    def reset():
-        bot.reset()
-        logger.info('[RESET]')
-        return None, [[None, None]]
+    # def bot(history, messages_history, key):
+    #     openai.api_key = key
+    #     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):
+    #     try:
+    #         messages_history += [{"role": "user", "content": message}]
+    #         response = openai.ChatCompletion.create(
+    #             model="gpt-3.5-turbo",
+    #             messages=messages_history
+    #         )
+    #         return response['choices'][0]['message']['content'], messages_history
+    #     except Exception as e:
+    #         print(e)
+    #         return e, messages_history
 
-    def retry(history):
-        return send('retry', history)
+    # def init_history(messages_history):
+    #     messages_history = []
+    #     return messages_history
 
-    send_btn.click(send, inputs=[msg, chatbot], outputs=[msg, chatbot], show_progress=True)
-    reset_btn.click(reset, inputs=None, outputs=[msg, chatbot])
-    retry_btn.click(retry, inputs=chatbot, outputs=[msg, chatbot])
+    submit.click(user, inputs=[msg, chatbot], outputs=[msg, chatbot], queue=True, api_name="submit").then(
+        bot, [chatbot, key], chatbot, api_name="bot_response"
+    )
+    clear.click(lambda: None, None, chatbot, queue=True, api_name="clear")
 
+    # clear.click(lambda: None, None, chatbot, queue=False, api_name="clear").then(init_history, [state], [state],api_name="init_history")
 
-demo.launch(share=args.share)
+demo.queue()
+demo.launch()