Update app.py

app.py

@@ -14,6 +14,8 @@ breeds = {
     "Dog": ["German Shepherd", "Labrador", "Bulldog", "Pomeranian", "Other"]
 }
 
+symptom_list = ["Vomiting", "Coughing", "Itching", "Diarrhea", "Loss of Appetite", "Limping", "Sneezing", "Other"]
+
 def update_breeds(pet):
     options = breeds.get(pet, [])
     return gr.update(choices=options, visible=True, value=None), gr.update(visible=False, value="")
@@ -21,11 +23,15 @@ def update_breeds(pet):
 def show_custom_breed(breed_selected):
     return gr.update(visible=(breed_selected == "Other"), value="")
 
-def ask_ai(pet, breed, custom_breed, age, gender, weight, topic, symptom, condition, user_detail):
+def show_custom_symptom(symptom_selected):
+    return gr.update(visible=(symptom_selected == "Other"), value="")
+
+def ask_ai(pet, breed, custom_breed, age, gender, weight, topic, symptom, custom_symptom, condition, user_detail):
     global chat_history, followup_count
     followup_count = 0
     user_detail = user_detail.strip()[:60]
     breed_name = custom_breed if breed == "Other" and custom_breed else breed
+    symptom_name = custom_symptom if symptom == "Other" and custom_symptom else symptom
 
     prompt = f"""Pet: {pet}
 Breed: {breed_name}
@@ -33,7 +39,7 @@ Age: {age}, Gender: {gender}, Weight: {weight}
 Topic: {topic}"""
 
     if topic == "Symptom Checker":
-        prompt += f"\nSymptom: {symptom}"
+        prompt += f"\nSymptom: {symptom_name}"
     elif topic == "Nutrition":
         prompt += f"\nCondition: {condition}"
     prompt += f"\nDetail: {user_detail}\nGive a short and relevant response. If critical, recommend a vet visit."
@@ -109,17 +115,21 @@ with gr.Blocks() as demo:
         weight = gr.Textbox(label="Weight (e.g. 10 kg)")
 
     topic = gr.Dropdown(["Symptom Checker", "Nutrition", "Breed Info"], label="Topic")
-    symptom = gr.Dropdown(["Vomiting", "Coughing", "Itching", "Diarrhea", "Loss of Appetite", "Limping", "Sneezing"], label="Select Symptom", visible=False)
+    symptom = gr.Dropdown(symptom_list, label="Select Symptom", visible=False)
+    custom_symptom = gr.Textbox(label="Enter Symptom", visible=False)
     condition = gr.Textbox(label="Condition (optional: slim, weak, etc.)", visible=False)
     user_detail = gr.Textbox(label="Problem Detail (max 60 characters)", max_lines=2)
 
     def toggle_topic_fields(selected_topic):
         return (
             gr.update(visible=(selected_topic == "Symptom Checker")),
-            gr.update(visible=(selected_topic == "Nutrition"))
+            gr.update(visible=(selected_topic == "Nutrition")),
+            gr.update(visible=False),
+            gr.update(visible=False)
         )
 
-    topic.change(toggle_topic_fields, topic, [symptom, condition])
+    topic.change(toggle_topic_fields, topic, [symptom, condition, custom_symptom, custom_breed])
+    symptom.change(show_custom_symptom, symptom, custom_symptom)
 
     submit_btn = gr.Button("Get Advice")
     output = gr.Textbox(label="AI Advice", lines=8)
@@ -131,7 +141,7 @@ with gr.Blocks() as demo:
 
     submit_btn.click(
         ask_ai,
-        inputs=[pet, breed, custom_breed, age, gender, weight, topic, symptom, condition, user_detail],
+        inputs=[pet, breed, custom_breed, age, gender, weight, topic, symptom, custom_symptom, condition, user_detail],
         outputs=output
     )