commit

app.py

@@ -254,8 +254,7 @@ def render_grid_image(grid: List[List[str]]) -> str:
     numeric_grid = np.array([[tile_map[cell] for cell in row] for row in grid])
 
     # Create output folder if not exists
-    os.makedirs("static", exist_ok=True)
-    image_path = "static/level.png"
+    image_path = "/tmp/level.png"
 
     fig, ax = plt.subplots(figsize=(6, 6))
     ax.imshow(numeric_grid, cmap=color_map, interpolation='none')
@@ -295,7 +294,7 @@ render_image_agent = create_react_agent(
         "- Only use the render_grid_image tool to render the grid.\n"
         "- When responding, ONLY return the raw file path returned by the tool.\n"
         "- Do NOT add any commentary or formatting.\n"
-        "- Example: /static/level.png"
+        "- Example: /tmp/level.png"
     ),
     name="render_image_agent"
 )
@@ -329,7 +328,7 @@ def run_streamed_supervisor(user_input):
 
     final_messages = chunk["supervisor"]["messages"]
 
-    # Search the last messages for image path
+    # Search the last messages for image path like /tmp/level.png
     for message in reversed(final_messages):
         if hasattr(message, "content") and isinstance(message.content, str):
             match = re.search(r"(/tmp/level\.png)", message.content)
@@ -337,9 +336,12 @@ def run_streamed_supervisor(user_input):
                 image_path = match.group(1)
                 break
 
-    # If path exists, open and return image
+    # ✅ Copy image to static/ and return filepath
     if image_path and os.path.exists(image_path):
-        return Image.open(image_path)
+        os.makedirs("static", exist_ok=True)
+        static_path = os.path.join("static", os.path.basename(image_path))
+        shutil.copy(image_path, static_path)
+        return static_path  # ✔️ Gradio will render it from here
 
     return "❌ Image path not found or file does not exist."
 
@@ -381,7 +383,7 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         inp = gr.Textbox(label="Instruction (e.g., 'design, test and render a spooky maze level with 1 enemy and 2 traps')")
-        out = gr.Image(label="Generated Maze")
+        out = gr.Image(label="Generated Maze", type="filepath")
 
     with gr.Row():
         btn = gr.Button("🎮 Generate Maze")