test

app.py

@@ -25,6 +25,13 @@ def start_ochat_server():
 
 start_ochat_server()
 
+def user(message, history):
+    return "", history + [[message, None]]
+
+
+def bot(history):
+    return chat_with_ochat(history[-1][0]), history + [[None, chat_with_ochat(history[-1][0])]]
+
 # Function to send a message to the ochat server and get a response
 def chat_with_ochat(message):
     url = "http://0.0.0.0:18888/v1/chat/completions"
@@ -50,20 +57,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     registry.hf.space/macadeliccc-openchat-3-5-chatbot:latest python app.py```")
 
     with gr.Row():
-        input_text = gr.Textbox(label="Your Message", placeholder="Type your message here")
-        submit_button = gr.Button("Send")
-    output_chat = gr.Chatbot()
+        message = gr.Textbox(label="Your Message", placeholder="Type your message here")
+        chatbot = gr.Chatbot()
+        clear = gr.Button("Clear")
 
     chat_history = State([])  # Session state for chat history
 
-    def update_output(input_message, chat_history):
-        server_response = chat_with_ochat(input_message)  # Server's response
-        chat_history.append((input_message, server_response))
-        return chat_history
-
-    submit_button.click(
-        fn=update_output, 
-        inputs=[input_text, chat_history],  # Ensure both inputs are passed
-        outputs=[output_chat]
+    message.submit(user, [message, chatbot], [message, chatbot], queue=False).then(
+        chat_with_ochat, chatbot, chatbot
     )
+    clear.click(lambda: None, None, chatbot, queue=False)
 app.launch()