Update app.py

app.py

@@ -1,6 +1,7 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, Tool
 from smolagents.default_tools import FinalAnswerTool, VisitWebpageTool
 from Gradio_UI import GradioUI
+from langchain.agents import load_tools
 
 import datetime
 import yaml
@@ -25,6 +26,7 @@ final_answer = FinalAnswerTool()
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 search_tool = DuckDuckGoSearchTool()
 visit_webpage_tool = VisitWebpageTool()
+wolfram_tool = Tool.from_langchain(load_tools(["wolfram-alpha"])[0])
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
@@ -33,7 +35,8 @@ agent = CodeAgent(
     model=model,
     tools=[
             final_answer,
-            knowledge_and_intelligence_tool,
+            wolfram_tool,
+            # knowledge_and_intelligence_tool,
             search_tool,
             image_generation_tool,
             convert_currency,