Update app.py

app.py

@@ -1,3 +1,106 @@
-import streamlit as st
+import gradio as gr
+import requests
 
-st.title("Hello from Animal AI Assistant")
+API_KEY = "2d19ae2e41864ddf81690cc806ffe941"  # Replace with your AIMLAPI API key
+
+def call_aimlapi(messages):
+    url = "https://api.aimlapi.com/v1/chat/completions"
+    headers = {
+        "Authorization": f"Bearer {API_KEY}",
+        "Content-Type": "application/json"
+    }
+    data = {
+        "model": "gpt-4o-mini",
+        "messages": messages
+    }
+    response = requests.post(url, headers=headers, json=data)
+    if response.status_code == 200:
+        return response.json()["choices"][0]["message"]["content"]
+    else:
+        return f"Error: {response.status_code} - {response.text}"
+
+def get_solution(animal_type, species, age, gender, weight, info_type, issue_desc, symptoms):
+    # Build a detailed prompt with symptom info (if health selected)
+    symptom_text = ""
+    if info_type.lower() == "health" and symptoms:
+        symptom_text = f"\nSymptoms reported: {', '.join(symptoms)}"
+
+    user_message = f"""
+Animal type: {animal_type}
+Species: {species}
+Age: {age}
+Gender: {gender}
+Weight: {weight}
+Information requested: {info_type}
+Description: {issue_desc}
+{symptom_text}
+
+Please provide a detailed, clear, and helpful answer based on these details.
+Include:
+- Emergency warnings if any critical symptoms are present.
+- Nutrition tips if relevant.
+- Breed-specific information if applicable.
+- Safe natural remedies and medicines (with a disclaimer to consult a vet).
+- And any other useful advice.
+
+Make the response friendly and easy to understand.
+"""
+
+    messages = [
+        {"role": "system", "content": "You are a knowledgeable and helpful veterinary assistant."},
+        {"role": "user", "content": user_message}
+    ]
+
+    answer = call_aimlapi(messages)
+    return answer
+
+# Common symptoms for health issues (can be expanded)
+COMMON_SYMPTOMS = [
+    "Coughing", "Vomiting", "Diarrhea", "Lethargy", "Loss of appetite",
+    "Limping", "Excessive scratching", "Swelling", "Discharge from eyes or nose",
+    "Difficulty breathing", "Seizures", "Excessive thirst", "Weight loss"
+]
+
+with gr.Blocks() as demo:
+    gr.Markdown("# 🐾 Pet & Animal Healthcare AI Assistant")
+
+    with gr.Row():
+        animal_type = gr.Dropdown(choices=["Pet", "Domestic", "Wild"], label="Animal Type", value="Pet", interactive=True)
+        species = gr.Textbox(label="Species (e.g., dog, cat, cow, lion)", placeholder="Enter species", interactive=True)
+
+    with gr.Row():
+        age = gr.Number(label="Age (years)", value=1, interactive=True, precision=1)
+        gender = gr.Dropdown(choices=["Male", "Female", "Unknown"], label="Gender", value="Unknown", interactive=True)
+        weight = gr.Number(label="Weight (kg)", value=1, interactive=True, precision=1)
+
+    info_type = gr.Dropdown(
+        choices=["Health", "Nutrition", "Breed", "Other"],
+        label="Information Requested",
+        value="Health",
+        interactive=True
+    )
+
+    issue_desc = gr.Textbox(label="Describe your issue or question (1-2 lines)", lines=3, placeholder="Enter description here...")
+
+    symptoms = gr.CheckboxGroup(
+        choices=COMMON_SYMPTOMS,
+        label="Select symptoms (only if Health info requested)",
+        visible=True
+    )
+
+    def toggle_symptoms_visibility(info_type):
+        return gr.update(visible=(info_type == "Health"))
+
+    info_type.change(fn=toggle_symptoms_visibility, inputs=info_type, outputs=symptoms)
+
+    submit_btn = gr.Button("Get Solution")
+    output = gr.Textbox(label="AI Solution", lines=20)
+
+    submit_btn.click(
+        fn=get_solution,
+        inputs=[animal_type, species, age, gender, weight, info_type, issue_desc, symptoms],
+        outputs=output
+    )
+
+if __name__ == "__main__":
+    demo.launch()