Update app.py

app.py

@@ -16,42 +16,36 @@ animal_options = {
 }
 
 def show_only_selected(category):
-    # Show only the selected category block
-    # Return visibility and selected category name
     return (
-        gr.update(visible=True, value=category),
+        gr.update(visible=True, value=f"**Selected Category:** {category}"),
         gr.update(visible=(category == "Pet")),
         gr.update(visible=(category == "Domestic")),
         gr.update(visible=(category == "Wild")),
-        gr.update(visible=True),  # animal dropdown visible
-        gr.update(choices=animal_options.get(category, []), value=None),
-        gr.update(visible=False),  # other animal input hidden initially
-        gr.update(visible=True),  # form visible
-        gr.update(visible=True),  # Back to Home button visible
-        gr.update(visible=False)  # category selection blocks hidden
+        gr.update(visible=True, choices=animal_options.get(category, []), value=None),
+        gr.update(visible=False),
+        gr.update(visible=True),
+        gr.update(visible=True),
     )
 
 def update_animals(selected_animal):
     return gr.update(visible=(selected_animal == "Other"), value="")
 
 def back_to_home():
-    # Show the 3 category blocks and hide the form & animal dropdown etc
     return (
-        gr.update(visible=False),  # selected_category_markdown hidden
-        gr.update(visible=True),   # pet_btn visible
-        gr.update(visible=True),   # domestic_btn visible
-        gr.update(visible=True),   # wild_btn visible
-        gr.update(visible=False),  # animal dropdown hidden
-        gr.update(value=None, visible=False),  # clear animal dropdown
-        gr.update(visible=False, value=""),   # other animal input hidden
-        gr.update(visible=False),              # form area hidden
-        gr.update(visible=False),              # back to home button hidden
-        gr.update(visible=True)                # category buttons visible
+        gr.update(visible=False, value=""),
+        gr.update(visible=True),
+        gr.update(visible=True),
+        gr.update(visible=True),
+        gr.update(visible=False, value=None),
+        gr.update(visible=False, value=""),
+        gr.update(visible=False),
+        gr.update(visible=False),
     )
 
-def ask_ai(category, selected_animal, other_animal, age, gender, height, weight, topic, user_detail):
+def ask_ai(category_md, selected_animal, other_animal, age, gender, height, weight, topic, user_detail):
     global chat_history, followup_count
     followup_count = 0
+    category = category_md.replace("**Selected Category:** ", "").strip()
     user_detail = user_detail.strip()[:60]
     animal = other_animal.strip() if selected_animal == "Other" else selected_animal
 
@@ -120,11 +114,11 @@ def handle_followup(user_input):
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
+
 with gr.Blocks() as demo:
 
     gr.Markdown("## 🐾 AI Vet Assistant")
 
-    # Category selection buttons - visible on home
     with gr.Row():
         pet_btn = gr.Button("Pet", visible=True)
         domestic_btn = gr.Button("Domestic", visible=True)
@@ -132,11 +126,9 @@ with gr.Blocks() as demo:
 
     selected_category_markdown = gr.Markdown("", visible=False)
 
-    # Animal dropdown and other input
     animal = gr.Dropdown(label="Select Animal", choices=[], visible=False)
     other_input = gr.Textbox(label="Enter Animal", visible=False)
 
-    # Form fields - visible only after category selected
     with gr.Row(visible=False) as form_row:
         age = gr.Textbox(label="Age (e.g. 2 years)")
         gender = gr.Dropdown(["Male", "Female", "Unknown"], label="Gender")
@@ -149,31 +141,46 @@ with gr.Blocks() as demo:
     submit_btn = gr.Button("Get Advice", visible=False)
     output = gr.Textbox(label="AI Advice", lines=8, visible=False)
 
-    # Follow-up question section
-    gr.Markdown("### 🔄 Follow-up Questions", visible=False, elem_id="followup_header")
+    followup_header = gr.Markdown("### 🔄 Follow-up Questions", visible=False)
     followup_input = gr.Textbox(label="Your Follow-up", visible=False)
     followup_btn = gr.Button("Ask Follow-up", visible=False)
     followup_output = gr.Textbox(label="Follow-up Response", lines=4, visible=False)
 
     back_home_btn = gr.Button("Back to Home", visible=False)
 
-    # Events for category buttons
     pet_btn.click(show_only_selected, inputs=[], outputs=[
-        selected_category_markdown, pet_btn, domestic_btn, wild_btn,
-        animal, other_input, form_row, back_home_btn, pet_btn, domestic_btn
-    ], _js=None)  # Remove _js to avoid error
+        selected_category_markdown,
+        pet_btn,
+        domestic_btn,
+        wild_btn,
+        animal,
+        other_input,
+        form_row,
+        back_home_btn
+    ])
 
     domestic_btn.click(show_only_selected, inputs=[], outputs=[
-        selected_category_markdown, pet_btn, domestic_btn, wild_btn,
-        animal, other_input, form_row, back_home_btn, pet_btn, domestic_btn
-    ], _js=None)
+        selected_category_markdown,
+        pet_btn,
+        domestic_btn,
+        wild_btn,
+        animal,
+        other_input,
+        form_row,
+        back_home_btn
+    ])
 
     wild_btn.click(show_only_selected, inputs=[], outputs=[
-        selected_category_markdown, pet_btn, domestic_btn, wild_btn,
-        animal, other_input, form_row, back_home_btn, pet_btn, domestic_btn
-    ], _js=None)
+        selected_category_markdown,
+        pet_btn,
+        domestic_btn,
+        wild_btn,
+        animal,
+        other_input,
+        form_row,
+        back_home_btn
+    ])
 
-    # Properly link animal dropdown to show/hide other input
     animal.change(update_animals, animal, other_input)
 
     submit_btn.click(
@@ -188,10 +195,33 @@ with gr.Blocks() as demo:
         back_to_home,
         inputs=[],
         outputs=[
-            selected_category_markdown, pet_btn, domestic_btn, wild_btn,
-            animal, other_input, form_row, back_home_btn, pet_btn, domestic_btn
+            selected_category_markdown,
+            pet_btn,
+            domestic_btn,
+            wild_btn,
+            animal,
+            other_input,
+            form_row,
+            back_home_btn
         ]
     )
 
-if __name__ == "__main__":
-    demo.launch()
+    def show_form_parts(show: bool):
+        return (
+            gr.update(visible=show),
+            gr.update(visible=show),
+            gr.update(visible=show),
+            gr.update(visible=show),
+            gr.update(visible=show),
+            gr.update(visible=show),
+            gr.update(visible=show),
+            gr.update(visible=show),
+        )
+
+    pet_btn.click(show_form_parts, inputs=[], outputs=[topic, user_detail, submit_btn, output, followup_header, followup_input, followup_btn, followup_output])
+    domestic_btn.click(show_form_parts, inputs=[], outputs=[topic, user_detail, submit_btn, output, followup_header, followup_input, followup_btn, followup_output])
+    wild_btn.click(show_form_parts, inputs=[], outputs=[topic, user_detail, submit_btn, output, followup_header, followup_input, followup_btn, followup_output])
+
+    back_home_btn.click(lambda: show_form_parts(False), inputs=[], outputs=[topic, user_detail, submit_btn, output, followup_header, followup_input, followup_btn, followup_output])
+
+demo.launch()