Update app.py

app.py

@@ -2,51 +2,45 @@ import gradio as gr
 import requests
 import os
 
-API_KEY = os.getenv("AIMLAPI_API_KEY") or "42959a7da8904b0785ad2bb304172793"
+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. Give short but useful answers regarding pet health, nutrition, breeds, and symptoms. 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 and animal health, nutrition, breeds, and conditions. Keep it simple and consult a vet in critical cases."
 
 chat_history = []
 followup_count = 0
 
 breeds = {
-    "Cat": ["Persian", "Maine Coon", "Siamese", "Turkish Van"],
-    "Dog": ["German Shepherd", "Labrador", "Bulldog", "Pomeranian"]
+    "Cat": ["Persian", "Maine Coon", "Siamese", "Turkish Van", "Other"],
+    "Dog": ["German Shepherd", "Labrador", "Bulldog", "Pomeranian", "Other"]
 }
 
-symptoms_list = ["Vomiting", "Loss of Appetite", "Limping", "Itchy Skin", "Coughing", "Diarrhea", "Lethargy"]
-
 def update_breeds(pet):
-    return gr.update(choices=breeds.get(pet, []), visible=True, value=None)
-
-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)
-    )
+    options = breeds.get(pet, [])
+    return gr.update(choices=options, visible=True, value=None), gr.update(visible=False, value="")
+
+def show_custom_breed(breed_selected):
+    return gr.update(visible=(breed_selected == "Other"), value="")
 
-def ask_ai(pet, breed, age, gender, weight, topic, symptom, condition, user_detail):
+def ask_ai(pet, breed, custom_breed, age, gender, weight, topic, 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
+
+    prompt = f"""Pet: {pet}
+Breed: {breed_name}
+Age: {age}, Gender: {gender}, Weight: {weight}
+Topic: {topic}"""
 
     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."
+        prompt += f"\nSymptom: {symptom}"
     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}"
+        prompt += f"\nCondition: {condition}"
+    prompt += f"\nDetail: {user_detail}\nGive a short and relevant response. If critical, recommend a vet visit."
 
     messages = [
         {"role": "system", "content": SYSTEM_PROMPT},
-        {"role": "user", "content": user_prompt}
+        {"role": "user", "content": prompt}
     ]
 
     payload = {
@@ -76,21 +70,17 @@ def handle_followup(user_input):
     global chat_history, followup_count
     if followup_count >= 5:
         return "⚠️ Max of 5 follow-up questions reached."
-
     chat_history.append({"role": "user", "content": user_input})
-
     payload = {
         "model": "gpt-4o-mini-2024-07-18",
         "messages": chat_history,
         "temperature": 0.5,
         "max_tokens": 200
     }
-
     headers = {
         "Content-Type": "application/json",
         "Authorization": f"Bearer {API_KEY}"
     }
-
     try:
         response = requests.post(API_URL, json=payload, headers=headers)
         response.raise_for_status()
@@ -102,29 +92,35 @@ def handle_followup(user_input):
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
-# Launch app
 with gr.Blocks() as demo:
     gr.Markdown("## 🐾 Smart Paws AI Assistant")
 
     with gr.Row():
         pet = gr.Dropdown(["Cat", "Dog"], label="Pet")
-        breed = gr.Dropdown(label="Select Breed")
+        breed = gr.Dropdown(label="Select Breed", visible=False)
+    custom_breed = gr.Textbox(label="Enter Breed Name", visible=False)
 
-    pet.change(update_breeds, pet, breed)
+    pet.change(update_breeds, pet, [breed, custom_breed])
+    breed.change(show_custom_breed, breed, custom_breed)
 
     with gr.Row():
-        age = gr.Textbox(label="Age (e.g. 2 years or 6 months)")
+        age = gr.Textbox(label="Age (e.g. 2 years or 8 months)")
         gender = gr.Dropdown(["Male", "Female", "Unknown"], label="Gender")
         weight = gr.Textbox(label="Weight (e.g. 10 kg)")
 
     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)
+    symptom = gr.Dropdown(["Vomiting", "Coughing", "Itching", "Diarrhea", "Loss of Appetite", "Limping", "Sneezing"], label="Select 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)
 
-    topic.change(update_ui_by_topic, topic, [symptom, condition, breed_info_trigger])
+    def toggle_topic_fields(selected_topic):
+        return (
+            gr.update(visible=(selected_topic == "Symptom Checker")),
+            gr.update(visible=(selected_topic == "Nutrition"))
+        )
+
+    topic.change(toggle_topic_fields, topic, [symptom, condition])
 
-    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)
 
@@ -135,9 +131,10 @@ with gr.Blocks() as demo:
 
     submit_btn.click(
         ask_ai,
-        inputs=[pet, breed, age, gender, weight, topic, symptom, condition, user_detail],
+        inputs=[pet, breed, custom_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__":