Update app.py

app.py

@@ -19,61 +19,37 @@ search_tool = DuckDuckGoSearchTool()
 ##Tool 2
 import wikipedia
 from smolagents import Tool
-from smolagents.models import InferenceClientModel
 
-class WikipediaQATool(Tool):
-    name = "wikipedia_qa"
+class WikipediaReaderTool(Tool):
+    name = "wikipedia_reader"
     description = (
-        "Searches Wikipedia for a topic, reads its content, and answers the input question "
-        "based on the content of the Wikipedia page."
+        "Use this tool to retrieve the full text of a Wikipedia article given a topic. "
+        "Useful when a question involves factual, historical, or biographical knowledge "
+        "that is likely found in Wikipedia. Input must be a single word or phrase representing the topic."
     )
 
     inputs = {
-        "question": {
-            "type": "string",
-            "description": "The question that should be answered using Wikipedia."
-        },
         "topic": {
             "type": "string",
-            "description": "The topic to search for on Wikipedia."
+            "description": "The Wikipedia article title to look up"
         }
     }
 
     output_type = "string"
 
-    def __init__(self, model=None):
-        super().__init__()
-        self.model = model or InferenceClientModel(
-            model="mistralai/Magistral-Small-2506", provider="featherless-ai"
-        )
-
-    def forward(self, question: str, topic: str) -> str:
+    def forward(self, topic: str) -> str:
         try:
             page = wikipedia.page(topic)
-            content = page.content[:2000]  # Limit for context
-
-            # Build QA prompt
-            prompt = (
-                f"You are a Wikipedia expert. Based only on the following content from the Wikipedia page on '{topic}', "
-                f"answer the question briefly and factually.\n\n"
-                f"=== Wikipedia Content ===\n{content}\n\n"
-                f"=== Question ===\n{question}\n\n"
-                f"Answer in a single line. Avoid any extra explanation.\n"
-                f"FINAL ANSWER:"
-            )
-
-            response = self.model(prompt)
-            return response.strip()
-
-        except wikipedia.DisambiguationError as e:
-            return f"Disambiguation error: multiple results found: {', '.join(e.options[:5])}"
-        except wikipedia.PageError:
-            return "Wikipedia page not found."
+            return page.content[:3000]  # return first 3000 characters (within LLM token limit)
+        except wikipedia.exceptions.DisambiguationError as e:
+            return f"Disambiguation error: Be more specific. Options: {', '.join(e.options[:5])}"
+        except wikipedia.exceptions.PageError:
+            return f"Error: No Wikipedia page found for '{topic}'"
         except Exception as e:
-            return f"Error while retrieving Wikipedia content: {str(e)}"
+            return f"Unexpected error: {str(e)}"
 
 
-#wiki_tool = WikipediaQATool()
+wiki_tool = WikipediaReaderTool()
 #excel_tool = ExcelAnalysisTool()
 #yt_tool = YouTubeDialogueTool()
 
@@ -83,7 +59,7 @@ async def run_and_submit_all(profile: gr.OAuthProfile | None):
     try:
         
         agent = ToolCallingAgent(
-            tools=[search_tool],
+            tools=[search_tool, wiki_tool],
                 model=OpenAIServerModel(model_id="gpt-4o", 
                                         api_key=os.environ["OPENAI_API_KEY"], 
                                        temperature=0.0),
@@ -141,7 +117,10 @@ async def run_and_submit_all(profile: gr.OAuthProfile | None):
 
                     If the answer is "st. petersberg" answer as "saint petersburg" (without abbreviations)
                     If the answer is "three" answer as "3".
-                    If the answer is "yamsaki, uehara" answer as "YAMASAKI, UEHARA" (capital letters)
+                    If the answer is "yamsaki, uehara" answer as "YAMASAKI, UEHARA" (capital letters).
+
+                    If the user asks a question like "who played Ray in the Polish-language version of Everybody Loves Raymond", use the `wikipedia_reader` tool with topic='Wszyscy kochają Romana, Magda M'.
+                    If you are unsure of the answer, or believe the question requires external information, call the relevant tool first.
                     """
             )
             full_prompt = system_prompt + f"Question: {question_text.strip()}"