Update app.py

app.py

@@ -122,6 +122,21 @@ class DocumentQuestionAnswering(BaseTool):
         response = self.qa_chain.run(query)
         return str(response)
 
+class DuckDuckGoSearchRun(BaseTool):
+    name = "DuckDuckGo"
+    description = "Useful for searching the internet for general information"
+
+    def _run(self, query: str) -> str:
+        url = "https://api.duckduckgo.com/"
+        params = {
+            "q": query,
+            "format": "json"
+        }
+        response = requests.get(url, params=params)
+        data = response.json()
+        answer = data["Abstract"]
+        return answer
+        
 # Function to handle different input types and choose the right tool
 def handle_input(user_prompt, image=None, audio=None, websearch=False, document=None):
     if audio: