Update app.py

app.py

@@ -4,27 +4,16 @@ import requests
 import inspect
 import pandas as pd
 import sympy
-import re
-from duckduckgo_search import ddg
+from duckduckgo_search import search
 from langgraph.graph import StateGraph, END
 from typing import TypedDict
 
-# (Keep Constants as is)
-# --- Constants ---
-DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
-
-# --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-from smolagent import Agent, Tool
-from duckduckgo_search import ddg
-import sympy
-import re
 # --- Tools ---
 
 def wikipedia_search_tool(input: str) -> str:
-    results = ddg(input, max_results=3)
+    results = search(input, max_results=3)
     if results:
-        return results[0]["body"]
+        return results[0].get("body", "No information found.")
     return "No information found."
 
 def math_solver_tool(input: str) -> str:
@@ -96,6 +85,8 @@ builder.add_edge("search", END)
 
 app_graph = builder.compile()
 
+# --- Agent wrapper ---
+
 class BasicAgent:
     def __init__(self):
         print("LangGraph Agent initialized.")