Update app.py

app.py

@@ -2,13 +2,13 @@ import gradio as gr
 import requests
 import os
 
-API_KEY = os.getenv("AIMLAPI_API_KEY") or "42959a7da8904b0785ad2bb304172793"
+API_KEY = os.getenv("AIML_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. "
     "Provide clear, thorough, and helpful responses to users about pets (dogs and cats). "
-    "Responses should be detailed based on the selected topic — covering diagnosis, treatment, food, precautions, recovery, and fun facts. "
-    "Always encourage consulting a vet for serious issues."
+    "Responses should be detailed, covering causes, symptoms, treatments, nutrition, precautions, "
+    "recovery time, and practical advice. Use friendly, empathetic language and encourage consulting a vet for serious concerns."
 )
 
 chat_history = []
@@ -63,13 +63,22 @@ def ask_ai(pet, breed, other_breed, age, gender, weight, topic, symptom, other_s
 
     if topic == "Symptom Checker":
         prompt_parts.append(f"Symptom: {final_symptom}")
-        prompt_parts.append(f"Describe the symptom in detail and explain possible causes, when it occurs, treatments, recovery time, and vet advice.")
+        prompt_parts.append(
+            "Please provide a detailed explanation including: typical causes, when and how this symptom usually appears, "
+            "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("Recommend suitable food and nutrients (available in Pakistan), with benefits, quantity, and feeding schedule.")
+        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."
+        )
     elif topic == "Breed Info":
         prompt_parts.append(f"Information Requested: {final_info}")
-        prompt_parts.append("Give detailed info specific to this topic for the selected breed.")
+        prompt_parts.append(
+            "Give detailed and interesting information on the selected topic related to the breed, covering key facts, practical tips, "
+            "common concerns, and any special advice."
+        )
 
     if user_detail:
         prompt_parts.append(f"User Detail: {user_detail}")
@@ -85,7 +94,7 @@ def ask_ai(pet, breed, other_breed, age, gender, weight, topic, symptom, other_s
         "model": "gpt-4o-mini-2024-07-18",
         "messages": messages,
         "temperature": 0.7,
-        "max_tokens": 800
+        "max_tokens": 900
     }
 
     headers = {
@@ -134,39 +143,38 @@ def handle_followup(user_input):
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
-# Launch UI
 with gr.Blocks() as demo:
     gr.Markdown("## 🐾 Smart Paws AI Assistant")
 
     with gr.Row():
-        pet = gr.Dropdown(["Cat", "Dog"], label="Pet Type")
-        breed = gr.Dropdown(label="Select Breed", visible=False)
-        other_breed = gr.Textbox(label="Enter Breed", visible=False)
+        pet = gr.Dropdown(["Cat", "Dog"], label="🐕 Pet Type")
+        breed = gr.Dropdown(label="🐾 Select Breed", visible=False)
+        other_breed = gr.Textbox(label="✍️ Enter Breed", visible=False)
 
     pet.change(update_breed, pet, [breed, other_breed])
     breed.change(handle_breed_other, breed, other_breed)
 
     with gr.Row():
-        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)")
+        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="📚 Select Topic")
 
-    topic = gr.Dropdown(["Symptom Checker", "Nutrition", "Breed Info"], label="Select Topic")
-    
     with gr.Column(visible=False) as symptom_section:
-        symptom = gr.Dropdown(symptom_list, label="Select Symptom")
-        other_symptom = gr.Textbox(label="Enter Symptom", visible=False)
+        symptom = gr.Dropdown(symptom_list, label="🤒 Select Symptom")
+        other_symptom = gr.Textbox(label="✍️ Enter Symptom", visible=False)
         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 (Optional, e.g. slim, weak)")
 
     with gr.Column(visible=False) as breed_info_section:
-        info_topic = gr.Dropdown(breed_info_topics, label="Select Info Type")
-        other_info = gr.Textbox(label="Enter Info Topic", visible=False)
+        info_topic = gr.Dropdown(breed_info_topics, label="ℹ️ Select Info Type")
+        other_info = gr.Textbox(label="✍️ Enter Info Topic", visible=False)
         info_topic.change(handle_info_other, info_topic, other_info)
 
-    user_detail = gr.Textbox(label="Extra Detail (optional, max 100 chars)", max_lines=2)
+    user_detail = gr.Textbox(label="📝 Extra Detail (optional, max 100 chars)", max_lines=2)
 
     def show_topic_fields(selected_topic):
         return [
@@ -178,12 +186,12 @@ with gr.Blocks() as demo:
     topic.change(show_topic_fields, topic, [symptom_section, nutrition_section, breed_info_section])
 
     submit_btn = gr.Button("Get Advice")
-    output = gr.Textbox(label="AI Response", lines=10)
+    output = gr.Textbox(label="💬 AI Response", lines=15)
 
     gr.Markdown("### 🔄 Ask Follow-up")
-    followup_input = gr.Textbox(label="Follow-up Question")
+    followup_input = gr.Textbox(label="🗣️ Follow-up Question")
     followup_btn = gr.Button("Ask Follow-up")
-    followup_output = gr.Textbox(label="Follow-up Response", lines=4)
+    followup_output = gr.Textbox(label="💡 Follow-up Response", lines=6)
 
     submit_btn.click(
         ask_ai,