Update app.py

app.py

@@ -12,8 +12,9 @@ from langchain_huggingface import HuggingFaceEndpoint
 from langchain_community.chat_models import ChatHuggingFace
 from langchain.llms.base import LLM
 from huggingface_hub import InferenceClient
-from smolagents import Agent
 from smolagents.tools import WebSearchTool
+from smolagents import Agent, CodeAgent, DuckDuckGoSearchTool, InferenceClientModel
+
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -37,16 +38,23 @@ class BasicAgent:
         print("BasicAgent initialized.")
 
         # Initialize the Agent with a text generation model (Falcon 7B instruct)
-        self.agent = Agent.from_model(
-            model="tiiuae/falcon-7b-instruct",
-            model_kwargs={"temperature": 0, "max_new_tokens": 512},
-            tools=[WebSearchTool()]
+        model = InferenceClientModel(
+            model_name="tiiuae/falcon-7b-instruct",
+            temperature=0,
+            max_new_tokens=512
+        )
+        self.agent = CodeAgent(
+            tools=[DuckDuckGoSearchTool()],
+            model=model
         )
 
+
     def __call__(self, question: str) -> str:
-        # Run the agent on the input question and return final answer
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        # Run the agent to get the answer
         answer = self.agent.run(question)
-        print(answer)
+        print(f"Agent returning answer: {answer.strip()}")
+        # Return the answer stripped (no extra spaces/newlines)
         return answer.strip()
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):