Update app.py

app.py

@@ -44,17 +44,23 @@ def chat_with_cerebras(user_input):
 # Gradio interface
 def gradio_ui():
     with gr.Blocks() as demo:
-        gr.Markdown("""# Cerebras AI Chatbot\nChat with a state-of-the-art AI model.""")
+        gr.Markdown("""# 🤖 Cerebras AI Chatbot\nChat with a state-of-the-art AI model!""")
 
         with gr.Row():
-            user_input = gr.Textbox(label="Your Message", placeholder="Type your message here...")
-            response = gr.Textbox(label="AI Response", interactive=False)
-            compute_time = gr.Textbox(label="Compute Time", interactive=False)
+            with gr.Column(scale=8):
+                chat_history = gr.Chatbot(label="Chat History")
+            with gr.Column(scale=2):
+                compute_time = gr.Textbox(label="Compute Time", interactive=False)
 
-        submit_button = gr.Button("Submit")
+        user_input = gr.Textbox(label="Type your message", placeholder="Ask me anything...", lines=2)
+        send_button = gr.Button("Send", variant="primary")
 
-        # Define interaction logic
-        submit_button.click(chat_with_cerebras, inputs=user_input, outputs=[response, compute_time])
+        def handle_chat(chat_history, user_input):
+            ai_response, compute_info = chat_with_cerebras(user_input)
+            chat_history.append((user_input, ai_response))
+            return chat_history, compute_info
+
+        send_button.click(handle_chat, inputs=[chat_history, user_input], outputs=[chat_history, compute_time])
 
     return demo