Update app.py (#3)

- Update app.py (7632d4b408fb48d8a1cda3bb24ca3f4957a91108)

app.py

@@ -7,7 +7,9 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-!pip install duckduckgo-search --upgrade
+pip install duckduckgo-search --upgrade
+
+from duckduckgo_search import DDGS
 
 from duckduckgo_search import DDGS
 
@@ -30,29 +32,50 @@ class VisitWebpageTool:
     def __call__(self, url: str):
         return f"Pretending to visit: {url}"
 
-visit_webpage = VisitWebpageTool()
+# Final answer tool to format and return the final response
+class FinalAnswerTool:
+    def __call__(self, results):
+        formatted_answer = "Final Answer:\n"
+        for result in results:
+            formatted_answer += f"- {str(result)}\n"
+        return formatted_answer
 
-# Dummy model and ToolCallingAgent setup
+# Dummy model
 class DummyModel:
     def call(self, input_text):
         return f"Model processing: {input_text}"
 
+# Modified ToolCallingAgent to use FinalAnswerTool
 class ToolCallingAgent:
-    def __init__(self, tools, model, max_steps=10):
+    def __init__(self, tools, model, final_answer_tool, max_steps=10):
         self.tools = tools
         self.model = model
+        self.final_answer_tool = final_answer_tool
         self.max_steps = max_steps
 
     def run(self, query):
         print(f"Running agent with query: {query}")
+        tool_outputs = []
         for tool in self.tools:
-            print("Tool output:", tool(query))
+            output = tool(query)
+            print("Tool output:", output)
+            tool_outputs.append(output)
+        # Use the final answer tool to format the collected outputs
+        final_result = self.final_answer_tool(tool_outputs)
+        print(final_result)
+        return final_result
+
+# Initialize tools and model
+model = DummyModel()
+search_tool = CustomDuckDuckGoSearchTool()
+visit_webpage = VisitWebpageTool()
+final_answer = FinalAnswerTool()
 
 # Initialize the agent
-model = DummyModel()
 web_agent = ToolCallingAgent(
-    tools=[CustomDuckDuckGoSearchTool(), visit_webpage],
+    tools=[search_tool, visit_webpage],
     model=model,
+    final_answer_tool=final_answer,
     max_steps=10
 )