fixed the chatbot

home_view/ui_home.py

@@ -66,11 +66,25 @@ def CreateGame():
         return game_ideas, game_ideas_dict[0]['Description'],game_ideas_dict[0]['Name']
     return None, constants.NO_GAME_TITLE, constants.NO_GAME
 
-def yes_man(message, history):
-    if message.endswith("?"):
-        return "Yes"
-    else:
-        return "Ask me anything!"
+def chatBot(message, history):
+    history_openai_format = []
+    for human, assistant in history:
+        history_openai_format.append({"role": "user", "content": human })
+        history_openai_format.append({"role": "assistant", "content":assistant})
+    history_openai_format.append({"role": "user", "content": message})
+
+    response = openai.ChatCompletion.create(
+        model='gpt-3.5-turbo',
+        messages= history_openai_format,
+        temperature=1.0,
+        stream=True
+    )
+
+    partial_message = ""
+    for chunk in response:
+        if len(chunk['choices'][0]['delta']) != 0:
+            partial_message = partial_message + chunk['choices'][0]['delta']['content']
+            yield partial_message
 
 df1 =gr.Dataframe(type="pandas", value=startDF)
 df2 =gr.Dataframe(type="pandas", value=slackDF)
@@ -95,9 +109,7 @@ with gr.Blocks() as ui:
         label=gr.Label()
     with gr.Row(visible=False) as chatRow:        
         chat = gr.ChatInterface(
-            yes_man,
-            chatbot=gr.Chatbot(height=300),
-            textbox=gr.Textbox(placeholder="Ask me a yes or no question", container=False, scale=7),
+            chatBot,
             title="Yes Man",
             description="Ask Yes Man any question",
             theme="soft",
@@ -107,6 +119,7 @@ with gr.Blocks() as ui:
             undo_btn="Delete Previous",
             clear_btn="Clear",
         )
+        
     sketch.change(Predict, inputs=[sketch], outputs=[label])
     playersCB.change(SetSelected, inputs=[playersCB], outputs=[])
     createBtn.click(CreateGame, inputs=[], outputs=[df4,desc,gameTitle])