refactor: update tool imports and instantiations for consistency; enhance Gradio UI error handling

app.py

@@ -6,11 +6,9 @@ from src.final_answer_tool import FinalAnswerTool
 from src.web_browsing_tool import WebBrowser
 from src.file_processing_tool import FileIdentifier
 from src.image_processing_tool import ImageProcessor
-# src.markdown_table_parser.py contains a function, will need a wrapper Tool class
-# from src.markdown_table_parser import parse_markdown_table 
+from src.markdown_table_parser import MarkdownTableParserTool # Updated
 from src.python_tool import CodeExecutionTool
-# src.speech_to_text.py contains a function, will need a wrapper Tool class
-# from src.speech_to_text import transcribe_audio
+from src.speech_to_text import SpeechToTextTool # Updated
 from src.spreadsheet_tool import SpreadsheetTool
 from src.text_reversal_tool import TextReversalTool
 from src.video_processing_tool import VideoProcessingTool
@@ -27,25 +25,16 @@ model = HfApiModel(
 )
 
 # Instantiate Tools
-# These instantiations assume the classes are now proper smolagents Tools
 final_answer_tool = FinalAnswerTool()
 web_browsing_tool = WebBrowser() 
 file_processing_tool = FileIdentifier() 
 image_processing_tool = ImageProcessor() 
+markdown_parser_tool = MarkdownTableParserTool() # Updated
 python_tool = CodeExecutionTool() 
+speech_to_text_tool = SpeechToTextTool() # Updated
 spreadsheet_tool = SpreadsheetTool() 
 text_reversal_tool = TextReversalTool()
 video_processing_tool = VideoProcessingTool() 
-ddg_search_tool = DuckDuckGoSearchTool()
-
-# For functions that need to be wrapped as tools (e.g., markdown_table_parser, speech_to_text):
-# These would need wrapper classes defined, potentially in their respective files or here.
-# Example (conceptual):
-# from src.markdown_table_parser import MarkdownTableParserTool # Assuming this wrapper exists
-# markdown_parser_tool = MarkdownTableParserTool()
-# from src.speech_to_text import SpeechToTextTool # Assuming this wrapper exists
-# speech_to_text_tool = SpeechToTextTool()
-
 
 # Load Prompts
 try:
@@ -60,29 +49,24 @@ except yaml.YAMLError as e:
 
 
 # Create the Agent
-# Note: Tools that are not yet proper smolagent tools (or their wrappers) are commented out.
-# They need to be converted/wrapped first and then added to this list.
 agent_tools = [
     final_answer_tool,
     web_browsing_tool,
     file_processing_tool,
     image_processing_tool,
-    # markdown_parser_tool, # Add when MarkdownTableParserTool wrapper is ready
+    markdown_parser_tool, # Updated
     python_tool,
-    # speech_to_text_tool, # Add when SpeechToTextTool wrapper is ready
+    speech_to_text_tool, # Updated
     spreadsheet_tool,
     text_reversal_tool,
-    video_processing_tool,
-    ddg_search_tool
+    video_processing_tool
 ]
 
 agent = CodeAgent(
     model=model,
     tools=agent_tools,
-    max_steps=15,  # Increased max_steps for potentially complex questions
-    verbosity_level=1, # 0 for less, 1 for normal, 2 for more verbose
-    # grammar=None, # Specify if you have a grammar for model output
-    # planning_interval=None, # Steps between planning phases
+    max_steps=15,
+    verbosity_level=1, 
     name="ComprehensiveQuestionAgent",
     description="An agent equipped with a suite of tools to answer diverse questions from the common_questions.json set.",
     prompt_templates=prompt_templates
@@ -91,12 +75,18 @@ agent = CodeAgent(
 # Launch the Gradio UI
 if __name__ == '__main__':
     print("Launching Gradio UI...")
-    # Ensure GradioUI can accept the agent instance
-    # If GradioUI is not designed to take an agent directly, this might need adjustment
-    # based on how GradioUI is implemented.
     try:
         ui = GradioUI(agent)
         ui.launch()
     except Exception as e:
-        print(f"Error launching Gradio UI: {e}")
-        print("Please ensure GradioUI is correctly defined and can be initialized with the agent.")
+        print(f"Failed to launch Gradio UI: {e}")
+        print("Attempting to run agent with a sample query directly...")
+        try:
+            # This is a placeholder for a direct test if Gradio fails.
+            # You might want to load a question from common_questions.json here.
+            sample_query = "What is the capital of France and reverse the answer?"
+            print(f"Testing agent with query: {sample_query}")
+            response = agent.run(sample_query)
+            print(f"Agent response: {response}")
+        except Exception as agent_e:
+            print(f"Error running agent directly: {agent_e}")