Update app.py

app.py

@@ -1,69 +1,94 @@
-# app.py
 import gradio as gr
 import matplotlib.pyplot as plt
 import pandas as pd
+import numpy as np
 import io
 from evo_transformer import EvoTransformer
 
-def run_evolution(generations):
-    evo = EvoTransformer()
-    evo.evolve(generations)
+# Global instance (resettable)
+evo = EvoTransformer()
 
-    history = evo.get_history()
-    final_eval = evo.evaluate()
-    
-    df = pd.DataFrame(history)
-    df_display = df.copy()
-    df_display["memory"] = df_display["memory"].apply(lambda x: "Enabled" if x else "Disabled")
-
-    # Radar chart
-    last_config = history[-1]
-    traits = ["layers", "attention_heads", "ffn_dim", "dropout", "memory"]
-    values = [last_config["layers"], last_config["attention_heads"],
-              last_config["ffn_dim"]/100, last_config["dropout"]*10, int(last_config["memory"])*10]
-    
-    fig, ax = plt.subplots(figsize=(6,6), subplot_kw=dict(polar=True))
-    angles = [n / float(len(traits)) * 2 * 3.14159 for n in range(len(traits))]
+# === Visualization Functions ===
+def plot_radar(config):
+    labels = ["Layers", "Attention Heads", "FFN Dim", "Dropout", "Memory"]
+    values = [
+        config["layers"],
+        config["attention_heads"],
+        config["ffn_dim"],
+        int(config["dropout"] * 100),
+        int(config["memory"])
+    ]
+    angles = np.linspace(0, 2 * np.pi, len(labels), endpoint=False).tolist()
     values += values[:1]
     angles += angles[:1]
-    ax.plot(angles, values, linewidth=2)
-    ax.fill(angles, values, alpha=0.3)
-    ax.set_xticks(angles[:-1])
-    ax.set_xticklabels(traits)
-    ax.set_title("Final Architecture Traits", size=15)
 
-    # File downloads
-    csv_file = io.StringIO(evo.export_csv())
-    json_file = io.StringIO(evo.export_json())
+    fig, ax = plt.subplots(figsize=(5, 5), subplot_kw=dict(polar=True))
+    ax.plot(angles, values, "o-", linewidth=2)
+    ax.fill(angles, values, alpha=0.25)
+    ax.set_thetagrids(np.degrees(angles[:-1]), labels)
+    ax.set_title("Final Architecture (Radar Chart)")
+    return fig
+
+def evolve_and_display(generations):
+    global evo
+    evo = EvoTransformer()  # Reset model
+    evo.evolve(generations)
+
+    df = evo.get_history_df()
+    final_config = evo.get_final_config()
+    accuracy, params = evo.evaluate()
+
+    fig = plot_radar(final_config)
+
+    json_file = io.BytesIO()
+    json_file.write(evo.get_history_json().encode("utf-8"))
+    json_file.seek(0)
+
+    csv_file = io.BytesIO()
+    df.to_csv(csv_file, index=False)
+    csv_file.seek(0)
 
     return (
-        df_display,
-        final_eval["accuracy"],
-        final_eval["params"],
+        accuracy,
+        params,
+        gr.Tabs.update(visible=True),
+        [gr.Textbox.update(value=str(row)) for _, row in df.iterrows()],
         fig,
-        ("evo_history.csv", csv_file.getvalue()),
-        ("evo_history.json", json_file.getvalue())
+        (csv_file, "evo_history.csv"),
+        (json_file, "evo_history.json"),
     )
 
-with gr.Blocks(title="EvoTransformer Demo") as demo:
-    gr.Markdown("# 🧬 EvoTransformer Live Demo")
-    gr.Markdown("This demo evolves a Transformer architecture and displays how traits change over generations.")
+# === Gradio UI ===
+with gr.Blocks(title="EvoTransformer Live Demo") as demo:
+    gr.Markdown(
+        "🚀 **EvoTransformer Live Demo**\n\n"
+        "This demo evolves a Transformer architecture and displays how traits change over generations."
+    )
 
     with gr.Row():
-        generations = gr.Slider(1, 10, value=5, step=1, label="Generations")
-        run_btn = gr.Button("Evolve Now 🚀")
+        generations = gr.Slider(1, 10, value=5, label="Generations")
+        evolve_btn = gr.Button("Evolve Now 🚀")
 
     with gr.Row():
-        acc = gr.Number(label="Estimated Accuracy")
-        params = gr.Number(label="Estimated Params (M)")
+        accuracy_out = gr.Number(label="Estimated Accuracy", value=0)
+        params_out = gr.Number(label="Estimated Params (M)", value=0)
 
-    table = gr.Dataframe(label="Evolution History", wrap=True)
-    plot = gr.Plot(label="Final Architecture (Radar Chart)")
+    tabbox = gr.Tabs(visible=False)
+    with tabbox:
+        with gr.Tab(label="Evolution History"):
+            history_display = [gr.Textbox(label=str(i+1), interactive=False) for i in range(10)]
+        with gr.Tab(label="Radar View"):
+            radar_plot = gr.Plot()
 
     with gr.Row():
-        csv_dl = gr.File(label="Download CSV", file_types=[".csv"], interactive=True)
-        json_dl = gr.File(label="Download JSON", file_types=[".json"], interactive=True)
+        csv_btn = gr.File(label="Download CSV")
+        json_btn = gr.File(label="Download JSON")
 
-    run_btn.click(fn=run_evolution, inputs=generations, outputs=[table, acc, params, plot, csv_dl, json_dl])
+    evolve_btn.click(
+        evolve_and_display,
+        inputs=[generations],
+        outputs=[accuracy_out, params_out, tabbox, history_display, radar_plot, csv_btn, json_btn]
+    )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()