Update app.py

app.py

@@ -1,10 +1,8 @@
 import gradio as gr
-import requests
 import os
 
+# Placeholder for API key
 API_KEY = os.getenv("AIMLAPI_API_KEY") or "your_aimlapi_key_here"
-API_URL = "https://api.aimlapi.com/v1/chat/completions"
-SYSTEM_PROMPT = "You are a helpful and knowledgeable veterinary assistant AI..."
 
 animal_options = {
     "Pet": ["Dog", "Cat", "Rabbit", "Hamster", "Parrot", "Other"],
@@ -13,18 +11,20 @@ animal_options = {
 }
 
 def show_only_selected(category):
-    # Return the selected category markdown text
-    # plus visibility and choices updates for UI components
     return (
-        f"**Selected Category:** {category}",  # Markdown text
-        gr.update(visible=(category == "Pet")),  # pet_btn visible if not selected
+        f"**Selected Category:** {category}",  # Selected category display
+        gr.update(visible=(category == "Pet")),
         gr.update(visible=(category == "Domestic")),
         gr.update(visible=(category == "Wild")),
-        gr.update(visible=True, choices=animal_options[category], value=None),  # animal dropdown
-        gr.update(visible=False),  # other_input hidden initially
-        gr.update(visible=True),   # form visible
-        gr.update(visible=True),   # back to home visible
-        gr.update(visible=False),  # category buttons hidden when one selected
+        gr.update(visible=True, choices=animal_options[category], value=None),  # Animal dropdown
+        gr.update(visible=False),  # Other animal input
+        gr.update(visible=True),  # Info form visible
+        gr.update(visible=True),  # Back button
+        gr.update(visible=True),  # Gender
+        gr.update(visible=True),  # Topic
+        gr.update(visible=True),  # User detail (problem)
+        gr.update(visible=True),  # Submit button
+        gr.update(visible=False),  # Output hidden initially
     )
 
 def update_animals(selected_animal):
@@ -32,16 +32,26 @@ def update_animals(selected_animal):
 
 def back_to_home():
     return (
-        gr.update(value="", visible=False),  # selected_category_markdown
-        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, value=None),  # animal dropdown hidden and cleared
-        gr.update(visible=False, value=""),  # other input hidden and cleared
-        gr.update(visible=False),  # form hidden
-        gr.update(visible=False)   # back home hidden
+        gr.update(value="", visible=False),  # Category label
+        gr.update(visible=True),   # Pet btn
+        gr.update(visible=True),   # Domestic btn
+        gr.update(visible=True),   # Wild btn
+        gr.update(visible=False, value=None),  # Animal dropdown
+        gr.update(visible=False, value=""),  # Other animal input
+        gr.update(visible=False),  # Info form
+        gr.update(visible=False),  # Back button
+        gr.update(visible=False),  # Gender
+        gr.update(visible=False),  # Topic
+        gr.update(visible=False),  # Problem detail
+        gr.update(visible=False),  # Submit button
+        gr.update(visible=False, value="")   # Output box
     )
 
+def ask_ai(category, animal, other, age, gender, height, weight, topic, problem):
+    final_animal = other if animal == "Other" else animal
+    prompt = f"Category: {category}\nAnimal: {final_animal}\nAge: {age}\nGender: {gender}\nHeight: {height}\nWeight: {weight}\nTopic: {topic}\nProblem: {problem}\n"
+    return f"🔍 **You asked:**\n{prompt}\n\n🤖 **AI's advice:**\nThis is a placeholder AI response for the given details."
+
 with gr.Blocks() as demo:
     gr.Markdown("## 🐾 AI Vet Assistant")
 
@@ -50,54 +60,78 @@ with gr.Blocks() as demo:
         domestic_btn = gr.Button("Domestic")
         wild_btn = gr.Button("Wild")
 
-    selected_category_markdown = gr.Markdown("", visible=False)
+    selected_category = gr.Markdown("", visible=False)
     animal = gr.Dropdown(label="Select Animal", choices=[], visible=False)
     other_input = gr.Textbox(label="Enter Animal", visible=False)
-    
+
     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")
         height = gr.Textbox(label="Height (e.g. 30 cm)")
         weight = gr.Textbox(label="Weight (e.g. 10 kg)")
 
+    gender = gr.Dropdown(["Male", "Female", "Unknown"], label="Gender", visible=False)
     topic = gr.Dropdown(["Health", "Nutrition", "Breed Info", "Natural Remedies"], label="Topic", visible=False)
     user_detail = gr.Textbox(label="Problem Detail (max 60 characters)", max_lines=2, visible=False)
-
     submit_btn = gr.Button("Get Advice", visible=False)
     output = gr.Textbox(label="AI Advice", lines=8, visible=False)
+    back_btn = gr.Button("Back to Home", visible=False)
 
-    back_home_btn = gr.Button("Back to Home", visible=False)
-
-    # Here is the key: pass the category name explicitly as input
+    # Button Click Events
     pet_btn.click(show_only_selected, inputs=[gr.State("Pet")], outputs=[
-        selected_category_markdown,
+        selected_category,
         pet_btn, domestic_btn, wild_btn,
         animal, other_input, form_row,
-        back_home_btn
+        back_btn,
+        gender, topic, user_detail,
+        submit_btn, output
     ])
 
     domestic_btn.click(show_only_selected, inputs=[gr.State("Domestic")], outputs=[
-        selected_category_markdown,
+        selected_category,
         pet_btn, domestic_btn, wild_btn,
         animal, other_input, form_row,
-        back_home_btn
+        back_btn,
+        gender, topic, user_detail,
+        submit_btn, output
     ])
 
     wild_btn.click(show_only_selected, inputs=[gr.State("Wild")], outputs=[
-        selected_category_markdown,
+        selected_category,
         pet_btn, domestic_btn, wild_btn,
         animal, other_input, form_row,
-        back_home_btn
+        back_btn,
+        gender, topic, user_detail,
+        submit_btn, output
     ])
 
-    animal.change(update_animals, animal, other_input)
+    animal.change(update_animals, inputs=animal, outputs=other_input)
 
-    back_home_btn.click(back_to_home, inputs=[], outputs=[
-        selected_category_markdown,
+    back_btn.click(back_to_home, outputs=[
+        selected_category,
         pet_btn, domestic_btn, wild_btn,
         animal, other_input, form_row,
-        back_home_btn
+        back_btn,
+        gender, topic, user_detail,
+        submit_btn, output
     ])
 
+    submit_btn.click(
+        ask_ai,
+        inputs=[
+            selected_category,
+            animal,
+            other_input,
+            age,
+            gender,
+            height,
+            weight,
+            topic,
+            user_detail
+        ],
+        outputs=output
+    )
+
+    output.change(lambda x: gr.update(visible=True), inputs=output, outputs=output)
+
 if __name__ == "__main__":
     demo.launch()