ok

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "python-envs.defaultEnvManager": "ms-python.python:system"
+}

controller.py

@@ -49,6 +49,7 @@ from pydantic_mongo_executor_model import ExecutorPayload, ExecutorResponse
 from report_service import FileBoxProps, ReportRequest, execute_report_generation
 from supabase_service import upload_bytes_to_supabase, upload_file_to_supabase
 from table_info_model import CsvFieldsRequest, CsvFieldsResponse
+from file_uploader import ExternalFileUploader
 
 # --- Configuration & Setup ---
 load_dotenv()
@@ -59,6 +60,8 @@ logging.basicConfig(
 )
 logger = logging.getLogger("API_Controller")
 
+uploader = ExternalFileUploader()
+
 # ==============================================================================
 # ASYNC CONNECTION POOL MANAGER (UPDATED WITH SSL FALLBACK)
 # ==============================================================================
@@ -630,7 +633,11 @@ async def execute_chart_endpoint(payload: ChartExecutionPayload, token: str = De
         image_bytes, error_msg, logs = await run_in_threadpool(execute_python_code, code=payload.code, csv_url=payload.csv_url)
         if error_msg: return ChartExecutionResponse(status="error", error=error_msg, output_log=logs, request_id=request_id)
         if payload.return_base64:
+            unique_name = f"{uuid.uuid4()}.png"
             base64_str = base64.b64encode(image_bytes).decode('utf-8')
+            file_url = await run_in_threadpool(uploader.upload_file, file_bytes=image_bytes, file_name=unique_name)
+            if file_url:
+                return ChartExecutionResponse(status="success", image_url=file_url, output_log=logs, request_id=request_id)
             return ChartExecutionResponse(status="success", base64_image=base64_str, output_log=logs, request_id=request_id)
         else:
             unique_name = f"{uuid.uuid4()}.png"

file_uploader.py

@@ -0,0 +1,47 @@
+import os
+import requests
+from typing import Optional
+
+
+class ExternalFileUploader:
+    def __init__(self):
+        self.base_url = os.getenv("FILE_UPLOAD_API_URL")
+        self.api_key = os.getenv("FILE_UPLOAD_API_KEY")
+
+    def upload_file(self, file_bytes: bytes, file_name: str) -> Optional[str]:
+        try:
+            files = {
+                "files": (file_name, file_bytes, "image/png")
+            }
+
+            headers = {
+                "X-API-Key": self.api_key
+            }
+
+            response = requests.post(
+                f"{self.base_url}/api/v1/files/bulk-upload",
+                headers=headers,
+                files=files,
+                timeout=30
+            )
+
+            if response.status_code != 200:
+                print("Upload failed:", response.text)
+                return None
+
+            data = response.json()
+
+            # Expecting array response
+            if not isinstance(data, list) or not data:
+                return None
+
+            first = data[0]
+
+            if not first.get("success"):
+                return None
+
+            return first["data"]["file_url"]
+
+        except Exception as e:
+            print("Uploader error:", str(e))
+            return None