Update app.py

app.py

@@ -58,13 +58,12 @@ def respond(
     except Exception as e:
         yield f"An error occurred: {str(e)}"
 
-# Define the UI layout with a more user-friendly design
+# Define the UI layout with improved structure
 with gr.Blocks() as demo:
     gr.Markdown("# 🧠 AI Chatbot Interface")
     gr.Markdown("### Customize your AI Chatbot's behavior and responses.")
     
     with gr.Row():
-        chatbot = gr.Chatbot()
         with gr.Column():
             system_message = gr.Textbox(
                 value="You are a helpful assistant knowledgeable in various topics. Provide clear, concise, and friendly responses.",
@@ -76,14 +75,43 @@ with gr.Blocks() as demo:
             top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)")
     
     with gr.Row():
-        message = gr.Textbox(label="Your message:", lines=1)
-        submit_btn = gr.Button("Send")
+        chatbot = gr.Chatbot()
+    
+    with gr.Row():
+        with gr.Column():
+            sample_prompt = gr.Dropdown(
+                choices=[
+                    "Can you explain the theory of relativity?",
+                    "What are some tips for improving productivity at work?",
+                    "Tell me a fun fact about space.",
+                    "How can I cook a perfect omelette?",
+                    "What's the latest news in technology?"
+                ],
+                label="Sample Prompts",
+                value="Can you explain the theory of relativity?",
+                type="value"
+            )
+            message = gr.Textbox(label="Your message:", lines=1)
+            submit_btn = gr.Button("Send")
+            clear_btn = gr.Button("Clear Chat")
+
+    # Handle sample prompt selection
+    def update_message(prompt):
+        return prompt
+
+    sample_prompt.change(fn=update_message, inputs=sample_prompt, outputs=message)
     
     # Update the chatbot with the new message and response
     submit_btn.click(respond, 
                      inputs=[message, chatbot, system_message, max_tokens, temperature, top_p], 
                      outputs=[chatbot],
                      show_progress=True)
+    
+    # Clear the chat history
+    def clear_chat():
+        return []
+
+    clear_btn.click(fn=clear_chat, inputs=None, outputs=chatbot)
 
 # Launch the Gradio interface
 if __name__ == "__main__":