new

app.py

@@ -2,6 +2,9 @@ import gradio as gr
 import os
 import openai
 import gradio as gr
+from gradio import ChatInterface
+import time
+
 # Get the value of the openai_api_key from environment variable
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
@@ -19,47 +22,35 @@ from langchain.schema import (
 from langchain.utilities import WikipediaAPIWrapper
 from langchain.tools import AIPluginTool
 
-# Setting up a system message for our Chatbot
-#system = SystemMessage(content = "You are a helpful AI assistant") # that translates English to Pirate English.")
+# Question- how can one set up a system message for their Chatbot while using ChatInterface
+# Example system message : system = SystemMessage(content = "You are a helpful AI assistant")
 
 # driver
 def predict(user_input, chatbot):
 
-    print(f"chatbot - {chatbot}")
-    print(f"user_input - {user_input}")
-
-    chat = ChatOpenAI(
-    temperature=1.0, 
-    streaming=True,
-    model='gpt-3.5-turbo-0613')
-    #messages = [system]
+    chat = ChatOpenAI(temperature=1.0, streaming=True, model='gpt-3.5-turbo-0613')
     messages=[]
 
-    if len(chatbot) != 0:
-        for conv in chatbot:
-            human = HumanMessage(content=conv[0])
-            ai = AIMessage(content=conv[1])
-            messages.append(human)
-            messages.append(ai)
-        messages.append(HumanMessage(content=user_input))
-        print(f"messages list is - {messages}")
-    
-    else: # for first user message
-        messages.append(HumanMessage(content=user_input))
-        print(f"messages list is - {messages}")
+    for conv in chatbot:
+        human = HumanMessage(content=conv[0])
+        ai = AIMessage(content=conv[1])
+        messages.append(human)
+        messages.append(ai)
+
+    messages.append(HumanMessage(content=user_input))
 
     # getting gpt3.5's response
     gpt_response = chat(messages)
-    print(f"gpt_response - {gpt_response}")
-    bot_message = gpt_response.content
-    print(f"bot_message - {bot_message}")
+    return gpt_response.content
 
-    #chatbot.append((user_input, bot_message))
 
-    #return "", chatbot, None #"", chatbot
-    return bot_message
 
+def echo_stream(message, history):
+    for i in range(len(message)):
+        time.sleep(0.3)
+        yield message[:i]
 
+ChatInterface(echo_stream).queue().launch(debug=True)
 
-chatbot = gr.Chatbot()
-gr.ChatInterface(predict, chatbot=chatbot, delete_last_btn="del").queue().launch(share=False, debug=True) #examples=["How are you?", "What's up?"],
+#chatbot = gr.Chatbot()
+#gr.ChatInterface(predict, delete_last_btn="del").queue().launch(share=False, debug=True) #examples=["How are you?", "What's up?"],