Update app.py

app.py

@@ -1,22 +1,24 @@
 import gradio as gr
 import requests
 import os
+from datetime import date
+import pandas as pd
 
+# ---------------------- API Config ----------------------
 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. "
-    "Always provide clear, complete, and well-structured responses about dogs and cats. "
-    "Include all relevant details such as symptoms, causes, treatments, precautions, nutrition, and recovery time where applicable. "
-    "Format the response cleanly using markdown: use bold headings, emojis for clarity, and bullet points where needed. "
-    "Avoid using symbols like ** or ## unless part of markdown. Make the text friendly, professional, and easy to read. "
-    "Always provide a complete and relevant answer within the given token limit. Avoid cutting off in the middle of a sentence."
+    "Provide clear, thorough, and helpful responses to users about pets (dogs and cats). "
+    "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."
 )
 
-
+# ---------------------- AI State ----------------------
 chat_history = []
 followup_count = 0
 
+# ---------------------- Dropdowns ----------------------
 breed_options = {
     "Cat": ["Persian", "Siamese", "Maine Coon", "British Shorthair", "Other"],
     "Dog": ["German Shepherd", "Labrador", "Bulldog", "Pomeranian", "Other"]
@@ -30,6 +32,7 @@ breed_info_topics = [
     "Favourite Food", "Most Common Diseases", "Precautions", "Allergies", "Favourite Activities", "Other"
 ]
 
+# ---------------------- Helper Functions ----------------------
 def update_breed(pet_type):
     return gr.update(choices=breed_options.get(pet_type, []), value=None, visible=True), gr.update(visible=False, value="")
 
@@ -97,7 +100,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": 300
+        "max_tokens": 500
     }
 
     headers = {
@@ -146,63 +149,123 @@ def handle_followup(user_input):
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
-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.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)")
-
-    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.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)")
-
-    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.change(handle_info_other, info_topic, other_info)
-
-    user_detail = gr.Textbox(label="📝 Extra Detail (optional, max 100 chars)", max_lines=2)
-
-    def show_topic_fields(selected_topic):
-        return [
-            gr.update(visible=(selected_topic == "Symptom Checker")),
-            gr.update(visible=(selected_topic == "Nutrition")),
-            gr.update(visible=(selected_topic == "Breed Info"))
-        ]
-
-    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=15)
-
-    gr.Markdown("### 🔄 Ask Follow-up")
-    followup_input = gr.Textbox(label="🗣️ Follow-up Question")
-    followup_btn = gr.Button("Ask Follow-up")
-    followup_output = gr.Textbox(label="💡 Follow-up Response", lines=6)
-
-    submit_btn.click(
-        ask_ai,
-        inputs=[pet, breed, other_breed, age, gender, weight, topic, symptom, other_symptom, condition, info_topic, other_info, user_detail],
-        outputs=output
-    )
+# ---------------------- Vaccination Tracker ----------------------
+vaccination_records = []
+
+def save_vaccination_record(pet_type, breed, pet_name, age, last_date, total_doses, upcoming_date, vaccine_type, notes):
+    record = {
+        "Pet Type": pet_type,
+        "Breed": breed,
+        "Pet Name": pet_name,
+        "Age": age,
+        "Last Vaccination Date": last_date,
+        "Total Doses": total_doses,
+        "Upcoming Dose Date": upcoming_date,
+        "Vaccine Type": vaccine_type,
+        "Notes": notes
+    }
+    vaccination_records.append(record)
+    return f"✅ Record saved for {pet_name} ({breed})"
 
-    followup_btn.click(handle_followup, inputs=followup_input, outputs=followup_output)
+def view_all_vaccination_records():
+    if not vaccination_records:
+        return "📭 No records saved yet."
+    df = pd.DataFrame(vaccination_records)
+    return df
 
+# ---------------------- UI ----------------------
+with gr.Blocks() as demo:
+    gr.Markdown("## 🐾 Smart Paws: AI Assistant + Vaccination Tracker")
+
+    with gr.Tabs():
+        # --- Tab 1: AI Assistant ---
+        with gr.TabItem("🧠 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.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)")
+
+            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.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)")
+
+            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.change(handle_info_other, info_topic, other_info)
+
+            user_detail = gr.Textbox(label="📝 Extra Detail (optional, max 100 chars)", max_lines=2)
+
+            def show_topic_fields(selected_topic):
+                return [
+                    gr.update(visible=(selected_topic == "Symptom Checker")),
+                    gr.update(visible=(selected_topic == "Nutrition")),
+                    gr.update(visible=(selected_topic == "Breed Info"))
+                ]
+
+            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=15)
+
+            gr.Markdown("### 🔄 Ask Follow-up")
+            followup_input = gr.Textbox(label="🗣️ Follow-up Question")
+            followup_btn = gr.Button("Ask Follow-up")
+            followup_output = gr.Textbox(label="💡 Follow-up Response", lines=6)
+
+            submit_btn.click(
+                ask_ai,
+                inputs=[pet, breed, other_breed, age, gender, weight, topic, symptom, other_symptom, condition, info_topic, other_info, user_detail],
+                outputs=output
+            )
+
+            followup_btn.click(handle_followup, inputs=followup_input, outputs=followup_output)
+
+        # --- Tab 2: Vaccination Records ---
+        with gr.TabItem("💉 Pet Vaccination Records"):
+            gr.Markdown("### 💉 Enter Vaccination Details")
+
+            with gr.Row():
+                v_pet_type = gr.Dropdown(["Dog", "Cat"], label="Pet Type")
+                v_breed = gr.Textbox(label="Breed")
+                v_pet_name = gr.Textbox(label="Pet Name")
+                v_age = gr.Textbox(label="Age")
+
+            with gr.Row():
+                v_last_vaccination = gr.Textbox(label="Last Vaccination Date (YYYY-MM-DD)")
+                v_total_doses = gr.Number(label="Total Doses Taken")
+                v_upcoming_dose = gr.Textbox(label="Upcoming Dose Date (YYYY-MM-DD)")
+
+            v_vaccine_type = gr.Dropdown(["Rabies", "DHPP", "FVRCP", "Bordetella", "Leptospirosis", "Other"], label="Vaccine Type")
+            v_notes = gr.Textbox(label="Additional Notes", lines=3)
+
+            save_btn = gr.Button("Save Record")
+            save_output = gr.Textbox(label="📋 Status")
+
+            view_btn = gr.Button("📑 View All Records")
+            table_output = gr.Dataframe(label="📊 Saved Records", visible=True)
+
+            save_btn.click(
+                save_vaccination_record,
+                inputs=[v_pet_type, v_breed, v_pet_name, v_age, v_last_vaccination, v_total_doses, v_upcoming_dose, v_vaccine_type, v_notes],
+                outputs=save_output
+            )
+
+            view_btn.click(view_all_vaccination_records, inputs=[], outputs=table_output)
+
+# ---------------------- Run App ----------------------
 if __name__ == "__main__":
     demo.launch()