Update app.py

app.py

@@ -9,27 +9,31 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
+@tool
 def crypto_duck(addr: str) -> str:
-    """Search DuckDuckGo for information about a cryptocurrency address and return the top result."""
+    """Search DuckDuckGo for information about a cryptocurrency address and return the top result.
+
+    Args:
+        addr: The cryptocurrency address to search for (e.g., a Bitcoin or Ethereum address).
+    """
     try:
         ddg = DuckDuckGoSearchTool(max_results=1)
         query = f'cryptocurrency address "{addr}"'
-        # Some versions expose .run; others are callable. Try .run first:
-        if hasattr(ddg, "run"):
-            res = ddg.run(query=query)
-        else:
-            res = ddg(query)
-        # res may already be a string; if it’s a list of dicts, format the first
+        # Support both callable and .run signatures across versions
+        res = ddg.run(query=query) if hasattr(ddg, "run") else ddg(query)
+
+        # If the tool returns a list[dict], format the first result
         if isinstance(res, list) and res:
             top = res[0]
             title = top.get("title", "")
             href = top.get("href") or top.get("link", "")
             snippet = top.get("body") or top.get("snippet", "")
             return f"{title}\n{snippet}\n{href}".strip()
+
+        # If it returns a markdown string, strip any header the custom class adds
         if isinstance(res, str):
-            if res.startswith("## Search Results"):
-                res = res.split("\n\n", 1)[-1]
-            return res or "No results found."
+            return res.split("\n\n", 1)[-1] if res.startswith("## Search Results") else (res or "No results found.")
+
         return "No results found."
     except Exception as e:
         return f"Search error: {str(e)}"