Update app.py

app.py

@@ -166,7 +166,85 @@ def arxiv_search(query: str, load_max_docs: int = 3) -> Dict[str, str]:
     )
     return {"arxiv_results": formatted_search_docs}
 
+def extract_keywords(text: str) -> list:
+    """
+    Simple keyword extractor that splits text into unique keywords.
+    
+    Args:
+        text (str): Input text.
 
+    Returns:
+        list: List of extracted keywords.
+    """
+    words = text.lower().split()
+    keywords = list(set([w.strip(".,!?") for w in words if len(w) > 3]))
+    return keywords
+
+import re
+
+def calculate_expression(expression: str) -> str:
+    """
+    Evaluates a simple mathematical expression and returns the result.
+
+    Args:
+        expression (str): A math expression (e.g., "12 * (3+5) / 4").
+
+    Returns:
+        str: The result of the calculation or an error message if invalid.
+    """
+    try:
+        # Allow only numbers, operators, parentheses, decimal points, and spaces
+        if not re.match(r'^[\d\s\+\-\*\/\(\)\.]+$', expression):
+            return "Invalid characters detected in expression."
+        
+        result = eval(expression)
+        return str(result)
+    except Exception as e:
+        return f"Error evaluating expression: {str(e)}"
+
+def basic_calculator(a: float, b: float, operation: str) -> str:
+    """
+    Perform basic arithmetic operations between two numbers.
+
+    Args:
+        a (float): First number.
+        b (float): Second number.
+        operation (str): The operation to perform. Options: "add", "subtract", "multiply", "divide".
+
+    Returns:
+        str: The result of the calculation or an error message.
+    """
+    try:
+        if operation == "add":
+            return str(a + b)
+        elif operation == "subtract":
+            return str(a - b)
+        elif operation == "multiply":
+            return str(a * b)
+        elif operation == "divide":
+            if b == 0:
+                return "Error: Division by zero is not allowed."
+            return str(a / b)
+        else:
+            return "Invalid operation. Use add, subtract, multiply, or divide."
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+def (items: list, reverse: bool = False) -> list:
+    """
+    Sort a list of numbers or strings in ascending or descending order.
+
+    Args:
+        items (list): List of elements (numbers or strings) to be sorted.
+        reverse (bool): If True, sort in descending order. Default is False.
+
+    Returns:
+        list: Sorted list.
+    """
+    try:
+        return sorted(items, reverse=reverse)
+    except Exception as e:
+        return [f"Error: {str(e)}"]
 
 
 
@@ -183,9 +261,11 @@ class BasicAgent:
             "web_search": web_search,
             "arxiv_search": arxiv_search,
             "extract_keywords": extract_keywords,
-            "extract_dates": extract_dates,
-            "extract_names": extract_names,
-            "normalize_text": normalize_text,
+            "calculate_expression":calculate_expression,
+            "basic_calculator":basic_calculator,
+            "sort_list":sort_list
+
+          
         }
         self._cache = {}
         self.max_iterations = max_iterations