Update app.py

app.py

@@ -4,90 +4,102 @@ import subprocess
 import shutil
 import numpy as np
 from PIL import Image
+import pywasm  # Add this to your requirements.txt
 
 # 1. Setup Persistent Paths
 SAVE_DIR = "/data"
 EMCC_DIR = os.path.join(SAVE_DIR, "emsdk")
 if not os.path.exists(SAVE_DIR): os.makedirs(SAVE_DIR, exist_ok=True)
 
-# 2. Function to ensure Emscripten exists in the bucket
+# Helper to get fresh file lists
+def get_wasm_files():
+    return [f for f in os.listdir(SAVE_DIR) if f.endswith('.wasm')]
+
+# --- COMPILER LOGIC ---
 def ensure_compiler():
-    # If already installed in bucket, just return the path
     emcc_path = os.path.join(EMCC_DIR, "upstream/emscripten/emcc")
-    if os.path.exists(emcc_path):
-        return emcc_path
-    
-    print("Installing Portable Emscripten to Bucket (One-time setup)...")
+    if os.path.exists(emcc_path): return emcc_path
     try:
-        # Clone EMSDK into the persistent bucket
         subprocess.run(["git", "clone", "https://github.com/emscripten-core/emsdk.git", EMCC_DIR], check=True)
         subprocess.run([os.path.join(EMCC_DIR, "emsdk"), "install", "latest"], check=True)
         subprocess.run([os.path.join(EMCC_DIR, "emsdk"), "activate", "latest"], check=True)
         return emcc_path
-    except Exception as e:
-        return f"Error installing compiler: {e}"
+    except Exception as e: return f"Error: {e}"
 
-def compile_c_logic(c_code, filename):
-    if not filename or not c_code: return "❌ Filename/Code required."
-    
-    # Ensure compiler is ready
+def compile_c(c_code, filename):
+    if not filename: return "❌ Need name."
     emcc_path = ensure_compiler()
     if "Error" in emcc_path: return emcc_path
+    
+    base = filename.split('.')[0]
+    c_f, out_js, out_wasm = f"/tmp/{base}.c", f"/tmp/{base}.js", f"/tmp/{base}.wasm"
+    with open(c_f, "w") as f: f.write(c_code)
+    
+    try:
+        # Exporting functions so they are callable by the runner
+        res = subprocess.run([emcc_path, c_f, "-s", "WASM=1", "-s", "EXPORTED_FUNCTIONS=['_main']", "-o", out_js], capture_output=True, text=True)
+        if res.returncode != 0: return f"❌ {res.stderr}"
+        shutil.move(out_wasm, os.path.join(SAVE_DIR, f"{base}.wasm"))
+        return f"✅ Compiled {base}.wasm"
+    except Exception as e: return f"❌ {e}"
 
-    base_name = filename.split('.')[0]
-    c_file = os.path.join("/tmp", f"{base_name}.c")
-    out_js = os.path.join("/tmp", f"{base_name}.js")
-    out_wasm = os.path.join("/tmp", f"{base_name}.wasm")
-    final_wasm = os.path.join(SAVE_DIR, f"{base_name}.wasm")
-
-    with open(c_file, "w") as f: f.write(c_code)
-
+# --- WASM RUNNER (The New "Execution" Tab) ---
+def run_wasm_func(wasm_file, func_name, args):
+    if not wasm_file: return "Select a file."
+    path = os.path.join(SAVE_DIR, wasm_file)
     try:
-        # Run compiler from the bucket path
-        result = subprocess.run(
-            [emcc_path, c_file, "-s", "WASM=1", "-o", out_js],
-            capture_output=True, text=True, timeout=60
-        )
-        
-        if result.returncode != 0: return f"❌ Compiler Error:\n{result.stderr}"
-        
-        if os.path.exists(out_wasm):
-            shutil.move(out_wasm, final_wasm)
-            return f"✅ Success! Compiled to: {final_wasm}"
-        return "❌ WASM not generated."
+        runtime = pywasm.load(path)
+        # Parse comma-separated args: "5, 10" -> [5, 10]
+        arg_list = [int(x.strip()) for x in args.split(",")] if args else []
+        result = runtime.exec(func_name, arg_list)
+        return f"Result: {result}"
     except Exception as e:
-        return f"❌ System Error: {str(e)}"
+        return f"Error running {func_name}: {e}"
 
-# --- WASM TO IMAGE LOGIC ---
-def render_wasm(wasm_filename):
-    if not wasm_filename: return None, "❌ Select a file."
-    path = os.path.join(SAVE_DIR, wasm_filename)
-    with open(path, "rb") as f: data = f.read()
-    
+# --- VISUALIZER ---
+def render_wasm(wasm_file):
+    if not wasm_file: return None, "Select a file."
+    with open(os.path.join(SAVE_DIR, wasm_file), "rb") as f: data = f.read()
     arr = np.frombuffer(data, dtype=np.uint8)
     canvas = np.zeros(1920 * 1080, dtype=np.uint8)
-    size = min(len(arr), 1920 * 1080)
-    canvas[:size] = arr[:size]
-    
-    img = Image.fromarray(canvas.reshape((1080, 1920)), mode='L')
-    return img, f"✅ Rendered {len(arr)} bytes to 1080p"
+    canvas[:min(len(arr), 1920*1080)] = arr[:min(len(arr), 1920*1080)]
+    return Image.fromarray(canvas.reshape((1080, 1920)), mode='L'), "Rendered."
 
-# --- GRADIO UI ---
+# --- UI ---
 with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
-    gr.Markdown("# 🛠️ LCPU Persistent Forge")
+    gr.Markdown("# 🛠️ LCPU: Forge, Run & Visualize")
     
-    with gr.Tab("1. Compiler"):
-        name = gr.Textbox(label="Filename", value="logic")
-        code = gr.TextArea(label="C Code", value="int add(int a, int b) { return a + b; }", lines=5)
-        btn = gr.Button("🚀 Build WASM")
-        stat = gr.Textbox(label="Status")
-        btn.click(compile_c_logic, [code, name], stat)
+    with gr.Tab("1. Forge"):
+        name = gr.Textbox(label="WASM Name", value="math_logic")
+        code = gr.TextArea(label="C Code", value='int add(int a, int b) { return a + b; }', lines=6)
+        comp_btn = gr.Button("🚀 Build")
+        status = gr.Textbox(label="Log")
 
-    with gr.Tab("2. Visualizer"):
-        files = gr.Dropdown(label="WASM Files", choices=[f for f in os.listdir(SAVE_DIR) if f.endswith('.wasm')])
-        v_btn = gr.Button("🎨 Render")
-        disp = gr.Image()
-        v_btn.click(render_wasm, files, [disp, stat])
+    with gr.Tab("2. Runner"):
+        with gr.Row():
+            run_select = gr.Dropdown(label="Select WASM", choices=get_wasm_files())
+            refresh_run = gr.Button("🔄", variant="secondary")
+        func_input = gr.Textbox(label="Function Name", value="add")
+        args_input = gr.Textbox(label="Arguments (comma separated)", placeholder="5, 10")
+        run_btn = gr.Button("⚡ Execute WASM")
+        run_out = gr.Textbox(label="Output")
+
+    with gr.Tab("3. Visualizer"):
+        with gr.Row():
+            vis_select = gr.Dropdown(label="Select WASM", choices=get_wasm_files())
+            refresh_vis = gr.Button("🔄", variant="secondary")
+        render_btn = gr.Button("🎨 Render 1080p")
+        img_out = gr.Image()
+
+    # Event Logic
+    comp_btn.click(compile_c, [code, name], status)
+    
+    # Live Refresh Logic
+    refresh_run.click(lambda: gr.Dropdown(choices=get_wasm_files()), outputs=run_select)
+    refresh_vis.click(lambda: gr.Dropdown(choices=get_wasm_files()), outputs=vis_select)
+    
+    run_btn.click(run_wasm_func, [run_select, func_input, args_input], run_out)
+    render_btn.click(render_wasm, vis_select, [img_out, status])
 
 if __name__ == "__main__":
     demo.launch()