Function executePythonCode

frontend/src/lib/python/executor.ts:8–176

(
  code: string
)

 * Execute Python code and capture outputs
 */
export async function executePythonCode(
  code: string
): Promise<PythonExecutionResult> {
  const pyodide = getPyodide();
  if (!pyodide) {
    throw new Error('Pyodide not initialized');
  }

  const startTime = performance.now();
  const outputs: CellOutput[] = [];
  let error: string | undefined;

  try {
    // Setup output capture
    await pyodide.runPython(`
      import sys
      import io
      from contextlib import redirect_stdout, redirect_stderr
      
      # Create string buffers for capturing output
      _stdout_buffer = io.StringIO()
      _stderr_buffer = io.StringIO()
      _output_captured = []
      
      # Custom print function to capture output
      _original_print = print
      def _capture_print(*args, **kwargs):
          # Capture in buffer
          _original_print(*args, file=_stdout_buffer, **kwargs)
          # Also print normally for console
          _original_print(*args, **kwargs)
      
      print = _capture_print
    `);

    // Execute the user code
    let result;
    try {
      // Check if code needs async handling
      const needsAsync =
        code.includes('sql(') ||
        code.includes('query(') ||
        code.includes('sql_bridge.') ||
        code.includes('await ') ||
        code.includes('micropip.install');

      if (needsAsync) {
        // Use runPythonAsync for code that might call async functions
        result = await pyodide.runPythonAsync(code);
      } else {
        // Use regular runPython for synchronous code
        result = pyodide.runPython(code);
      }
    } catch (pythonError: any) {
      error = `${pythonError.name}: ${pythonError.message}`;

      outputs.push({
        id: crypto.randomUUID(),