test

README.md

@@ -2,7 +2,7 @@
 title: Fast OpenChat-3.5
 emoji: 🌍
 colorFrom: indigo
-colorTo: blue
+colorTo: green
 sdk: gradio
 sdk_version: 4.1.2
 app_file: app.py

app.py

@@ -46,19 +46,27 @@ def chat_with_ochat(message):
     except requests.RequestException as e:
         return f"Error: {e}"
 
-
+# Chat history variable
+chat_history = []
 
 # Create a Gradio Blocks interface
 with gr.Blocks() as app:
-    gr.Markdown("### ochat Chat Interface")
+    gr.Markdown("## vLLM OpenChat-3.5 Interface")
     gr.Markdown("Type your message and get a response from the ochat server.")
     with gr.Row():
-        input_text = gr.Textbox(label="Your Message")
+        input_text = gr.Textbox(
+            label="Your Message",
+            placeholder="Type your message here",
+            )
         submit_button = gr.Button("Send")
-    output_text = gr.Textbox(label="ochat Response", interactive=False)
+    output_text = gr.Textbox(label="ochat Response", interactive=True, scale=4)
 
     def update_output(input_message):
-        return chat_with_ochat(input_message)
+        global chat_history
+        user_message = f"You: {input_message}"
+        server_response = f"ochat: {chat_with_ochat(input_message)}"
+        chat_history.extend([user_message, server_response])
+        return "\n".join(chat_history)
 
     submit_button.click(fn=update_output, inputs=[input_text], outputs=[output_text])