Update app.py

app.py

@@ -2,38 +2,47 @@ import gradio as gr
 import requests
 import os
 
-API_KEY = os.getenv("AIMLAPI_API_KEY") or "your_aimlapi_key_here"
+API_KEY = os.getenv("AIMLAPI_API_KEY") or "42959a7da8904b0785ad2bb304172793"
 API_URL = "https://api.aimlapi.com/v1/chat/completions"
-SYSTEM_PROMPT = "You are a helpful and knowledgeable veterinary assistant AI. Give short but useful answers regarding pet and animal health, nutrition, breeds, and natural remedies. Keep it simple and consult a vet in critical cases."
+SYSTEM_PROMPT = "You are a helpful and knowledgeable veterinary assistant AI. Give short but useful answers regarding pet health, nutrition, breeds, and symptoms. Keep it simple and consult a vet in critical cases."
 
 chat_history = []
 followup_count = 0
 
-animal_options = {
-    "Pet": ["Dog", "Cat", "Rabbit", "Hamster", "Parrot", "Other"],
-    "Domestic": ["Cow", "Goat", "Sheep", "Chicken", "Horse", "Buffalo", "Camel", "Donkey", "Other"],
-    "Wild": ["Tiger", "Lion", "Elephant", "Deer", "Bear", "Leopard", "Wolf", "Giraffe", "Other"]
+breeds = {
+    "Cat": ["Persian", "Maine Coon", "Siamese", "Turkish Van"],
+    "Dog": ["German Shepherd", "Labrador", "Bulldog", "Pomeranian"]
 }
 
-def update_animals(category):
-    animals = animal_options.get(category, [])
-    return gr.update(choices=animals, visible=True, value=None), gr.update(visible=False, value="")
+symptoms_list = ["Vomiting", "Loss of Appetite", "Limping", "Itchy Skin", "Coughing", "Diarrhea", "Lethargy"]
 
-def handle_other_animal(selected):
-    return gr.update(visible=(selected == "Other"), value="")
+def update_breeds(pet):
+    return gr.update(choices=breeds.get(pet, []), visible=True, value=None)
 
-def ask_ai(category, selected_animal, other_animal, age, gender, height, weight, topic, user_detail):
+def update_ui_by_topic(topic):
+    symptom_visible = topic == "Symptom Checker"
+    condition_visible = topic == "Nutrition"
+    breed_info = topic == "Breed Info"
+    return (
+        gr.update(visible=symptom_visible),
+        gr.update(visible=condition_visible),
+        gr.update(visible=breed_info)
+    )
+
+def ask_ai(pet, breed, age, gender, weight, topic, symptom, condition, user_detail):
     global chat_history, followup_count
     followup_count = 0
     user_detail = user_detail.strip()[:60]
-    animal = other_animal.strip() if selected_animal == "Other" else selected_animal
 
-    user_prompt = f"""Animal Category: {category}
-Animal: {animal}
-Age: {age}, Gender: {gender}, Height: {height}, Weight: {weight}
-Topic: {topic}
-Detail: {user_detail}
-Give a short and relevant response."""
+    if topic == "Symptom Checker":
+        user_prompt = f"Pet: {pet}, Breed: {breed}, Age: {age}, Gender: {gender}, Symptom: {symptom}\nDiagnose possible diseases and suggest if vet is needed."
+    elif topic == "Nutrition":
+        user_prompt = f"Pet: {pet}, Breed: {breed}, Age: {age}, Gender: {gender}, Weight: {weight}, Condition: {condition}\nSuggest appropriate diet and supplements in Pakistan."
+    elif topic == "Breed Info":
+        user_prompt = f"Give detailed breed info for a {breed} {pet} including favorite food, common diseases, allergies, activity level, precautions, grooming needs, and health care."
+
+    if user_detail:
+        user_prompt += f"\nExtra detail: {user_detail}"
 
     messages = [
         {"role": "system", "content": SYSTEM_PROMPT},
@@ -95,25 +104,27 @@ def handle_followup(user_input):
 
 # Launch app
 with gr.Blocks() as demo:
-    gr.Markdown("## 🐾 AI Vet Assistant")
+    gr.Markdown("## 🐾 Smart Paws AI Assistant")
 
     with gr.Row():
-        category = gr.Dropdown(["Pet", "Domestic", "Wild"], label="Animal Category")
-        animal = gr.Dropdown(label="Select Animal", visible=False)
-        other_input = gr.Textbox(label="Enter Animal", visible=False)
+        pet = gr.Dropdown(["Cat", "Dog"], label="Pet")
+        breed = gr.Dropdown(label="Select Breed")
 
-    category.change(update_animals, category, [animal, other_input])
-    animal.change(handle_other_animal, animal, other_input)
+    pet.change(update_breeds, pet, breed)
 
     with gr.Row():
-        age = gr.Textbox(label="Age (e.g. 2 years)")
+        age = gr.Textbox(label="Age (e.g. 2 years or 6 months)")
         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)")
 
-    topic = gr.Dropdown(["Health", "Nutrition", "Breed Info", "Natural Remedies"], label="Topic")
-    user_detail = gr.Textbox(label="Problem Detail (max 60 characters)", max_lines=2)
+    topic = gr.Dropdown(["Symptom Checker", "Nutrition", "Breed Info"], label="Topic")
+    symptom = gr.Dropdown(choices=symptoms_list, label="Select Symptom", visible=False)
+    condition = gr.Textbox(label="Condition (optional - e.g. weak, slim)", visible=False)
+    breed_info_trigger = gr.Textbox(visible=False)
+
+    topic.change(update_ui_by_topic, topic, [symptom, condition, breed_info_trigger])
 
+    user_detail = gr.Textbox(label="Extra Detail (optional, max 60 characters)", max_lines=2)
     submit_btn = gr.Button("Get Advice")
     output = gr.Textbox(label="AI Advice", lines=8)
 
@@ -122,7 +133,11 @@ with gr.Blocks() as demo:
     followup_btn = gr.Button("Ask Follow-up")
     followup_output = gr.Textbox(label="Follow-up Response", lines=4)
 
-    submit_btn.click(ask_ai, inputs=[category, animal, other_input, age, gender, height, weight, topic, user_detail], outputs=output)
+    submit_btn.click(
+        ask_ai,
+        inputs=[pet, breed, age, gender, weight, topic, symptom, condition, user_detail],
+        outputs=output
+    )
     followup_btn.click(handle_followup, inputs=followup_input, outputs=followup_output)
 
 if __name__ == "__main__":