add python_interpreter tool

agent.py

@@ -23,10 +23,11 @@ from youtube_transcript_api import YouTubeTranscriptApi
 from PIL import Image
 import pytesseract
 import fitz  # PyMuPDF
+from dotenv import load_dotenv
+from contextlib import redirect_stdout
 
 # Load environment variables from .env file
 # in HF Spaces, the .env file is saved in Variables and secrets in settings
-from dotenv import load_dotenv
 load_dotenv()
 
 # === System Prompt ===
@@ -179,11 +180,38 @@ def read_jsonl(jsonl_path: str) -> str:
     except Exception as e:
         return f"Error reading JSONL file: {e}"
 
+@tool
+def python_interpreter(code: str) -> str:
+    """Execute Python code and return the output. Supports data analysis, plotting, and general Python operations."""
+    try:
+
+        
+        # Set up a safe globals environment
+        safe_globals = {
+            'pd': __import__('pandas'),
+            'np': __import__('numpy'),
+            'plt': __import__('matplotlib.pyplot'),
+            'json': __import__('json'),
+            're': __import__('re'),
+            'math': __import__('math'),
+        }
+        
+        # Capture output
+        buffer = io.StringIO()
+        with redirect_stdout(buffer):
+            # Execute the code in a safe environment
+            exec(code, safe_globals)
+            
+        return buffer.getvalue() or "Code executed successfully (no output)"
+    except Exception as e:
+        return f"Error executing Python code: {e}"
+
 # Update tools list
 tools: List[StructuredTool] = [
     calculate, web_search, wikipedia_search, image_recognition,
     read_pdf, read_csv, read_spreadsheet, transcribe_audio,
-    youtube_transcript_tool, youtube_transcript_api, read_jsonl  # Add read_jsonl
+    youtube_transcript_tool, youtube_transcript_api, read_jsonl,
+    python_interpreter  # Add python_interpreter
 ]
 
 class AgentState(TypedDict):