Update app.py

app.py

@@ -1,157 +1,85 @@
 import gradio as gr
-import subprocess
-import tempfile
-import os
+import io
 import sys
-from pathlib import Path
+import matplotlib.pyplot as plt
+from PIL import Image
 
-def execute_code(code, language):
-    """Execute code and return the output and any errors."""
-    with tempfile.TemporaryDirectory() as tmp_dir:
-        if language == "python":
-            file_path = Path(tmp_dir) / "script.py"
-            with open(file_path, "w") as f:
-                f.write(code)
-            
-            # Execute the code and capture output
-            try:
-                result = subprocess.run(
-                    [sys.executable, file_path],
-                    capture_output=True,
-                    text=True,
-                    timeout=10  # 10 second timeout for safety
-                )
-                stdout = result.stdout
-                stderr = result.stderr
-                
-                if result.returncode != 0:
-                    return f"Error:\n{stderr}\n\nOutput:\n{stdout}"
-                return stdout
-            except subprocess.TimeoutExpired:
-                return "Execution timed out (limit: 10 seconds)"
-            except Exception as e:
-                return f"Error: {e}"
-        
-        elif language == "javascript":
-            file_path = Path(tmp_dir) / "script.js"
-            with open(file_path, "w") as f:
-                f.write(code)
-            
-            try:
-                result = subprocess.run(
-                    ["node", file_path],
-                    capture_output=True,
-                    text=True,
-                    timeout=10
-                )
-                stdout = result.stdout
-                stderr = result.stderr
-                
-                if result.returncode != 0:
-                    return f"Error:\n{stderr}\n\nOutput:\n{stdout}"
-                return stdout
-            except subprocess.TimeoutExpired:
-                return "Execution timed out (limit: 10 seconds)"
-            except FileNotFoundError:
-                return "Error: Node.js is not installed or not in PATH"
-            except Exception as e:
-                return f"Error: {e}"
-        
-        else:
-            return f"Language '{language}' is not supported"
+# Function to execute code and capture output
+def run_code(code, namespace):
+    output = io.StringIO()
+    sys.stdout = output
+    images = []
+    try:
+        exec(code, namespace)
+        # Capture any matplotlib plots
+        figs = [plt.figure(i) for i in plt.get_fignums()]
+        for fig in figs:
+            buf = io.BytesIO()
+            fig.savefig(buf, format='png')
+            buf.seek(0)
+            img = Image.open(buf)
+            images.append(img)
+        plt.close('all')
+    except Exception as e:
+        print(f"Error: {e}")
+    sys.stdout = sys.__stdout__
+    return output.getvalue(), images, namespace
 
-# Example code for each language
-python_example = """# Python Example
-def fibonacci(n):
-    if n <= 0:
-        return []
-    elif n == 1:
-        return [0]
-    elif n == 2:
-        return [0, 1]
-    
-    fib = [0, 1]
-    for i in range(2, n):
-        fib.append(fib[i-1] + fib[i-2])
-    
-    return fib
+# Function to load code for the selected file
+def select_file(file_name, files):
+    return files[file_name]
 
-# Generate first 10 Fibonacci numbers
-result = fibonacci(10)
-print(f"First 10 Fibonacci numbers: {result}")
-"""
+# Function to update the files state when code is edited
+def update_code(code, file_name, files):
+    files[file_name] = code
+    return files
 
-js_example = """// JavaScript Example
-function fibonacci(n) {
-    if (n <= 0) return [];
-    if (n === 1) return [0];
-    if (n === 2) return [0, 1];
-    
-    const fib = [0, 1];
-    for (let i = 2; i < n; i++) {
-        fib.push(fib[i-1] + fib[i-2]);
-    }
-    
-    return fib;
-}
+# Function to add a new file
+def add_file(files):
+    new_file_name = f"file_{len(files)}.py"
+    files[new_file_name] = "# New file"
+    return files, list(files.keys()), new_file_name
 
-// Generate first 10 Fibonacci numbers
-const result = fibonacci(10);
-console.log(`First 10 Fibonacci numbers: ${result}`);
-"""
+# Function to delete the current file
+def delete_file(file_name, files):
+    if len(files) > 1:  # Prevent deleting the last file
+        del files[file_name]
+        new_file_name = list(files.keys())[0]
+        return files, list(files.keys()), new_file_name, files[new_file_name]
+    else:
+        return files, list(files.keys()), file_name, files[file_name]
 
-# Create the Gradio interface
-with gr.Blocks(title="Simple Code IDE") as app:
-    gr.Markdown("# Simple Code IDE")
-    gr.Markdown("Write, run, and see the output of your code!")
+# Build the interface
+with gr.Blocks(title="Simple IDE with Gradio") as demo:
+    # Initial state
+    initial_files = {"main.py": "# Write your code here\nprint('Hello, World!')"}
     
-    with gr.Row():
-        with gr.Column(scale=2):
-            language = gr.Dropdown(
-                ["python", "javascript"], 
-                label="Programming Language", 
-                value="python"
-            )
-            
-            code_input = gr.Code(
-                label="Code Editor", 
-                language="python", 
-                value=python_example
-            )
-            
-            # Update code example when language changes
-            def update_example(lang):
-                if lang == "python":
-                    return python_example, "python"
-                elif lang == "javascript":
-                    return js_example, "javascript"
-            
-            language.change(
-                update_example, 
-                inputs=language, 
-                outputs=[code_input, code_input.language]
-            )
-            
-            run_button = gr.Button("Run Code", variant="primary")
-        
-        with gr.Column(scale=1):
-            output = gr.Textbox(label="Output", lines=20)
+    # State components
+    files = gr.State(initial_files)
+    namespace = gr.State({})
     
-    run_button.click(
-        execute_code, 
-        inputs=[code_input, language], 
-        outputs=output
-    )
-    
-    gr.Markdown("""
-    ## Instructions
-    1. Select your programming language
-    2. Write or edit the code in the editor
-    3. Click "Run Code" to execute and see the output
-    
-    Note: Code execution is limited to 10 seconds for safety.
-    """)
+    # UI components
+    file_selector = gr.Dropdown(choices=list(initial_files.keys()), label="Select File")
+    code_input = gr.Code(label="Code Editor", language="python", value=initial_files["main.py"])
+    with gr.Row():
+        run_button = gr.Button("Run")
+        add_file_button = gr.Button("Add File")
+        delete_file_button = gr.Button("Delete File")
+        clear_button = gr.Button("Clear Output")
+        reset_button = gr.Button("Reset Namespace")
+    output_display = gr.Textbox(label="Output")
+    image_gallery = gr.Gallery(label="Plots")
+
+    # Event handlers
+    file_selector.change(select_file, inputs=[file_selector, files], outputs=code_input)
+    code_input.change(update_code, inputs=[code_input, file_selector, files], outputs=files)
+    add_file_button.click(add_file, inputs=files, outputs=[files, file_selector, file_selector])
+    delete_file_button.click(delete_file, inputs=[file_selector, files], 
+                            outputs=[files, file_selector, file_selector, code_input])
+    run_button.click(run_code, inputs=[code_input, namespace], 
+                     outputs=[output_display, image_gallery, namespace])
+    clear_button.click(lambda: ("", []), outputs=[output_display, image_gallery])
+    reset_button.click(lambda: ({}, "", []), outputs=[namespace, output_display, image_gallery])
 
-# Launch the app
-if __name__ == "__main__":
-    app.launch()
+# Launch the interface
+demo.launch()