test

app.py

@@ -48,22 +48,19 @@ def chat_with_ochat(message):
 
 
 
-def clear_chat():
-    return "Chat cleared. You can start a new conversation."
+# Create a Gradio Blocks interface
+with gr.Blocks() as app:
+    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")
+        submit_button = gr.Button("Send")
+    output_text = gr.Textbox(label="ochat Response", interactive=False)
 
-# Create a Gradio Interface with an improved layout
-iface = gr.Interface(
-    fn=chat_with_ochat, 
-    inputs=gr.Textbox(lines=2, placeholder="Type your message here..."),
-    outputs=[
-        gr.Textbox(label="ochat Response", placeholder="Response will appear here...", interactive=False),
-        gr.Button("Clear Chat", fn=clear_chat)
-    ],
-    title="ochat Chat Interface",
-    description="Engage in a conversation with the ochat server. Type your message below and get a response.",
-    layout="vertical",
-    theme="huggingface",  # This is an example theme, you can choose others
-    css=".input_text {height: 50px;} .output_text {height: 100px;}"
-)
+    def update_output():
+        response = chat_with_ochat(input_text.value)
+        output_text.update(value=response)
 
-iface.launch()
+    submit_button.click(fn=update_output, inputs=input_text, outputs=output_text)
+
+app.launch()