Update app.py

app.py

@@ -4,38 +4,75 @@ import requests
 import inspect
 import pytz
 import datetime
+import wikipedia
 
 import pandas as pd
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+
+@tool
+def get_current_time_in_timezone(timezone: str) -> str:
+    """Fetches the current local time in a specified timezone.
+    Args:
+        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+    """
+    try:
+        tz = pytz.timezone(timezone)
+        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+        return f"The current local time in {timezone} is: {local_time}"
+    except Exception as e:
+        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+
+        
+@tool
+def wikipedia_search(query: str, sentences: int = 2) -> str:
+    """Searches Wikipedia for information.
+    Args:
+        query: The search term.
+        sentences: Number of summary sentences to return (default: 2).
+    """
+    try:
+        return wikipedia.summary(query, sentences=sentences)
+    except wikipedia.exceptions.DisambiguationError as e:
+        return f"Disambiguation error: {e.options}"
+    except wikipedia.exceptions.PageError:
+        return "No Wikipedia page found for this query."
+    except Exception as e:
+        return f"Wikipedia search failed: {str(e)}"
+
+
+
 class BasicAgent:
     def __init__(self):
         model = HfApiModel(
         max_tokens=2096,
-        temperature=0.5,
+        temperature=0.3,
         model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
         custom_role_conversions=None,
         )
 
-     def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
-        try:
-            # Create timezone object
-            tz = pytz.timezone(timezone)
-            # Get current time in that timezone
-            local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-            return f"The current local time in {timezone} is: {local_time}"
-        except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+        self.search_tool = DuckDuckGoSearchTool()
+        self.tools = {
+            "current_time": get_current_time_in_timezone,
+            "wikipedia": wikipedia_search,
+            "search": self.search_tool
+        }
+         self.agent = CodeAgent(
+            model=self.model,
+            tools=self.tools,
+            system_prompt="You are a helpful AI assistant with access to various tools. "
+                        "Analyze questions carefully and use the appropriate tools when needed. "
+                        "Provide clear, concise, and accurate answers."
+        )
+
+    
         
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")