Update app.py

app.py

@@ -4,21 +4,7 @@ import requests
 import inspect
 import pandas as pd
 
-import re
-from transformers import (
-    pipeline,
-    AutoModelForCausalLM,
-    AutoTokenizer,
-    BlenderbotForConditionalGeneration,
-    BlenderbotTokenizer
-)
-import torch
-print(f"PyTorch version: {torch.__version__}")
-print(f"CUDA available: {torch.cuda.is_available()}")
-print(f"GPU memory: {torch.cuda.get_device_properties(0).total_memory/1e9}GB" if torch.cuda.is_available() else "No GPU")
-
-from typing import Optional
-from huggingface_hub import InferenceClient
+from smolagents import DuckDuckGoSearchTool
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -41,7 +27,8 @@ from tools import (
 
 class BasicAgent:
     def __init__(self):
-        self.client = InferenceClient(model="google/flan-t5-xxl")
+        self.search_tool = DuckDuckGoSearchTool()
+
         
     def __call__(self, question, question_attach=""):
         try:
@@ -54,22 +41,7 @@ class BasicAgent:
                 question = question[::-1]
                 print(f"Reversed question: {question}")
                 
-            prompt = f"""<|system|>
-            Answer the following question concisely and factually.
-            If you don't know, say "I don't know" rather than guessing.
-            </s>
-            <|user|>
-            {question}
-            </s>
-            <|assistant|>"""
-            
-            response = self.client.text_generation(
-                prompt,
-                model="google/flan-t5-xxl",
-                max_new_tokens=100,
-            )
-            
-            return response.strip()
+            self.search_tool(question)
             
         except Exception as e:
             print(f"Error: {str(e)}")