Update app.py

app.py

@@ -95,6 +95,61 @@ class KeywordsExtractorTool(Tool):
         except Exception as e:
             return f"Error during extracting most common words: {e}"
 
+
+@tool
+def calculator(inputs: Union[str, dict]):
+    """
+    Perform mathematical operations based on the operation provided.
+    Supports both binary (a, b) operations and list operations.
+    """
+
+    # If input is a JSON string, parse it
+    if isinstance(inputs, str):
+        try:
+            import json
+            inputs = json.loads(inputs)
+        except Exception as e:
+            return f"Invalid input format: {e}"
+
+    # Handle list-based operations like SUM
+    if "list" in inputs:
+        nums = inputs.get("list", [])
+        op = inputs.get("operation", "").lower()
+
+        if not isinstance(nums, list) or not all(isinstance(n, (int, float)) for n in nums):
+            return "Invalid list input. Must be a list of numbers."
+
+        if op == "sum":
+            return sum(nums)
+        elif op == "multiply":
+            return reduce(operator.mul, nums, 1)
+        else:
+            return f"Unsupported list operation: {op}"
+
+    # Handle basic two-number operations
+    a = inputs.get("a")
+    b = inputs.get("b")
+    operation = inputs.get("operation", "").lower()
+
+    if a is None or b is None or not isinstance(a, (int, float)) or not isinstance(b, (int, float)):
+        return "Both 'a' and 'b' must be numbers."
+
+    if operation == "add":
+        return a + b
+    elif operation == "subtract":
+        return a - b
+    elif operation == "multiply":
+        return a * b
+    elif operation == "divide":
+        if b == 0:
+            return "Error: Division by zero"
+        return a / b
+    elif operation == "modulus":
+        return a % b
+    else:
+        return f"Unknown operation: {operation}"
+
+
 @tool
 def parse_excel_to_json(task_id: str) -> dict:
     """
@@ -197,19 +252,39 @@ class BasicAgent:
         video_transcription_tool = VideoTranscriptionTool()
 
         system_prompt = f"""
-You are my general AI assistant. Your task is to answer the question I asked.
-First, provide an explanation of your reasoning, step by step, to arrive at the answer.
-Then, return your final answer in a single line, formatted as follows: "FINAL ANSWER: [YOUR FINAL ANSWER]".
-[YOUR FINAL ANSWER] should be a number, a string, or a comma-separated list of numbers and/or strings, depending on the question.
-If the answer is a number, do not use commas or units (e.g., $, %) unless specified.
-If the answer is a string, do not use articles or abbreviations (e.g., for cities), and write digits in plain text unless specified.
-If the answer is a comma-separated list, apply the above rules for each element based on whether it is a number or a string.
+You are an advanced, helpful, and highly analytical research assistant. Your goal is to provide accurate, comprehensive, and well-structured answers to user queries, leveraging all available tools efficiently.
+
+**Follow this robust process:**
+
+1.  **Understand the User's Need:** Carefully analyze the user's question, including any attached files or specific requests (e.g., "summarize," "analyze data," "find facts").
+2.  **Formulate a Detailed Plan:** Before acting, create a clear, step-by-step plan. This plan should outline:
+    * What information needs to be gathered.
+    * Which tools are most appropriate for each step (e.g., `duckduckgo_search` for general web search, `wiki_search` for encyclopedic facts, `transcript_video` for YouTube, `file_analysis` or `data_analysis` for local files).
+    * How you will combine information from different sources.
+    * How you will verify or synthesize the findings.
+3.  **Execute the Plan Using Tools:** Call the necessary tools, providing clear and correct arguments. If a tool fails, try to understand why and adapt your plan (e.g., try a different search query or tool).
+4.  **Synthesize and Verify Information:** Once you have gathered sufficient information, synthesize it into a coherent answer. Do not just list facts; explain their significance and how they relate to the original question. If there are contradictions or uncertainties, mention them.
+5.  **Formulate the Final Answer:**
+    * Present your answer clearly and concisely.
+    * Always begin your ultimate response with "FINAL ANSWER:".
+    * If the answer is a single number, provide only the number.
+    * If the answer is a list, provide comma-separated values.
+    * For complex answers, use structured formats like bullet points or JSON where appropriate to enhance readability.
+    * **Crucially, always include sources or references (e.g., URLs, Wikipedia titles, file names) where you obtained the information.** This builds trust and allows for verification.
+    * If you used `file_analysis` or `data_analysis` tools on an uploaded file, explicitly state that you analyzed the provided file.
+
+**Important Considerations:**
+* **Prioritize:** If the query involves a specific file, start by analyzing that file if appropriate.
+* **Ambiguity:** If the question is ambiguous, ask for clarification.
+* **Limitations:** If you cannot answer a question with the available tools, state that clearly.
+* **Conciseness:** Be as concise as possible while providing a complete and accurate answer.
 """
         self.agent = CodeAgent(
             model=model,
             tools=[search_tool, wiki_search_tool, str_reverse_tool, keywords_extract_tool, speech_to_text_tool, visit_webpage_tool, \
                    final_answer_tool, parse_excel_to_json, video_transcription_tool],
-            add_base_tools=True
+            add_base_tools=True,
+            max_steps=15 
         )
         self.agent.prompt_templates["system_prompt"] = self.agent.prompt_templates["system_prompt"] + system_prompt