Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import requests
 import os
 
-API_KEY = os.getenv("AIML_API_KEY") or "your_aimlapi_key_here"
+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 highly knowledgeable and friendly veterinary assistant AI. "
@@ -39,10 +39,26 @@ def handle_symptom_other(symptom):
 def handle_info_other(info):
     return gr.update(visible=(info == "Other"), value="")
 
+def validate_inputs(pet, breed, other_breed, age, gender, weight, topic, symptom, other_symptom, condition, info_topic, other_info):
+    if not pet or not age.strip() or not gender or not weight.strip() or not topic:
+        return False
+    if topic == "Symptom Checker" and (not symptom or (symptom == "Other" and not other_symptom.strip())):
+        return False
+    if topic == "Nutrition" and not condition.strip():
+        return False
+    if topic == "Breed Info" and (not info_topic or (info_topic == "Other" and not other_info.strip())):
+        return False
+    if breed == "Other" and not other_breed.strip():
+        return False
+    return True
+
 def ask_ai(pet, breed, other_breed, age, gender, weight, topic, symptom, other_symptom, condition, info_topic, other_info, user_detail):
     global chat_history, followup_count
     followup_count = 0
 
+    if not validate_inputs(pet, breed, other_breed, age, gender, weight, topic, symptom, other_symptom, condition, info_topic, other_info):
+        return "⚠️ Please complete all required fields before getting advice."
+
     pet = pet.strip()
     final_breed = other_breed.strip() if breed == "Other" else breed
     final_symptom = other_symptom.strip() if symptom == "Other" else symptom
@@ -68,7 +84,7 @@ def ask_ai(pet, breed, other_breed, age, gender, weight, topic, symptom, other_s
             "recommended treatments and home care, expected recovery timeline, preventive measures, and when to seek a vet."
         )
     elif topic == "Nutrition":
-        prompt_parts.append(f"Condition: {condition if condition else 'None'}")
+        prompt_parts.append(f"Condition: {condition}")
         prompt_parts.append(
             "Provide a comprehensive nutrition guide for this breed/condition: best foods available locally (Pakistan), benefits of each food, "
             "recommended feeding quantity and frequency, and tips for maintaining optimal health."
@@ -167,7 +183,7 @@ with gr.Blocks() as demo:
         symptom.change(handle_symptom_other, symptom, other_symptom)
 
     with gr.Column(visible=False) as nutrition_section:
-        condition = gr.Textbox(label="🩺 Condition (Optional, e.g. slim, weak)")
+        condition = gr.Textbox(label="🩺 Condition (Required)")
 
     with gr.Column(visible=False) as breed_info_section:
         info_topic = gr.Dropdown(breed_info_topics, label="ℹ️ Select Info Type")