Initial commit with full functionality extend app req tools

app.py

@@ -7,7 +7,7 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Helper function for text extraction
+@tool
 def extract_text_tool(url: str) -> str:
     """A tool that extracts clean text content from a webpage
     Args:
@@ -20,7 +20,10 @@ def extract_text_tool(url: str) -> str:
     except Exception as e:
         return f"Error extracting text: {str(e)}"
 
-# Time zone tool from the example
+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:
@@ -33,9 +36,14 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# Initialize components
+get_current_time_in_timezone.name = "get_time"
+get_current_time_in_timezone.description = "Gets current time in a specific timezone"
+
+# Initialize tools
 final_answer = FinalAnswerTool()
 search_tool = DuckDuckGoSearchTool()
+search_tool.name = "web_search"
+search_tool.description = "Searches the web for information"
 
 # Setup the model
 model = HfApiModel(
@@ -57,10 +65,10 @@ agent = CodeAgent(
     model=model,
     tools=[
         final_answer,
-        search_tool,  # For web search
-        tool(extract_text_tool),  # For text extraction
-        tool(get_current_time_in_timezone),  # For timezone checks
-        image_generation_tool  # For image generation
+        search_tool,
+        tool(extract_text_tool),
+        tool(get_current_time_in_timezone),
+        image_generation_tool
     ],
     max_steps=6,
     verbosity_level=1,

tools/final_answer.py

@@ -1,7 +1,12 @@
 # tools/final_answer.py
 from typing import Any
+from smolagents import tool
 
+@tool
 class FinalAnswerTool:
+    name = "final_answer"  # Required by smolagents
+    description = "A tool to provide the final answer"
+
     def __call__(self, answer: str) -> str:
         """Tool to provide a final answer
         Args:
@@ -10,4 +15,4 @@ class FinalAnswerTool:
         return answer
 
     def __str__(self) -> str:
-        return "final_answer"
+        return self.name