Update app.py

app.py

@@ -2,9 +2,11 @@ import gradio as gr
 import numpy as np
 import pandas as pd
 import matplotlib.pyplot as plt
+import tempfile
+
 from minimal_self_full import MinimalSelf, MovingObstacle, SocialEntity, run_simulation, compute_phi
 
-plt.switch_backend("Agg")  # Ensure headless plotting
+plt.switch_backend("Agg")  # ensure headless plotting
 
 def run_agent(
     steps=500,
@@ -46,9 +48,10 @@ def run_agent(
     # Plot metrics
     fig1, axes = plt.subplots(4, 1, figsize=(10, 9), sharex=True)
     metrics = ["predictive_rate", "C_min", "body_bit_strength", "reward"]
+    colors = ["#2b8", "#06c", "#a5a", "#e67"]
     for i, m in enumerate(metrics):
         if m in df.columns:
-            axes[i].plot(df["t"], df[m], label=m, color=["#2b8","#06c","#a5a","#e67"][i])
+            axes[i].plot(df["t"], df[m], label=m, color=colors[i])
             axes[i].set_ylabel(m)
             axes[i].grid(True)
             axes[i].legend()
@@ -58,7 +61,7 @@ def run_agent(
     fig1.suptitle("Metrics over time")
     fig1.tight_layout()
 
-    # Plot path (agent + optional obstacle/social)
+    # Plot path
     fig2, ax = plt.subplots(figsize=(6, 6))
     ax.set_title("Agent and environment paths")
     ax.set_xlabel("X")
@@ -88,14 +91,16 @@ def run_agent(
 
     ax.legend()
 
-    # Prepare CSV
-    csv_bytes = df.to_csv(index=False).encode("utf-8")
+    # Save CSV to a temporary file and return path
+    tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".csv")
+    df.to_csv(tmp.name, index=False)
+    tmp.close()
 
     return (
         gr.Plot(fig1),
         gr.Plot(fig2),
         f"{final_phi:.2f}",
-        csv_bytes
+        tmp.name  # return path, not bytes
     )
 
 with gr.Blocks(title="RFT Minimal Self: 3×3 Agent") as demo:
@@ -117,7 +122,7 @@ with gr.Blocks(title="RFT Minimal Self: 3×3 Agent") as demo:
     metrics_plot = gr.Plot(label="Metrics over time")
     path_plot = gr.Plot(label="Paths in 3×3 world")
     final_phi = gr.Textbox(label="Final Φ_min (toy measure)", interactive=False)
-    csv_out = gr.File(label="Download results.csv", file_count="single")
+    csv_out = gr.File(label="Download results.csv", file_types=[".csv"])
 
     run_btn.click(
         fn=run_agent,