Update app.py

app.py

@@ -13,7 +13,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, BitsAndBytesConfig
 from smolagents import CodeAgent, VisitWebpageTool, PromptTemplates, PlanningPromptTemplate, ManagedAgentPromptTemplate, FinalAnswerPromptTemplate
 from smolagents.models import ChatMessage
-from custom_tools import WebpageStructureAnalyzerTool, CachedWebSearchTool, CachedWikiTool, PreloadedPythonTool
+from custom_tools import WebpageStructureAnalyzerTool, SummarizeWebpageContentTool, ExtractTableFromWebpageTool, CachedWebSearchTool, CachedWikiTool, PreloadedPythonTool
 
 subprocess.run(["playwright", "install"], check=True)
 
@@ -173,6 +173,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         llm_model = BasicModel(model_id=MODEL_ID, hf_token=hf_token)
         # (2) Create the tools
         web_structure_analyzer_tool = WebpageStructureAnalyzerTool()
+        summarizer_tool = SummarizeWebpageContentTool()
+        table_extractor_tool = ExtractTableFromWebpageTool()
         wiki_tool = CachedWikiTool()
         search_tool = CachedWebSearchTool()
         python_tool = PreloadedPythonTool()
@@ -194,7 +196,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
                                                                                    post_messages=data['final_answer']['post_messages']))
         # (5) Create the CodeAgent, passing the LLM wrapper and tools
         agent = CodeAgent(model=llm_model, 
-                          tools=[web_structure_analyzer_tool, wiki_tool, search_tool, python_tool, html_parse_tool], 
+                          tools=[web_structure_analyzer_tool, summarizer_tool, table_extractor_tool, wiki_tool, search_tool, python_tool, html_parse_tool], 
                           prompt_templates=my_prompt_templates,
                           max_steps=8,
                           add_base_tools=True,