Update app.py

app.py

@@ -4,10 +4,7 @@ import os
 
 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
+SYSTEM_PROMPT = "You are a helpful and knowledgeable veterinary assistant AI..."
 
 animal_options = {
     "Pet": ["Dog", "Cat", "Rabbit", "Hamster", "Parrot", "Other"],
@@ -16,15 +13,18 @@ animal_options = {
 }
 
 def show_only_selected(category):
+    # Return the selected category markdown text
+    # plus visibility and choices updates for UI components
     return (
-        gr.update(visible=True, value=f"**Selected Category:** {category}"),
-        gr.update(visible=(category == "Pet")),
+        f"**Selected Category:** {category}",  # Markdown text
+        gr.update(visible=(category == "Pet")),  # pet_btn visible if not selected
         gr.update(visible=(category == "Domestic")),
         gr.update(visible=(category == "Wild")),
-        gr.update(visible=True, choices=animal_options.get(category, []), value=None),
-        gr.update(visible=False),
-        gr.update(visible=True),
-        gr.update(visible=True),
+        gr.update(visible=True, choices=animal_options[category], value=None),  # animal dropdown
+        gr.update(visible=False),  # other_input hidden initially
+        gr.update(visible=True),   # form visible
+        gr.update(visible=True),   # back to home visible
+        gr.update(visible=False),  # category buttons hidden when one selected
     )
 
 def update_animals(selected_animal):
@@ -32,103 +32,28 @@ def update_animals(selected_animal):
 
 def back_to_home():
     return (
-        gr.update(visible=False, value=""),
-        gr.update(visible=True),
-        gr.update(visible=True),
-        gr.update(visible=True),
-        gr.update(visible=False, value=None),
-        gr.update(visible=False, value=""),
-        gr.update(visible=False),
-        gr.update(visible=False),
+        gr.update(value="", visible=False),  # selected_category_markdown
+        gr.update(visible=True),   # pet_btn visible
+        gr.update(visible=True),   # domestic_btn visible
+        gr.update(visible=True),   # wild_btn visible
+        gr.update(visible=False, value=None),  # animal dropdown hidden and cleared
+        gr.update(visible=False, value=""),  # other input hidden and cleared
+        gr.update(visible=False),  # form hidden
+        gr.update(visible=False)   # back home hidden
     )
 
-def ask_ai(category_md, selected_animal, other_animal, age, gender, height, weight, topic, user_detail):
-    global chat_history, followup_count
-    followup_count = 0
-    category = category_md.replace("**Selected Category:** ", "").strip()
-    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)}"
-
-
 with gr.Blocks() as demo:
-
     gr.Markdown("## 🐾 AI Vet Assistant")
 
     with gr.Row():
-        pet_btn = gr.Button("Pet", visible=True)
-        domestic_btn = gr.Button("Domestic", visible=True)
-        wild_btn = gr.Button("Wild", visible=True)
+        pet_btn = gr.Button("Pet")
+        domestic_btn = gr.Button("Domestic")
+        wild_btn = gr.Button("Wild")
 
     selected_category_markdown = 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)")
         gender = gr.Dropdown(["Male", "Female", "Unknown"], label="Gender")
@@ -141,87 +66,38 @@ with gr.Blocks() as demo:
     submit_btn = gr.Button("Get Advice", visible=False)
     output = gr.Textbox(label="AI Advice", lines=8, 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_home_btn = gr.Button("Back to Home", visible=False)
 
-    pet_btn.click(show_only_selected, inputs=[], outputs=[
+    # Here is the key: pass the category name explicitly as input
+    pet_btn.click(show_only_selected, inputs=[gr.State("Pet")], outputs=[
         selected_category_markdown,
-        pet_btn,
-        domestic_btn,
-        wild_btn,
-        animal,
-        other_input,
-        form_row,
+        pet_btn, domestic_btn, wild_btn,
+        animal, other_input, form_row,
         back_home_btn
     ])
 
-    domestic_btn.click(show_only_selected, inputs=[], outputs=[
+    domestic_btn.click(show_only_selected, inputs=[gr.State("Domestic")], outputs=[
         selected_category_markdown,
-        pet_btn,
-        domestic_btn,
-        wild_btn,
-        animal,
-        other_input,
-        form_row,
+        pet_btn, domestic_btn, wild_btn,
+        animal, other_input, form_row,
         back_home_btn
     ])
 
-    wild_btn.click(show_only_selected, inputs=[], outputs=[
+    wild_btn.click(show_only_selected, inputs=[gr.State("Wild")], outputs=[
         selected_category_markdown,
-        pet_btn,
-        domestic_btn,
-        wild_btn,
-        animal,
-        other_input,
-        form_row,
+        pet_btn, domestic_btn, wild_btn,
+        animal, other_input, form_row,
         back_home_btn
     ])
 
     animal.change(update_animals, animal, other_input)
 
-    submit_btn.click(
-        ask_ai,
-        inputs=[selected_category_markdown, animal, other_input, age, gender, height, weight, topic, user_detail],
-        outputs=output
-    )
-
-    followup_btn.click(handle_followup, inputs=followup_input, outputs=followup_output)
-
-    back_home_btn.click(
-        back_to_home,
-        inputs=[],
-        outputs=[
-            selected_category_markdown,
-            pet_btn,
-            domestic_btn,
-            wild_btn,
-            animal,
-            other_input,
-            form_row,
-            back_home_btn
-        ]
-    )
+    back_home_btn.click(back_to_home, inputs=[], outputs=[
+        selected_category_markdown,
+        pet_btn, domestic_btn, wild_btn,
+        animal, other_input, form_row,
+        back_home_btn
+    ])
 
-    def show_form_parts(show: bool):
-        return (
-            gr.update(visible=show),
-            gr.update(visible=show),
-            gr.update(visible=show),
-            gr.update(visible=show),
-            gr.update(visible=show),
-            gr.update(visible=show),
-            gr.update(visible=show),
-            gr.update(visible=show),
-        )
-
-    pet_btn.click(show_form_parts, inputs=[], outputs=[topic, user_detail, submit_btn, output, followup_header, followup_input, followup_btn, followup_output])
-    domestic_btn.click(show_form_parts, inputs=[], outputs=[topic, user_detail, submit_btn, output, followup_header, followup_input, followup_btn, followup_output])
-    wild_btn.click(show_form_parts, inputs=[], outputs=[topic, user_detail, submit_btn, output, followup_header, followup_input, followup_btn, followup_output])
-
-    back_home_btn.click(lambda: show_form_parts(False), inputs=[], outputs=[topic, user_detail, submit_btn, output, followup_header, followup_input, followup_btn, followup_output])
-
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()