Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import requests
 import os
 
-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."
 
@@ -15,60 +15,25 @@ animal_options = {
     "Wild": ["Tiger", "Lion", "Elephant", "Deer", "Bear", "Leopard", "Wolf", "Giraffe", "Other"]
 }
 
-def show_only_selected(category):
-    return (
-        gr.update(visible=True, value=f"### Selected Category: {category}"),
-        gr.update(visible=False),
-        gr.update(visible=False),
-        gr.update(visible=False),
-        gr.update(choices=animal_options.get(category, []), value=None, visible=True),
-        gr.update(value="", visible=False),  # other input
-        gr.update(visible=True),  # form row
-        gr.update(visible=True),  # topic
-        gr.update(visible=True),  # user detail
-        gr.update(visible=True),  # submit
-        gr.update(visible=False),  # output box hidden initially
-        gr.update(visible=True),  # back button
-        gr.update(visible=False),  # followup input
-        gr.update(visible=False),  # followup btn
-        gr.update(visible=False),  # followup output
-        gr.update(visible=False)   # followup header
-    )
-
-def update_animals(selected_animal):
-    return gr.update(visible=(selected_animal == "Other"), value="")
-
-def back_to_home():
-    return (
-        gr.update(visible=False),  # category markdown
-        gr.update(visible=True),   # pet
-        gr.update(visible=True),   # domestic
-        gr.update(visible=True),   # wild
-        gr.update(visible=False),  # animal dropdown
-        gr.update(visible=False, value=""),  # other animal
-        gr.update(visible=False),  # form row
-        gr.update(visible=False),  # topic
-        gr.update(visible=False),  # user detail
-        gr.update(visible=False),  # submit
-        gr.update(visible=False),  # output
-        gr.update(visible=False),  # back
-        gr.update(visible=False),  # followup input
-        gr.update(visible=False),  # followup btn
-        gr.update(visible=False),  # followup output
-        gr.update(visible=False)   # followup header
-    )
-
-def ask_ai(category_text, selected_animal, other_animal, age, gender, height, weight, topic, user_detail):
+def update_animals(category):
+    animals = animal_options.get(category, [])
+    return gr.update(choices=animals, visible=True, value=None), gr.update(visible=False, value="")
+
+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
-    category = category_text.replace("### Selected Category: ", "").strip()
-    animal = other_animal if selected_animal == "Other" else selected_animal
+    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.strip()[:60]}"""
+Detail: {user_detail}
+Give a short and relevant response."""
 
     messages = [
         {"role": "system", "content": SYSTEM_PROMPT},
@@ -94,15 +59,9 @@ Detail: {user_detail.strip()[:60]}"""
         answer = result['choices'][0]['message']['content'].strip()
         chat_history.clear()
         chat_history.extend(messages + [{"role": "assistant", "content": answer}])
-        return (
-            answer,
-            gr.update(visible=True),
-            gr.update(visible=True),
-            gr.update(visible=True),
-            gr.update(visible=True)
-        )
+        return answer
     except Exception as e:
-        return (f"❌ Error: {str(e)}", gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False))
+        return f"❌ Error: {str(e)}"
 
 def handle_followup(user_input):
     global chat_history, followup_count
@@ -134,73 +93,37 @@ def handle_followup(user_input):
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
+# Launch app
 with gr.Blocks() as demo:
     gr.Markdown("## 🐾 AI Vet Assistant")
 
     with gr.Row():
-        pet_btn = gr.Button("Pet")
-        domestic_btn = gr.Button("Domestic")
-        wild_btn = gr.Button("Wild")
+        category = gr.Dropdown(["Pet", "Domestic", "Wild"], label="Animal Category")
+        animal = gr.Dropdown(label="Select Animal", visible=False)
+        other_input = gr.Textbox(label="Enter Animal", visible=False)
 
-    selected_category = gr.Markdown(visible=False)
-    animal = gr.Dropdown(label="Select Animal", choices=[], visible=False)
-    other_input = gr.Textbox(label="Enter Animal", visible=False)
+    category.change(update_animals, category, [animal, other_input])
+    animal.change(handle_other_animal, animal, other_input)
 
-    with gr.Row(visible=False) as form_row:
-        age = gr.Textbox(label="Age")
+    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")
-        weight = gr.Textbox(label="Weight")
-
-    topic = gr.Dropdown(["Health", "Nutrition", "Breed Info", "Natural Remedies"], label="Topic", visible=False)
-    user_detail = gr.Textbox(label="Problem Detail", max_lines=2, visible=False)
-    submit_btn = gr.Button("Get Advice", visible=False)
-    output = gr.Textbox(label="AI Advice", lines=6, visible=False)
-
-    followup_header = gr.Markdown("### 🔄 Follow-up Questions", visible=False)
-    followup_input = gr.Textbox(label="Your Follow-up", visible=False)
-    followup_btn = gr.Button("Ask Follow-up", visible=False)
-    followup_output = gr.Textbox(label="Follow-up Response", lines=4, visible=False)
-
-    back_btn = gr.Button("Back to Home", visible=False)
-
-    pet_btn.click(show_only_selected, inputs=["Pet"], outputs=[
-        selected_category, pet_btn, domestic_btn, wild_btn,
-        animal, other_input, form_row, topic, user_detail,
-        submit_btn, output, back_btn, followup_input,
-        followup_btn, followup_output, followup_header
-    ])
-
-    domestic_btn.click(show_only_selected, inputs=["Domestic"], outputs=[
-        selected_category, pet_btn, domestic_btn, wild_btn,
-        animal, other_input, form_row, topic, user_detail,
-        submit_btn, output, back_btn, followup_input,
-        followup_btn, followup_output, followup_header
-    ])
-
-    wild_btn.click(show_only_selected, inputs=["Wild"], outputs=[
-        selected_category, pet_btn, domestic_btn, wild_btn,
-        animal, other_input, form_row, topic, user_detail,
-        submit_btn, output, back_btn, followup_input,
-        followup_btn, followup_output, followup_header
-    ])
-
-    animal.change(update_animals, inputs=animal, outputs=other_input)
-
-    submit_btn.click(
-        ask_ai,
-        inputs=[selected_category, animal, other_input, age, gender, height, weight, topic, user_detail],
-        outputs=[output, followup_input, followup_btn, followup_output, followup_header]
-    )
+        height = gr.Textbox(label="Height (e.g. 30 cm)")
+        weight = gr.Textbox(label="Weight (e.g. 10 kg)")
 
-    followup_btn.click(handle_followup, inputs=followup_input, outputs=followup_output)
+    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)
 
-    back_btn.click(back_to_home, outputs=[
-        selected_category, pet_btn, domestic_btn, wild_btn,
-        animal, other_input, form_row, topic, user_detail,
-        submit_btn, output, back_btn, followup_input,
-        followup_btn, followup_output, followup_header
-    ])
+    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)
+
+    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)
 
 if __name__ == "__main__":
     demo.launch()