Update app.py

app.py

@@ -371,6 +371,7 @@ if _output_buffer:
 executor = SecureCodeExecutor()
  
 # MCP Functions
+# Replace the execute_python_code function with this enhanced version
 def execute_python_code(code: str, session_id: str = None) -> str:
     """
     Execute Python code safely with visualization support.
@@ -380,9 +381,39 @@ def execute_python_code(code: str, session_id: str = None) -> str:
         session_id (str, optional): Session ID for persistent context
     
     Returns:
-        str: JSON string with execution results
+        str: JSON string with execution results, including image data for LibreChat
     """
-    return executor.execute_code(code, "python", session_id)
+    result_json = executor.execute_code(code, "python", session_id)
+    result = json.loads(result_json)
+    
+    # If there's a plot, we need to format it specially for LibreChat
+    if result.get("plot"):
+        plot_base64 = result["plot"]
+        
+        # Create a response that LibreChat can understand
+        # Include both the text output and indicate there's an image
+        response = {
+            "success": result["success"],
+            "stdout": result.get("stdout", ""),
+            "stderr": result.get("stderr", ""),
+            "session_id": result.get("session_id"),
+            "execution_time": result.get("execution_time"),
+            # Add image information
+            "images": [
+                {
+                    "type": "image/png",
+                    "data": plot_base64,
+                    "description": "Generated matplotlib plot"
+                }
+            ],
+            # Also include a message about the image
+            "message": f"{result.get('stdout', '')}\n\n📊 Generated visualization (matplotlib plot)"
+        }
+        
+        return json.dumps(response, indent=2)
+    
+    # If no plot, return the original result
+    return result_json
  
 def execute_javascript_code(code: str, session_id: str = None) -> str:
     """