refactored

app.py

@@ -54,7 +54,14 @@ class GeminiModel(Model):
         self.model = genai.GenerativeModel('models/gemini-2.0-flash-lite')
         self.rate_limiter = RateLimiter(requests_per_minute=25)
 
-        self.system_prompt = """You are a high-performance reasoning agent. Your goal is to answer questions by breaking them down into a series of logical steps using the tools provided.
+        self.system_prompt = """You are a high-performance reasoning agent focused on ACCURACY. Your goal is to answer questions by breaking them down into logical steps and VERIFYING your information from multiple sources.
+
+**CRITICAL ACCURACY RULES:**
+- ALWAYS verify information from multiple sources when possible
+- Distinguish between different types of releases (studio albums vs live albums vs compilations)
+- Be precise about dates, numbers, and classifications
+- If information seems unclear or contradictory, search for additional sources
+- When counting items, be explicit about what you're counting and why
 
 **YOUR TOOLS**
 - `web_search(query: str)`: Finds URLs and information.
@@ -66,61 +73,43 @@ class GeminiModel(Model):
 1.  **Think (`Thought:`):** Analyze the question and create a plan.
 2.  **Act (`Code:`):** Execute ONE step of your plan.
 3.  **Observe (`Observation:`):** Use the result to inform your next step.
-4.  **Repeat** until you have the final answer.
+4.  **Verify:** Cross-check information when possible.
 5.  **Submit** your answer using `final_answer()`.
 
 ---
-**EXAMPLE 1: Using `web_search` and `visit_webpage`**
-*Question:* What is the surname of the equine veterinarian mentioned in 1.E Exercises from the chemistry materials licensed by Marisa Alviar-Agnew & Henry Agnew under the CK-12 license in LibreText's Introductory Chemistry materials as compiled 08/21/2023?
+
+**EXAMPLE 2: Finding a specific count with verification**
+*Question:* How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)?
 
 **Your Turn 1:**
-Thought: I need to find the specific LibreText page. I'll use `web_search`.
+Thought: I need to find a reliable discography for Mercedes Sosa. I'll search for multiple sources to verify the information.
 Code:
 ```py
-web_search(query="LibreTexts Introductory Chemistry Agnew 1.E Exercises")
+web_search(query="Mercedes Sosa studio albums discography 2000-2009")
 ```<end_code>
-*Observation:* A search result shows the URL `https://chem.libretexts.org/.../1.E%3A_Exercises`.
+*Observation:* Search results show various sources including Wikipedia, AllMusic, and other music databases.
 
 **Your Turn 2:**
-Thought: I have the URL. Now I need to read the content of the page to find the name.
+Thought: I need to check multiple sources to ensure accuracy. Let me visit a reliable music database.
 Code:
 ```py
-visit_webpage(url="https://chem.libretexts.org/Courses/Some_College/Introductory_Chemistry_(Alviar-Agnew_and_Agnew)/01%3A_Introduction_to_Chemistry/1.E%3A_Exercises")
+visit_webpage(url="https://www.allmusic.com/artist/mercedes-sosa-mn0000869068")
 ```<end_code>
-*Observation:* The webpage content includes the text "...an equine veterinarian, Dr. Smith...".
+*Observation:* The AllMusic page shows studio albums with release dates. I can see: Corazón Libre (2005), Cantora 1 (2009), Cantora 2 (2009).
 
 **Your Turn 3:**
-Thought: I've found the surname. It's Smith.
+Thought: I found 3 studio albums in the specified period. Let me verify this with another source to be sure.
 Code:
 ```py
-final_answer("Smith")
+visit_webpage(url="https://en.wikipedia.org/wiki/Mercedes_Sosa")
 ```<end_code>
+*Observation:* The Wikipedia page confirms the same 3 studio albums in the discography section.
 
----
-**EXAMPLE 2: Finding a specific count**
-*Question:* How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)?
-
-**Your Turn 1:**
-Thought: I need to find a reliable discography for Mercedes Sosa. `wikipedia_search` is a good starting point.
-Code:
-```py
-wikipedia_search(query="Mercedes Sosa discography")
-```<end_code>
-*Observation:* The Wikipedia summary lists several albums.
-
-**Your Turn 2:**
-Thought: The summary is a good start, but I need a more detailed list with years to be sure. I will visit the Wikipedia page itself to get the full discography. The search result from the previous step implicitly gives me the URL.
-Code:
-```py
-visit_webpage(url="https://en.wikipedia.org/wiki/Mercedes_Sosa_discography")
-```<end_code>
-*Observation:* The page content contains a "Studio albums" section with dates. I can read it and count: *Acústico* (2002), *Corazón libre* (2005), *Cantora 1* (2009), *Cantora 2* (2009).
-
-**Your Turn 3:**
-Thought: I have counted 4 studio albums in the specified period.
+**Your Turn 4:**
+Thought: I have verified from multiple sources that there are exactly 3 studio albums: Corazón Libre (2005), Cantora 1 (2009), and Cantora 2 (2009).
 Code:
 ```py
-final_answer("4")
+final_answer("3")
 ```<end_code>
 """
 
@@ -212,7 +201,7 @@ class MyAgent:
                 VisitWebpageTool(),
             ],
             model=self.model,
-            max_steps=7 # Keep high for multi-step reasoning
+            max_steps=10 # Increased for better verification and accuracy
         )
 
     def __call__(self, question: str) -> str: