Update app.py

app.py

@@ -2,82 +2,23 @@ import streamlit as st
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
 
 
-examples = """
-Here are a few examples:
-
-Task: Today, I had an appointment with Bella, a 5-year-old Golden Retriever owned by John Smith. She came in for her annual check-up. Bella weighs 25 kilograms and is in great condition overall.
-
-During the exam, I checked her vital signs: her temperature was 38.5°C, her heart rate was 80 bpm, and her respiratory rate was 20 breaths per minute—all perfectly normal. Her coat was shiny and free of any abnormalities, her eyes were clear with no discharge, and her ears showed no signs of infection. I did notice some mild tartar on her teeth, which I mentioned to John, and her skin was healthy with no rashes or lesions.
-
-As part of her routine care, I administered her rabies vaccine, which was due, and confirmed that her distemper vaccination is up to date. We also performed a fecal test and a heartworm test, both of which came back negative.
-
-For prevention, I prescribed her a monthly heartworm medication and provided her with flea and tick prevention. I told John that Bella is healthy and has no current health concerns. I recommended he continue her balanced diet, keep up her regular exercise, and return in a year for her next check-up.
-
-It was a pleasure to see Bella, and I look forward to her next visit. – Dr. Emily Carter
-
-
-Final answer: 
-
-Date of Visit	January 9, 2025
-Pet's Name	Bella
-Species	Dog
-Breed	Golden Retriever
-Age	5 years
-Weight	25 kg
-Owner's Name	John Smith
-Reason for Visit	Annual Check-up
-Vital Signs	- Temperature: 38.5°C
-- Heart Rate: 80 bpm
-- Respiratory Rate: 20/min
-Physical Exam	- Coat: Shiny, no abnormalities
-- Eyes: Clear, no discharge
-- Ears: No signs of infection
-- Teeth: Mild tartar
-- Skin: No rashes or lesions
-Vaccinations	- Rabies Vaccine: Administered
-- Distemper: Up to date
-Tests Performed	- Fecal Test: Negative
-- Heartworm Test: Negative
-Medications	- Monthly Heartworm Prevention prescribed
-- Flea and Tick prevention given
-Diagnosis	Healthy; No concerns detected
-Recommendations	- Continue balanced diet
-- Regular exercise
-- Return in 1 year for next annual check-up
-Veterinarian	Dr. Emily Carter
-
-"""
-
-sys_admin_prompt = f"""
-You are a systems admin at a veterinary clinic. It is your job to organize my reports. 
-
-Here is my report:
-
-{{task}}
-
-Here are a few examples of previous tasks
-
-{{additional_prompting}}
-"""
-
-
-admin_agent = CodeAgent(
-    tools=[],
+web_agent = ToolCallingAgent(
+    tools=[DuckDuckGoSearchTool(max_results=3), VisitWebpageTool()],
     model=HfApiModel(),
 )
 
-managed_admin_agent = ManagedAgent(
-    agent=admin_agent,
-    name="system admin",
-    managed_agent_prompt=sys_admin_prompt,
-    additional_prompting=examples,
-    description="Returns a rough draft of an organized report",
+managed_web_agent = ManagedAgent(
+    agent=web_agent,
+    name="search",
+    description="Runs veterinarian web searches. Give it your query as an argument. Also, this agent should link to the sources you are using. If any specific sites are provided, this agent will only those sources.",
 )
 
 manager_agent = CodeAgent(
     tools=[],
-    model=HfApiModel(),
-    managed_agents=[managed_admin_agent],
+    model=HfApiModel(model_id = "Qwen/Qwen2.5-Coder-32B-Instruct"),
+    managed_agents=[managed_web_agent],
+    additional_authorized_imports=['pyparsing', 'matplotlib', 'datetime', 'statistics', 'bs4', 'request', 'unicodedata', 'queue', 'time', 'collections', 're', 'math', 'stat', 'random', 
+'itertools'],
 )
 
 # Function to log agent actions
@@ -89,7 +30,7 @@ def log_agent_action(prompt, result, agent_name):
     st.code(result, language="text")
 
 # Streamlit app title
-st.title("AI Veterinary Assistant Agent organizing reports")
+st.title("AI Veterinary Assistant Agent researching your query and summarizing it")
 
 # App description
 st.write("Generate reports enriched with real-time insights using the AI Veterinary Report Writing Agent powered by SmolAgents and DuckDuckGo.")
@@ -109,7 +50,7 @@ if st.button("Generate Report"):
                 st.write(result)
 
                 # Log backend activity
-                log_agent_action(prompt, result, "Blog Writing Agent with DuckDuckGo")
+                log_agent_action(prompt, result, "Veterinary Research Agent with DuckDuckGo")
             except Exception as e:
                 st.error(f"An error occurred: {e}")
     else: