Update app.py

app.py

@@ -10,14 +10,151 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+import os
+from typing import List, Dict, Any, Optional
+# from smolagents import CodeAgent, Tool
+# from smolagents.models import ApiModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool, InferenceClientModel, Tool, tool, VisitWebpageTool
+
+import json
+from duckduckgo_search import DDGS
+import wikipedia
+import re
+from datetime import datetime
+
+class WebSearchTool(Tool):
+    name = "web_search"
+    description = "Search the web for information on a given query using DuckDuckGo"
+    
+    def search(self, query: str, max_results: int = 5) -> List[Dict[str, str]]:
+        """
+        Search the web for information using DuckDuckGo.
+        
+        Args:
+            query: The search query
+            max_results: Maximum number of results to return
+            
+        Returns:
+            List of search results with title, body, and href
+        """
+        try:
+            with DDGS() as ddgs:
+                results = list(ddgs.text(query, max_results=max_results))
+                return results
+        except Exception as e:
+            return [{"title": "Error", "body": f"Search error: {str(e)}", "href": ""}]
+
+class WikipediaTool(Tool):
+    name = "wikipedia"
+    description = "Search Wikipedia for information on a given topic"
+    
+    def search(self, query: str, sentences: int = 3) -> str:
+        """
+        Search Wikipedia for information on a topic.
+        
+        Args:
+            query: The search query
+            sentences: Number of sentences to return in the summary
+            
+        Returns:
+            Wikipedia summary as a string
+        """
+        try:
+            result = wikipedia.summary(query, sentences=sentences)
+            return result
+        except wikipedia.exceptions.DisambiguationError as e:
+            return f"Multiple results found. Try one of these: {', '.join(e.options[:5])}"
+        except wikipedia.exceptions.PageError:
+            return f"No Wikipedia page found for '{query}'"
+        except Exception as e:
+            return f"Wikipedia search error: {str(e)}"
+
+class DateTimeTool(Tool):
+    name = "datetime_tool"
+    description = "Get current date and time information"
+    
+    def get_current_datetime(self) -> str:
+        """Get the current date and time."""
+        now = datetime.now()
+        return now.strftime("%Y-%m-%d %H:%M:%S")
+
 class BasicAgent:
     def __init__(self):
+        """
+        Initialize the GAIA dataset agent with SmoLagents.
+        
+        Args:
+            api_key: API key for the LLM provider
+            model_name: Name of the LLM model to use
+        """
         print("BasicAgent initialized.")
+        
+        # Initialize tools
+        self.web_search_tool = DuckDuckGoSearchTool()
+        # self.wiki_tool = WikipediaTool()
+        self.datetime_tool = DateTimeTool()
+        
+        # Set up model
+        self.model = InferenceClientModel()
+        
+        # Initialize the agent
+        self.agent = CodeAgent(
+            model=self.model,
+            tools=[
+                self.web_search_tool, 
+                # self.wiki_tool, 
+                self.datetime_tool,
+            ],
+            verbose=False  # Set to True for detailed logs
+        )
+    
     def __call__(self, question: str) -> str:
+        """
+        Process a question and return an answer.
+        
+        Args:
+            question: The question to answer
+            
+        Returns:
+            The answer to the question
+        """
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        
+        # Create a prompt for the agent
+        system_prompt = """
+        You are a knowledgeable assistant answering questions from the GAIA dataset. 
+        Your task is to provide accurate, well-researched answers based on the best available information.
+        Follow these guidelines:
+        1. Use search tools to find relevant information
+        2. Check Wikipedia for background knowledge
+        3. Verify facts when possible
+        4. Be transparent about uncertainty
+        5. Provide clear, concise answers with supporting evidence
+        6. If the question is unclear, ask for clarification
+        7. If information is unavailable, say so rather than making things up
+        
+        Always cite your sources in your answer.
+        """
+        
+        full_prompt = f"{system_prompt}\n\nQuestion: {question}\n\nAnswer:"
+        
+        try:
+            # Run the agent
+            response = self.agent.run(full_prompt)
+            
+            # Clean up the response
+            # Remove any system-prompt-like text at the beginning
+            cleaned_response = re.sub(r'^.*?Answer:', '', response, flags=re.DOTALL).strip()
+            
+            if not cleaned_response:
+                cleaned_response = response  # Fallback to original if cleaning removes everything
+            
+            print(f"Agent returning answer (first 50 chars): {cleaned_response[:50]}...")
+            return cleaned_response
+        except Exception as e:
+            error_msg = f"Error processing question: {str(e)}"
+            print(error_msg)
+            return error_msg
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """