Update app.py

app.py

@@ -22,16 +22,54 @@ from tools import (
     query_image,
     query_video
 )
-"""from langchain_community.tools import DuckDuckGoSearchRun
+from langchain_community.tools import DuckDuckGoSearchRun
 
 search_tool = DuckDuckGoSearchRun()
-results = search_tool.invoke("Who's the current President of France?")
-print(results)"""
+
+"""
+from huggingface_hub import InferenceClient
+
+class StrictAnswerAgent:
+    def __init__(self):
+        
+        self.llm = InferenceClient("meta-llama/Llama-3.2-3B-Instruct")
+    
+    def _web_search(self, query, num_results=3):
+        'Get search results from DuckDuckGo'
+        try:
+            results = ddg(query, max_results=num_results)
+            return [self._clean_text(r['body']) for r in results]
+        except:
+            return []
+    
+    def answer(self, question):
+        # Step 1: Get web search context
+        search_results = self._web_search(question)
+        context = "\n".join(search_results) if search_results else "No context found"
+        
+        # Step 2: Generate answer with strict 2-word limit
+        prompt = f'Context: {context}
+        Question: {question}
+        Answer concisely in exactly 2 words:'
+        
+        response = self.llm.text_generation(
+            prompt,
+            max_new_tokens=4,  # Strict limit for 2 words
+            temperature=0.3,
+            truncate=2000
+        )
+        
+        # Post-process to ensure 2-word maximum
+        words = response.strip().replace(".", "").split()[:2]
+        return ' '.join(words) if words else "Unknown"
+
+
+"""
     
 
 class BasicAgent:
     def __init__(self):
-        self.search_tool = DuckDuckGoSearchTool()
+        search_tool = DuckDuckGoSearchRun()
 
         
     def __call__(self, question, question_attach=""):
@@ -45,7 +83,7 @@ class BasicAgent:
                 question = question[::-1]
                 print(f"Reversed question: {question}")
                 
-            return self.search_tool(question)
+            return self.search_tool.invoke(question)
             
         except Exception as e:
             print(f"Error: {str(e)}")