Update chat.py

chat.py

@@ -115,7 +115,7 @@ def make_completion(input:str, history: List[Message]) -> Optional[str]:
     return prediction
 
 
-def predict(input, history):
+def predict(input: str, history: List[Message]):
 
     """
     Predict the response of the chatbot and complete a running list of chat history.
@@ -123,27 +123,30 @@ def predict(input, history):
 
     response = make_completion(input, history)
 
-    history.append({"role": "user", "content": input})
-    history.append({"role": "assistant", "content": response})
+    return response
 
-    messages = [
-        (history[i]["content"], history[i+1]["content"]) 
-        for i in range(0, len(history)-1, 2)
-    ]
+#    history.append({"role": "user", "content": input})
+#    history.append({"role": "assistant", "content": response})
 
-    return messages, history
+#    messages = [
+#        (history[i]["content"], history[i+1]["content"]) 
+#        for i in range(0, len(history)-1, 2)
+#    ]
+
+#    return messages, history
 
 
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot(label="CHAT", layout="bubble", likeable=True, show_copy_button=True)
-    state = gr.State([])
-    with gr.Row():
-        txt = gr.Textbox(show_label=True, placeholder="Enter text and press enter")
-    txt.submit(predict, [txt, state], [chatbot, state])
+#    state = gr.State([])
+#    with gr.Row():
+#        txt = gr.Textbox(show_label=True, placeholder="Enter text and press enter")
+#    txt.submit(predict, [txt, state], [chatbot, state])
 
     gr.ChatInterface(
         fn=predict, chatbot=chatbot,
         title="Chat with information about your insurance policy",
+        textbox=gr.Textbox(show_label=True, placeholder="Enter text and press enter"),
         examples=["What is my policy number?", "Are dental expenses covered in my policy?"],
         show_progress="full",
         concurrency_limit=8