commit

app.py

@@ -239,6 +239,9 @@ def render_grid_image(grid: List[List[str]]) -> str:
     import os
     from matplotlib.colors import ListedColormap
 
+    # Ensure the static folder exists
+    os.makedirs("static", exist_ok=True)
+
     # Mapping from tile type to index
     tile_map = {"P": 0, "W": 1, "E": 2, "T": 3, "S": 4, "G": 5}
     colors = [
@@ -253,9 +256,10 @@ 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
-    image_path = "/tmp/level.png"
+    # Save path
+    image_path = "static/level.png"
 
+    # Render image
     fig, ax = plt.subplots(figsize=(6, 6))
     ax.imshow(numeric_grid, cmap=color_map, interpolation='none')
 
@@ -265,16 +269,16 @@ def render_grid_image(grid: List[List[str]]) -> str:
     ax.grid(which='minor', color='black', linewidth=0.5)
     ax.tick_params(which='both', bottom=False, left=False, labelbottom=False, labelleft=False)
 
-    # Add emoji/text labels for key tiles
+    # Emoji/text labels for key tiles
     for i in range(len(grid)):
         for j in range(len(grid[0])):
             cell = grid[i][j]
             if cell in ["S", "G", "E", "T"]:
                 label = {
-                    "S": "S",  # or "🏁"
-                    "G": "G",  # or "🎯"
-                    "E": "⚠️",  # or "E"
-                    "T": "💣",  # or "T"
+                    "S": "S",    # Start
+                    "G": "G",    # Goal
+                    "E": "⚠️",   # Enemy
+                    "T": "💣",   # Trap
                 }.get(cell, "")
                 ax.text(j, i, label, ha='center', va='center', fontsize=12, color='black')
 
@@ -282,7 +286,7 @@ def render_grid_image(grid: List[List[str]]) -> str:
     plt.savefig(image_path, bbox_inches="tight", dpi=150)
     plt.close()
 
-    return image_path  # ✅ Return file path to be used in gr.Image(type="filepath")
+    return image_path  # ✅ Serve this in gr.Image(type="filepath")
 
 
 
@@ -294,7 +298,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: /tmp/level.png"
+        "- Example: static/level.png"
     ),
     name="render_image_agent"
 )
@@ -324,24 +328,21 @@ def run_streamed_supervisor(user_input):
             {"role": "user", "content": user_input}
         ]
     }):
-        pass  # just get the final chunk
+        pass  # Wait for final chunk
 
     final_messages = chunk["supervisor"]["messages"]
 
-    # Search the last messages for image path like /tmp/level.png
+    # Look for static image path in last messages
     for message in reversed(final_messages):
         if hasattr(message, "content") and isinstance(message.content, str):
-            match = re.search(r"(/tmp/level\.png)", message.content)
+            match = re.search(r"(static/level\.png)", message.content)
             if match:
                 image_path = match.group(1)
                 break
 
-    # ✅ Copy image to static/ and return filepath
+    # ✅ Return static file path directly (already saved by render_grid_image)
     if image_path and os.path.exists(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  # Gradio will use it directly
 
     return "❌ Image path not found or file does not exist."