Update app.py

app.py

@@ -2,72 +2,133 @@ import gradio as gr
 import os
 import subprocess
 import shutil
+import numpy as np
+from PIL import Image
 
-# Define the persistent directory
+# 1. Configuration & Paths
 SAVE_DIR = "/data"
 if not os.path.exists(SAVE_DIR):
     os.makedirs(SAVE_DIR, exist_ok=True)
 
+# 2. Compiler Logic
 def compile_c_logic(c_code, filename):
     if not filename or not c_code:
         return "❌ Error: Filename and C code are required."
     
-    # Ensure filename doesn't have extension for the command
+    # Strip extensions for processing
     base_name = filename.split('.')[0]
     c_file_path = os.path.join("/tmp", f"{base_name}.c")
-    wasm_output_path = os.path.join(SAVE_DIR, f"{base_name}.wasm")
     
-    # 1. Write the C code to a temp file
+    # Write the C code to /tmp
     with open(c_file_path, "w") as f:
         f.write(c_code)
     
-    # 2. Run Emscripten (emcc)
-    # Note: Hugging Face runs on Linux, so we call emcc directly
     try:
-        # -s WASM=1 tells it to produce a .wasm file
-        # -o name.js is required by emcc to generate the .wasm
         temp_js = os.path.join("/tmp", f"{base_name}.js")
+        temp_wasm = os.path.join("/tmp", f"{base_name}.wasm")
+        dest_wasm = os.path.join(SAVE_DIR, f"{base_name}.wasm")
+        
+        # We use 'emcc' directly. If packages.txt is processed, it should be in PATH.
+        # If it fails, we try calling it via its common absolute path.
+        compiler_cmd = "emcc"
+        
         result = subprocess.run(
-            ["emcc", c_file_path, "-s", "WASM=1", "-o", temp_js],
-            capture_output=True, text=True
+            [compiler_cmd, c_file_path, "-s", "WASM=1", "-o", temp_js],
+            capture_output=True, text=True, timeout=30
         )
         
         if result.returncode != 0:
             return f"❌ Compilation Error:\n{result.stderr}"
         
-        # 3. Move the generated .wasm from /tmp to persistent /data
-        temp_wasm = os.path.join("/tmp", f"{base_name}.wasm")
+        # Move generated .wasm to persistent storage
         if os.path.exists(temp_wasm):
-            shutil.move(temp_wasm, wasm_output_path)
-            return f"✅ Success! Compiled to: {wasm_output_path}\n\nConsole:\n{result.stdout}"
+            shutil.move(temp_wasm, dest_wasm)
+            return f"✅ Success! Compiled to: {dest_wasm}\n\nConsole:\n{result.stdout}"
         else:
             return "❌ Error: .wasm file was not generated."
             
     except Exception as e:
-        return f"❌ System Error: {str(e)}"
+        return f"❌ System Error: {str(e)}\n(Check if 'emscripten' is in packages.txt)"
+
+# 3. WASM to 1080p Image Logic
+def generate_wasm_canvas(wasm_filename):
+    if not wasm_filename:
+        return None, "❌ Select a WASM file first."
+    
+    path = os.path.join(SAVE_DIR, wasm_filename)
+    
+    with open(path, "rb") as f:
+        binary_data = f.read()
+    
+    data_array = np.frombuffer(binary_data, dtype=np.uint8)
+    canvas_size = 1920 * 1080
+    
+    # Padding/Truncating to fit 1080p exactly
+    if len(data_array) < canvas_size:
+        padded = np.zeros(canvas_size, dtype=np.uint8)
+        padded[:len(data_array)] = data_array
+        data_array = padded
+    else:
+        data_array = data_array[:canvas_size]
+    
+    img_array = data_array.reshape((1080, 1920))
+    img = Image.fromarray(img_array, mode='L')
+    
+    output_png_name = wasm_filename.replace(".wasm", ".png")
+    output_png_path = os.path.join(SAVE_DIR, output_png_name)
+    img.save(output_png_path)
+    
+    return img, f"✅ Generated 1080p Canvas: {output_png_name}"
+
+# 4. File Management Logic
+def refresh_list():
+    files = os.listdir(SAVE_DIR)
+    wasm_files = [f for f in files if f.endswith('.wasm')]
+    all_files = files
+    return gr.Dropdown(choices=wasm_files), gr.Dropdown(choices=all_files)
 
-# --- Reuse your previous file manager logic here ---
-def list_files():
-    return gr.Dropdown(choices=os.listdir(SAVE_DIR))
+def delete_file(filename):
+    if not filename: return "❌ No file selected.", refresh_list()[1]
+    try:
+        os.remove(os.path.join(SAVE_DIR, filename))
+        return f"🗑️ Deleted {filename}", refresh_list()[1]
+    except Exception as e:
+        return f"❌ Error: {str(e)}", refresh_list()[1]
 
+# 5. UI Layout
 with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
-    gr.Markdown("# 🛠️ LCPU: C-to-WASM & Bucket Manager")
+    gr.Markdown("# 🛠️ LCPU: Arithmetic-to-Spatial Engine")
     
-    with gr.Tab("C-to-WASM Compiler"):
-        gr.Markdown("### Write C logic and compile to Binary Pixels")
+    with gr.Tab("1. C-to-WASM Forge"):
         with gr.Row():
-            wasm_filename = gr.Textbox(label="Output Name", placeholder="adder_logic")
-        c_editor = gr.TextArea(label="C Source Code", placeholder="int main() { return 5 + 5; }", lines=10)
+            wasm_name = gr.Textbox(label="Output Name", placeholder="adder_logic")
+        c_code = gr.TextArea(label="C Source Code", placeholder="int add(int a, int b) { return a + b; }", lines=8)
         compile_btn = gr.Button("🚀 Compile to WASM", variant="primary")
-        compile_status = gr.Textbox(label="Compiler Output")
+        comp_status = gr.Textbox(label="Compiler Output")
 
-    with gr.Tab("Bucket Management"):
-        # (Insert your previous Save/Delete/Refresh logic here)
-        file_list = gr.Dropdown(label="Files in /data", choices=os.listdir(SAVE_DIR))
-        refresh_btn = gr.Button("🔄 Refresh")
-        refresh_btn.click(fn=list_files, outputs=file_list)
+    with gr.Tab("2. Spatial Visualizer"):
+        wasm_select = gr.Dropdown(label="Select WASM from Bucket", choices=[f for f in os.listdir(SAVE_DIR) if f.endswith('.wasm')])
+        render_btn = gr.Button("🎨 Render to 1080p Canvas", variant="primary")
+        canvas_display = gr.Image(label="1080p Logic Map")
+        render_status = gr.Textbox(label="Render Status")
 
-    compile_btn.click(fn=compile_c_logic, inputs=[c_editor, wasm_filename], outputs=compile_status)
+    with gr.Tab("3. Bucket Manager"):
+        all_files_list = gr.Dropdown(label="All Files in /data", choices=os.listdir(SAVE_DIR))
+        with gr.Row():
+            ref_btn = gr.Button("🔄 Refresh List")
+            del_btn = gr.Button("🗑️ Delete Selected", variant="stop")
+        status_box = gr.Textbox(label="Action Status")
+
+    # Interactions
+    compile_btn.click(fn=compile_c_logic, inputs=[c_code, wasm_name], outputs=comp_status)
+    
+    render_btn.click(fn=generate_wasm_canvas, inputs=wasm_select, outputs=[canvas_display, render_status])
+    
+    del_btn.click(fn=delete_file, inputs=all_files_list, outputs=[status_box, all_files_list])
+    
+    ref_btn.click(fn=lambda: (gr.Dropdown(choices=[f for f in os.listdir(SAVE_DIR) if f.endswith('.wasm')]), 
+                              gr.Dropdown(choices=os.listdir(SAVE_DIR))), 
+                  outputs=[wasm_select, all_files_list])
 
 if __name__ == "__main__":
     demo.launch()