Update app.py

app.py

@@ -18,64 +18,57 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
-@tool
-def extended_calculator(operation: str) -> str:
-    """An extended calculator that safely evaluates math expressions.
-    
-    Supports:
-      - Basic operations: +, -, *, /, and parentheses.
-      - Exponentiation: using either '^' or '**'.
-      - Math functions: sqrt, sin, cos, tan, log, exp.
-      
+def duckduckgo_search(query: str) -> str:
+    """
+    A simple search tool that queries DuckDuckGo's Instant Answer API and returns a result snippet.
+
     Args:
-        operation: A string mathematical expression (e.g., "2 + 2", "sqrt(16)", "2^3").
-    
+        query (str): The search query string.
+
     Returns:
-        str: The result of the calculation or an error message.
+        str: A summary from DuckDuckGo or an error message.
     """
     try:
-        # Replace caret '^' with Python's exponentiation operator '**'
-        operation = operation.replace("^", "**")
-        
-        # Define allowed names for math functions and constants.
-        safe_env = {
-            "sqrt": math.sqrt,
-            "sin": math.sin,
-            "cos": math.cos,
-            "tan": math.tan,
-            "log": math.log,   # natural logarithm by default
-            "exp": math.exp,
-            "pi": math.pi,
-            "e": math.e,
-            "abs": abs,
-            "pow": pow,
+        url = "https://api.duckduckgo.com/"
+        params = {
+            "q": query,
+            "format": "json",
+            "no_html": 1,
+            "skip_disambig": 1,
         }
+        response = requests.get(url, params=params)
+        if response.status_code != 200:
+            return f"Error: Received status code {response.status_code} from DuckDuckGo."
+
+        data = response.json()
+        
+        # Use the AbstractText if available as a summary.
+        abstract_text = data.get("AbstractText", "")
+        if abstract_text:
+            return f"Result: {abstract_text}"
         
-        # Use a regex to allow only safe characters (digits, letters, operators, parentheses, commas, and periods)
-        # This helps reject inputs with disallowed symbols.
-        if not re.match(r'^[0-9a-zA-Z\s+\-*/().,]+$', operation):
-            return "Error: Expression contains invalid characters."
+        # If no abstract is provided, try returning the first related topic's text.
+        related_topics = data.get("RelatedTopics", [])
+        for topic in related_topics:
+            # Some topics might be nested; check both levels.
+            if isinstance(topic, dict):
+                if "Text" in topic:
+                    return f"Result: {topic['Text']}"
+                if "Topics" in topic:
+                    for subtopic in topic["Topics"]:
+                        if "Text" in subtopic:
+                            return f"Result: {subtopic['Text']}"
         
-        # Evaluate the expression using the restricted environment.
-        result = eval(operation, {"__builtins__": None}, safe_env)
-        return f"Result: {result}"
+        return "No results found."
     
     except Exception as e:
-        return f"Error calculating {operation}: {str(e)}"
+        return f"Error performing search: {str(e)}"
 
 # Example usage:
 if __name__ == "__main__":
-    expressions = [
-        "2 + 2",
-        "3 * (4 + 5)",
-        "2^3",           # using '^' for exponentiation
-        "sqrt(16)",
-        "sin(pi/2)",
-        "log(e)"
-    ]
+    search_query = "Python programming"
+    print(duckduckgo_search(search_query))
     
-    for expr in expressions:
-        print(f"{expr} = {extended_calculator(expr)}")
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.