Update app.py

app.py

@@ -1,8 +1,13 @@
 import gradio as gr
+import requests
 import os
 
-# Placeholder for API key
-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 and animal health, nutrition, breeds, and natural remedies. Keep it simple and consult a vet in critical cases."
+
+chat_history = []
+followup_count = 0
 
 animal_options = {
     "Pet": ["Dog", "Cat", "Rabbit", "Hamster", "Parrot", "Other"],
@@ -10,128 +15,147 @@ animal_options = {
     "Wild": ["Tiger", "Lion", "Elephant", "Deer", "Bear", "Leopard", "Wolf", "Giraffe", "Other"]
 }
 
-def show_only_selected(category):
+def update_animals(category):
+    animals = animal_options.get(category, [])
     return (
-        f"**Selected Category:** {category}",  # Selected category display
-        gr.update(visible=(category == "Pet")),
-        gr.update(visible=(category == "Domestic")),
-        gr.update(visible=(category == "Wild")),
-        gr.update(visible=True, choices=animal_options[category], value=None),  # Animal dropdown
-        gr.update(visible=False),  # Other animal input
-        gr.update(visible=True),  # Info form visible
-        gr.update(visible=True),  # Back button
-        gr.update(visible=True),  # Gender
-        gr.update(visible=True),  # Topic
-        gr.update(visible=True),  # User detail (problem)
-        gr.update(visible=True),  # Submit button
-        gr.update(visible=False),  # Output hidden initially
+        gr.update(choices=animals, visible=True, value=None),
+        gr.update(visible=False, value=""),
+        gr.update(visible=True),  # form visible
+        gr.update(visible=True),  # back button visible
+        gr.update(visible=False),  # Pet button hidden
+        gr.update(visible=False),  # Domestic button hidden
+        gr.update(visible=False),  # Wild button hidden
+        gr.update(visible=True, value=category)  # Show only selected category at top
     )
 
-def update_animals(selected_animal):
-    return gr.update(visible=(selected_animal == "Other"), value="")
-
-def back_to_home():
+def handle_other_animal(selected):
+    return gr.update(visible=(selected == "Other"), value="")
+
+def ask_ai(category, selected_animal, other_animal, age, gender, height, weight, topic, 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."""
+
+    messages = [
+        {"role": "system", "content": SYSTEM_PROMPT},
+        {"role": "user", "content": user_prompt}
+    ]
+
+    payload = {
+        "model": "gpt-4o-mini-2024-07-18",
+        "messages": messages,
+        "temperature": 0.7,
+        "max_tokens": 500
+    }
+
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {API_KEY}"
+    }
+
+    try:
+        response = requests.post(API_URL, json=payload, headers=headers)
+        response.raise_for_status()
+        result = response.json()
+        answer = result['choices'][0]['message']['content'].strip()
+        chat_history.clear()
+        chat_history.extend(messages + [{"role": "assistant", "content": answer}])
+        return answer
+    except Exception as e:
+        return f"❌ Error: {str(e)}"
+
+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()
+        result = response.json()
+        reply = result['choices'][0]['message']['content'].strip()
+        chat_history.append({"role": "assistant", "content": reply})
+        followup_count += 1
+        return reply
+    except Exception as e:
+        return f"❌ Error: {str(e)}"
+
+def go_home():
     return (
-        gr.update(value="", visible=False),  # Category label
-        gr.update(visible=True),   # Pet btn
-        gr.update(visible=True),   # Domestic btn
-        gr.update(visible=True),   # Wild btn
-        gr.update(visible=False, value=None),  # Animal dropdown
-        gr.update(visible=False, value=""),  # Other animal input
-        gr.update(visible=False),  # Info form
-        gr.update(visible=False),  # Back button
-        gr.update(visible=False),  # Gender
-        gr.update(visible=False),  # Topic
-        gr.update(visible=False),  # Problem detail
-        gr.update(visible=False),  # Submit button
-        gr.update(visible=False, value="")   # Output box
+        gr.update(visible=False),  # form hidden
+        gr.update(visible=True),  # Pet visible
+        gr.update(visible=True),  # Domestic visible
+        gr.update(visible=True),  # Wild visible
+        gr.update(visible=False),  # animal dropdown
+        gr.update(visible=False),  # other input
+        gr.update(visible=False),  # back button
+        gr.update(visible=False, value="")  # selected category label hidden
     )
 
-def ask_ai(category, animal, other, age, gender, height, weight, topic, problem):
-    final_animal = other if animal == "Other" else animal
-    prompt = f"Category: {category}\nAnimal: {final_animal}\nAge: {age}\nGender: {gender}\nHeight: {height}\nWeight: {weight}\nTopic: {topic}\nProblem: {problem}\n"
-    return f"🔍 **You asked:**\n{prompt}\n\n🤖 **AI's advice:**\nThis is a placeholder AI response for the given details."
-
 with gr.Blocks() as demo:
-    gr.Markdown("## 🐾 AI Vet Assistant")
+    gr.Markdown("## 🐾 AI Animal Assistant")
+
+    selected_label = gr.Textbox(visible=False, interactive=False)
 
     with gr.Row():
-        pet_btn = gr.Button("Pet")
+        pet_btn = gr.Button("Pets")
         domestic_btn = gr.Button("Domestic")
         wild_btn = gr.Button("Wild")
 
-    selected_category = gr.Markdown("", visible=False)
-    animal = gr.Dropdown(label="Select Animal", choices=[], visible=False)
-    other_input = gr.Textbox(label="Enter Animal", visible=False)
-
-    with gr.Row(visible=False) as form_row:
-        age = gr.Textbox(label="Age (e.g. 2 years)")
-        height = gr.Textbox(label="Height (e.g. 30 cm)")
-        weight = gr.Textbox(label="Weight (e.g. 10 kg)")
-
-    gender = gr.Dropdown(["Male", "Female", "Unknown"], label="Gender", visible=False)
-    topic = gr.Dropdown(["Health", "Nutrition", "Breed Info", "Natural Remedies"], label="Topic", visible=False)
-    user_detail = gr.Textbox(label="Problem Detail (max 60 characters)", max_lines=2, visible=False)
-    submit_btn = gr.Button("Get Advice", visible=False)
-    output = gr.Textbox(label="AI Advice", lines=8, visible=False)
-    back_btn = gr.Button("Back to Home", visible=False)
-
-    # Button Click Events
-    pet_btn.click(show_only_selected, inputs=[gr.State("Pet")], outputs=[
-        selected_category,
-        pet_btn, domestic_btn, wild_btn,
-        animal, other_input, form_row,
-        back_btn,
-        gender, topic, user_detail,
-        submit_btn, output
-    ])
-
-    domestic_btn.click(show_only_selected, inputs=[gr.State("Domestic")], outputs=[
-        selected_category,
-        pet_btn, domestic_btn, wild_btn,
-        animal, other_input, form_row,
-        back_btn,
-        gender, topic, user_detail,
-        submit_btn, output
-    ])
-
-    wild_btn.click(show_only_selected, inputs=[gr.State("Wild")], outputs=[
-        selected_category,
-        pet_btn, domestic_btn, wild_btn,
-        animal, other_input, form_row,
-        back_btn,
-        gender, topic, user_detail,
-        submit_btn, output
-    ])
-
-    animal.change(update_animals, inputs=animal, outputs=other_input)
-
-    back_btn.click(back_to_home, outputs=[
-        selected_category,
-        pet_btn, domestic_btn, wild_btn,
-        animal, other_input, form_row,
-        back_btn,
-        gender, topic, user_detail,
-        submit_btn, output
-    ])
-
-    submit_btn.click(
-        ask_ai,
-        inputs=[
-            selected_category,
-            animal,
-            other_input,
-            age,
-            gender,
-            height,
-            weight,
-            topic,
-            user_detail
-        ],
-        outputs=output
-    )
+    with gr.Column(visible=False) as form_section:
+        category = selected_label
+        animal = gr.Dropdown(label="Select Animal")
+        other_input = gr.Textbox(label="Enter Animal", visible=False)
+        animal.change(handle_other_animal, animal, other_input)
+
+        with gr.Row():
+            age = gr.Textbox(label="Age (e.g. 2 years)")
+            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)
+
+        submit_btn = gr.Button("Get Advice")
+        output = gr.Textbox(label="AI Advice", lines=8)
+
+        gr.Markdown("### 🔄 Follow-up Questions")
+        followup_input = gr.Textbox(label="Your Follow-up")
+        followup_btn = gr.Button("Ask Follow-up")
+        followup_output = gr.Textbox(label="Follow-up Response", lines=4)
+
+        back_btn = gr.Button("🔙 Back to Home")
+
+    pet_btn.click(fn=update_animals, inputs=[pet_btn], outputs=[animal, other_input, form_section, back_btn, pet_btn, domestic_btn, wild_btn, selected_label], show_progress=False)
+    domestic_btn.click(fn=update_animals, inputs=[domestic_btn], outputs=[animal, other_input, form_section, back_btn, pet_btn, domestic_btn, wild_btn, selected_label], show_progress=False)
+    wild_btn.click(fn=update_animals, inputs=[wild_btn], outputs=[animal, other_input, form_section, back_btn, pet_btn, domestic_btn, wild_btn, selected_label], show_progress=False)
 
-    output.change(lambda x: gr.update(visible=True), inputs=output, outputs=output)
+    submit_btn.click(ask_ai, inputs=[category, animal, other_input, age, gender, height, weight, topic, user_detail], outputs=output)
+    followup_btn.click(handle_followup, inputs=followup_input, outputs=followup_output)
+    back_btn.click(fn=go_home, outputs=[form_section, pet_btn, domestic_btn, wild_btn, animal, other_input, back_btn, selected_label], show_progress=False)
 
 if __name__ == "__main__":
     demo.launch()