Initial commit with full functionality extend app req tools

app.py

@@ -5,47 +5,11 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-from Gradio_UI import GradioUI
-
-@tool
-def extract_text_tool(url: str) -> str:
-    """A tool that extracts clean text content from a webpage
-    Args:
-        url: The URL of the webpage to analyze
-    """
-    try:
-        response = requests.get(url, timeout=10)
-        response.raise_for_status()
-        return response.text
-    except Exception as e:
-        return f"Error extracting text: {str(e)}"
-
-extract_text_tool.name = "extract_text"
-extract_text_tool.description = "Extracts text content from a URL"
-
-@tool
-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:
-        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)}"
 
-get_current_time_in_timezone.name = "get_time"
-get_current_time_in_timezone.description = "Gets current time in a specific timezone"
-
-# Initialize tools
+# Initialize FinalAnswerTool
 final_answer = FinalAnswerTool()
-search_tool = DuckDuckGoSearchTool()
-search_tool.name = "web_search"
-search_tool.description = "Searches the web for information"
 
-# Setup the model
+# Initialize model
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -53,23 +17,17 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-# Import image generation tool
+# Load tools
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 # Load prompt templates
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Create the agent with all tools
+# Create and configure agent
 agent = CodeAgent(
     model=model,
-    tools=[
-        final_answer,
-        search_tool,
-        tool(extract_text_tool),
-        tool(get_current_time_in_timezone),
-        image_generation_tool
-    ],
+    tools=[final_answer],  # Start with just the final_answer tool
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -79,8 +37,11 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Launch the Gradio interface
+# Import GradioUI after agent creation to avoid circular imports
+from Gradio_UI import GradioUI
+
 if __name__ == "__main__":
+    # Launch Gradio interface
     GradioUI(agent).launch(
         server_name="0.0.0.0",
         server_port=7860

tools/final_answer.py

@@ -1,5 +1,5 @@
 # tools/final_answer.py
-from typing import Any
+from typing import Callable, Any
 from smolagents import tool
 
 @tool
@@ -9,30 +9,26 @@ class FinalAnswerTool:
     This tool is used to format and return the final response from the agent.
     It ensures that the response is properly formatted and ready for presentation
     to the user.
-    
-    Attributes:
-        name (str): The name identifier for the tool
-        description (str): A brief description of the tool's purpose
     """
     
-    name = "final_answer"
-    description = "A tool to provide the final answer"
+    name: str = "final_answer"
+    description: str = "Tool for providing the final response to the user"
 
-    def __call__(self, answer: str) -> str:
-        """Process and return the final answer.
+    def __call__(self, response: str) -> str:
+        """Provide the final answer.
         
         Args:
-            answer: The final answer text to be returned to the user
+            response: The response to be returned as the final answer
             
         Returns:
-            str: The formatted final answer
+            The formatted final answer as a string
         """
-        return answer
+        return response
 
     def __str__(self) -> str:
-        """Return string representation of the tool.
+        """Get string representation.
         
         Returns:
-            str: The tool's name
+            The tool's name
         """
         return self.name