Update main_gradio.py

main_gradio.py

@@ -1,6 +1,24 @@
 import gradio as gr
 import random
 import time
+from pydantic import BaseModel
+import openai
+openai.api_key = os.environ["apikey"]
+
+def chatgpt(input, chat_history):
+    try:
+        messages = []
+        for message, bot_message in chat_history:
+            messages.append({"role": "user", "content": message})
+            messages.append({"role": "assistant", "content": bot_message})
+        prompt = {"role": "user", "content": input}
+        messages.append(prompt)
+        result = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages)
+        output = result.get('choices')[0].get('message').get('content')
+    except Exception as e:
+        output = str(e)
+    return output
+
 
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
@@ -8,7 +26,8 @@ with gr.Blocks() as demo:
     clear = gr.Button("Clear")
 
     def respond(message, chat_history):
-        bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
+        # bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
+        bot_message = chatgpt(message, chat_history)
         chat_history.append((message, bot_message))
         time.sleep(1)
         return "", chat_history