update tools

agent.py

@@ -21,8 +21,9 @@ import pandas as pd
 from transformers import BlipProcessor, BlipForConditionalGeneration
 from PIL import Image
 import torch
+import matplotlib.pyplot as plt
 import cmath
-from code_interpreter import CodeInterpreter
+# from code_interpreter import CodeInterpreter
 import uuid
 import tempfile
 import requests
@@ -147,69 +148,69 @@ def blip_image_caption(image_path: str) -> str:
     except Exception as e:
         return f"Failed to process image with BLIP: {str(e)}"
     
-@tool
-def execute_code_multilang(code: str, language: str = "python") -> str:
-    """Execute code in multiple languages (Python, Bash, SQL, C, Java) and return results.
-    Args:
-        code (str): The source code to execute.
-        language (str): The language of the code. Supported: "python", "bash", "sql", "c", "java".
-    Returns:
-        A string summarizing the execution results (stdout, stderr, errors, plots, dataframes if any).
-    """
-    supported_languages = ["python", "bash", "sql", "c", "java"]
-    language = language.lower()
-    interpreter_instance = CodeInterpreter()
-
-    if language not in supported_languages:
-        return f"❌ Unsupported language: {language}. Supported languages are: {', '.join(supported_languages)}"
-
-    result = interpreter_instance.execute_code(code, language=language)
-
-    response = []
-
-    if result["status"] == "success":
-        response.append(f"✅ Code executed successfully in **{language.upper()}**")
-
-        if result.get("stdout"):
-            response.append(
-                "\n**Standard Output:**\n```\n" + result["stdout"].strip() + "\n```"
-            )
-
-        if result.get("stderr"):
-            response.append(
-                "\n**Standard Error (if any):**\n```\n"
-                + result["stderr"].strip()
-                + "\n```"
-            )
-
-        if result.get("result") is not None:
-            response.append(
-                "\n**Execution Result:**\n```\n"
-                + str(result["result"]).strip()
-                + "\n```"
-            )
-
-        if result.get("dataframes"):
-            for df_info in result["dataframes"]:
-                response.append(
-                    f"\n**DataFrame `{df_info['name']}` (Shape: {df_info['shape']})**"
-                )
-                df_preview = pd.DataFrame(df_info["head"])
-                response.append("First 5 rows:\n```\n" + str(df_preview) + "\n```")
-
-        if result.get("plots"):
-            response.append(
-                f"\n**Generated {len(result['plots'])} plot(s)** (Image data returned separately)"
-            )
-
-    else:
-        response.append(f"❌ Code execution failed in **{language.upper()}**")
-        if result.get("stderr"):
-            response.append(
-                "\n**Error Log:**\n```\n" + result["stderr"].strip() + "\n```"
-            )
-
-    return "\n".join(response)
+# @tool
+# def execute_code_multilang(code: str, language: str = "python") -> str:
+#     """Execute code in multiple languages (Python, Bash, SQL, C, Java) and return results.
+#     Args:
+#         code (str): The source code to execute.
+#         language (str): The language of the code. Supported: "python", "bash", "sql", "c", "java".
+#     Returns:
+#         A string summarizing the execution results (stdout, stderr, errors, plots, dataframes if any).
+#     """
+#     supported_languages = ["python", "bash", "sql", "c", "java"]
+#     language = language.lower()
+#     interpreter_instance = CodeInterpreter()
+
+#     if language not in supported_languages:
+#         return f"❌ Unsupported language: {language}. Supported languages are: {', '.join(supported_languages)}"
+
+#     result = interpreter_instance.execute_code(code, language=language)
+
+#     response = []
+
+#     if result["status"] == "success":
+#         response.append(f"✅ Code executed successfully in **{language.upper()}**")
+
+#         if result.get("stdout"):
+#             response.append(
+#                 "\n**Standard Output:**\n```\n" + result["stdout"].strip() + "\n```"
+#             )
+
+#         if result.get("stderr"):
+#             response.append(
+#                 "\n**Standard Error (if any):**\n```\n"
+#                 + result["stderr"].strip()
+#                 + "\n```"
+#             )
+
+#         if result.get("result") is not None:
+#             response.append(
+#                 "\n**Execution Result:**\n```\n"
+#                 + str(result["result"]).strip()
+#                 + "\n```"
+#             )
+
+#         if result.get("dataframes"):
+#             for df_info in result["dataframes"]:
+#                 response.append(
+#                     f"\n**DataFrame `{df_info['name']}` (Shape: {df_info['shape']})**"
+#                 )
+#                 df_preview = pd.DataFrame(df_info["head"])
+#                 response.append("First 5 rows:\n```\n" + str(df_preview) + "\n```")
+
+#         if result.get("plots"):
+#             response.append(
+#                 f"\n**Generated {len(result['plots'])} plot(s)** (Image data returned separately)"
+#             )
+
+#     else:
+#         response.append(f"❌ Code execution failed in **{language.upper()}**")
+#         if result.get("stderr"):
+#             response.append(
+#                 "\n**Error Log:**\n```\n" + result["stderr"].strip() + "\n```"
+#             )
+
+#     return "\n".join(response)
 
 @tool
 def save_and_read_file(content: str, filename: Optional[str] = None) -> str:
@@ -291,7 +292,7 @@ def analyze_csv_file(file_path: str, query: str) -> str:
         return f"Error analyzing CSV file: {str(e)}"
 tools = [multiply, add, subtract, divide, modulus, 
          wiki_search, web_search, arvix_search, read_excel_file, extract_text_from_pdf,
-         blip_image_caption, execute_code_multilang, save_and_read_file, download_file_from_url, analyze_csv_file]
+         blip_image_caption, save_and_read_file, download_file_from_url, analyze_csv_file]
 
 # ------------------- SYSTEM PROMPT -------------------
 system_prompt_path = "system_prompt.txt"