Update app.py

app.py

@@ -135,7 +135,7 @@ def create_pyodide_interface():
                     captured_output.getvalue()
                 `);
                 
-                // Check for plots
+                // Check for plots separately - don't mix with stdout
                 let plotData = null;
                 try {
                     plotData = pyodide.runPython(`
@@ -155,21 +155,31 @@ def create_pyodide_interface():
                 
                 outputDiv.style.display = 'block';
                 
-                let output = stdout || (result !== undefined ? String(result) : '');
-                outputText.textContent = output || 'Code executed successfully (no output)';
+                // Clean text output - exclude any base64 data that might have leaked
+                let cleanOutput = stdout || (result !== undefined ? String(result) : '');
+                // Filter out any base64 image data that might have accidentally been printed
+                cleanOutput = cleanOutput.replace(/iVBORw0KGgoAAAANSUhEUgAA[A-Za-z0-9+/=]+/g, '[Image data filtered]');
+                
+                outputText.textContent = cleanOutput || 'Code executed successfully (no output)';
                 
                 // Display plot if generated
-                if (plotData) {
+                if (plotData && plotData.length > 100) { // Valid base64 should be long
                     plotContainer.innerHTML = `<img src="data:image/png;base64,${plotData}" style="max-width: 100%; height: auto; border: 1px solid #ddd;" alt="Generated Plot">`;
-                    outputText.textContent += '\\n\\n📊 Plot generated successfully!';
+                    
+                    // Add plot success message to text (but not the raw base64)
+                    if (cleanOutput.trim()) {
+                        outputText.textContent += '\\n\\n📊 Plot generated and displayed below!';
+                    } else {
+                        outputText.textContent = '📊 Plot generated and displayed below!';
+                    }
                     updateStatus('✅ Code executed with plot generated!', 'green');
+                    return 'Code executed successfully with plot generated';
                 } else {
                     plotContainer.innerHTML = '';
                     updateStatus('✅ Code executed successfully!', 'green');
+                    return cleanOutput || 'Code executed successfully';
                 }
                 
-                return output + (plotData ? '\\n📊 Plot displayed above' : '');
-                
             } catch (error) {
                 console.error('Execution error:', error);
                 document.getElementById('output-text').textContent = 'Error: ' + error.toString();