Update chat.py

chat.py

@@ -123,33 +123,22 @@ def predict(input: str, history: List[Message]):
 
     response = make_completion(input, history)
 
-    return response
+    history.append({"role": "user", "content": input})
+    history.append({"role": "assistant", "content": response})
 
-#    history.append({"role": "user", "content": input})
-#    history.append({"role": "assistant", "content": response})
+    messages = [
+        (history[i]["content"], history[i+1]["content"]) 
+        for i in range(0, len(history)-1, 2)
+    ]
 
-#    messages = [
-#        (history[i]["content"], history[i+1]["content"]) 
-#        for i in range(0, len(history)-1, 2)
-#    ]
-
-#    return messages, history
+    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])
-
-    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
-    )
+    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])
 
 demo.launch(auth=("demouser", os.getenv('PASSWD')))