Update app.py

app.py

@@ -16,13 +16,23 @@ If the information is not in the context, respond with "I don't find that inform
 Keep responses to 1-2 lines maximum.
 """.strip()
 
-# Pre-populated questions
+# Expanded pre-populated questions
 PREDEFINED_QUESTIONS = [
     "Select a question...",
     "Tell me how can I navigate to a specific pose - include replanning aspects in your answer.",
-    "Can you provide me with code for this task?"
+    "Can you provide me with code for this task?",
+    "How do I set up obstacle avoidance in ROS2 navigation?",
+    "What are the key parameters for tuning the nav2 planner?",
+    "How do I integrate custom recovery behaviors?"
 ]
 
+# Helper text for tooltip
+DROPDOWN_TOOLTIP = """
+You can either:
+• Select a predefined question from this dropdown
+• Type your own question in the text box below
+"""
+
 def generate_prompt(context: str, question: str, system_prompt: str = DEFAULT_SYSTEM_PROMPT) -> str:
     return f"""
 [INST] <<SYS>>
@@ -68,6 +78,10 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
         # Initialize chat history if None
         history = history or []
         
+        if not message.strip():
+            history.append((message, "Please enter a question or select one from the dropdown menu."))
+            return history
+            
         model, tokenizer = initialize_model()
         
         # Get context from database
@@ -105,18 +119,28 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
         history.append((message, f"An error occurred: {str(e)}"))
         return history
 
+def clear_input():
+    return gr.Textbox.update(value="")
+
 # Create the Gradio interface
 with gr.Blocks(title="ROS2 Expert Assistant") as demo:
     gr.Markdown("# ROS2 Expert Assistant")
     gr.Markdown("Ask questions about ROS2, navigation, and robotics. I'll provide concise answers based on the available documentation.")
     
     with gr.Row():
-        # Dropdown for predefined questions
-        question_dropdown = gr.Dropdown(
-            choices=PREDEFINED_QUESTIONS,
-            value="Select a question...",
-            label="Pre-defined Questions"
-        )
+        with gr.Column(scale=8):
+            # Dropdown for predefined questions
+            question_dropdown = gr.Dropdown(
+                choices=PREDEFINED_QUESTIONS,
+                value="Select a question...",
+                label="Pre-defined Questions"
+            )
+        with gr.Column(scale=1):
+            # Info icon with tooltip
+            gr.Markdown(
+                """<div title="{}">ℹ️</div>""".format(DROPDOWN_TOOLTIP),
+                elem_classes=["tooltip"]
+            )
         
     with gr.Row():
         # Chat interface
@@ -162,6 +186,16 @@ with gr.Blocks(title="ROS2 Expert Assistant") as demo:
             label="Top-p"
         )
     
+    # Add custom CSS for tooltip
+    gr.Markdown("""
+        <style>
+        .tooltip {
+            cursor: help;
+            font-size: 1.2em;
+        }
+        </style>
+    """)
+    
     # Event handlers
     question_dropdown.change(
         question_selected,
@@ -169,8 +203,14 @@ with gr.Blocks(title="ROS2 Expert Assistant") as demo:
         outputs=[msg]
     )
     
+    def submit_and_clear(message, history, system_message, max_tokens, temperature, top_p):
+        # First get the response
+        new_history = respond(message, history, system_message, max_tokens, temperature, top_p)
+        # Then clear the input
+        return new_history, gr.Textbox.update(value="")
+    
     submit.click(
-        respond,
+        submit_and_clear,
         inputs=[
             msg,
             chatbot,
@@ -179,12 +219,12 @@ with gr.Blocks(title="ROS2 Expert Assistant") as demo:
             temperature,
             top_p
         ],
-        outputs=[chatbot]
+        outputs=[chatbot, msg]
     )
     
-    clear.click(lambda: None, None, chatbot, queue=False)
+    clear.click(lambda: (None, ""), None, [chatbot, msg], queue=False)
     msg.submit(
-        respond,
+        submit_and_clear,
         inputs=[
             msg,
             chatbot,
@@ -193,7 +233,7 @@ with gr.Blocks(title="ROS2 Expert Assistant") as demo:
             temperature,
             top_p
         ],
-        outputs=[chatbot]
+        outputs=[chatbot, msg]
     )
 
 if __name__ == "__main__":