Update app.py

app.py

@@ -83,11 +83,12 @@ def run_flow(blocks):
 def add_block(blocks, block_type):
     blocks.append(create_block(block_type))
 
-    choices = [
-        f"{i}: {b['title']}" for i, b in enumerate(blocks)
-    ]
+    choices = [f"{i}: {b['title']}" for i, b in enumerate(blocks)]
+
+    selected = choices[-1] if choices else None
+
+    return blocks, choices, selected
 
-    return blocks, choices
 
 def load_block(blocks, selected):
     if not selected:
@@ -98,6 +99,7 @@ def load_block(blocks, selected):
 
     return b["var"], b["value"], b["prompt"], b["system"], b["output"]
 
+
 def save_block(blocks, selected, var, value, prompt, system):
     if not selected:
         return blocks
@@ -125,24 +127,24 @@ with gr.Blocks() as demo:
 
     with gr.Row():
 
-        # ---------------- SIDEBAR ----------------
+        # SIDEBAR
         with gr.Column(scale=1):
             gr.Markdown("## 🧩 Blocks")
-
             add_text = gr.Button("➕ Text Input")
             add_ai = gr.Button("➕ AI Block")
 
-        # ---------------- CANVAS ----------------
+        # CANVAS
         with gr.Column(scale=2):
             gr.Markdown("## 🧠 Canvas")
 
             block_list = gr.Radio(
                 choices=[],
+                value=None,
                 label="Blocks",
                 interactive=True
             )
 
-        # ---------------- INSPECTOR ----------------
+        # INSPECTOR
         with gr.Column(scale=2):
             gr.Markdown("## ⚙️ Inspector")
 
@@ -152,9 +154,8 @@ with gr.Blocks() as demo:
             system = gr.Textbox(label="System Prompt")
             output = gr.Textbox(label="Output", interactive=False)
 
-    # ---------------- RUN ----------------
+    # RUN
     run_btn = gr.Button("▶ Run Flow")
-
     variables_out = gr.JSON(label="Variables")
     logs = gr.Textbox(label="Logs")
 
@@ -163,13 +164,13 @@ with gr.Blocks() as demo:
     add_text.click(
         fn=lambda b: add_block(b, "text"),
         inputs=blocks_state,
-        outputs=[blocks_state, block_list]
+        outputs=[blocks_state, block_list, block_list]
     )
 
     add_ai.click(
         fn=lambda b: add_block(b, "ai"),
         inputs=blocks_state,
-        outputs=[blocks_state, block_list]
+        outputs=[blocks_state, block_list, block_list]
     )
 
     block_list.change(
@@ -208,7 +209,7 @@ with gr.Blocks() as demo:
         outputs=[blocks_state, variables_out, logs]
     )
 
-# ✅ FIXED CSS placement (Gradio 6+)
+# ✅ CSS FIX (Gradio 6)
 demo.launch(css="""
 body { background-color: #0f172a; color: white; }
 """)