Update app.py

app.py

@@ -5,7 +5,7 @@ import boto3
 import re
 
 # -----------------------------
-# STATE HELPERS
+# HELPERS
 # -----------------------------
 
 def create_block(block_type):
@@ -21,7 +21,11 @@ def create_block(block_type):
     }
 
 def resolve_vars(text, variables):
-    return re.sub(r"\{\{(\w+)\}\}", lambda m: variables.get(m.group(1), m.group(0)), text or "")
+    return re.sub(
+        r"\{\{(\w+)\}\}",
+        lambda m: variables.get(m.group(1), m.group(0)),
+        text or ""
+    )
 
 # -----------------------------
 # AI CALL
@@ -45,7 +49,7 @@ def call_claude(prompt, system):
     return result["content"][0]["text"]
 
 # -----------------------------
-# CORE ENGINE
+# ENGINE
 # -----------------------------
 
 def run_flow(blocks):
@@ -73,47 +77,62 @@ def run_flow(blocks):
     return blocks, variables, "\n".join(logs)
 
 # -----------------------------
-# UI ACTIONS
+# ACTIONS
 # -----------------------------
 
 def add_block(blocks, block_type):
     blocks.append(create_block(block_type))
-    return blocks, blocks
 
-def select_block(blocks, idx):
-    return blocks, idx, blocks[idx]
+    choices = [
+        f"{i}: {b['title']}" for i, b in enumerate(blocks)
+    ]
+
+    return blocks, choices
+
+def load_block(blocks, selected):
+    if not selected:
+        return "", "", "", "", ""
+
+    idx = int(selected.split(":")[0])
+    b = blocks[idx]
+
+    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
+
+    idx = int(selected.split(":")[0])
+
+    blocks[idx].update({
+        "var": var,
+        "value": value,
+        "prompt": prompt,
+        "system": system
+    })
 
-def update_block(blocks, idx, data):
-    blocks[idx].update(data)
     return blocks
 
 # -----------------------------
-# MAIN UI
+# UI
 # -----------------------------
 
-with gr.Blocks(css="""
-body {background-color: #0f172a; color: white;}
-""") as demo:
+with gr.Blocks() as demo:
 
     blocks_state = gr.State([])
-    selected_idx = gr.State(None)
 
     gr.Markdown("# ⚡ Blocksmith AI")
 
     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")
 
@@ -123,68 +142,73 @@ body {background-color: #0f172a; color: white;}
                 interactive=True
             )
 
-        # -----------------------------
-        # ⚙️ INSPECTOR PANEL
-        # -----------------------------
+        # ---------------- INSPECTOR ----------------
         with gr.Column(scale=2):
             gr.Markdown("## ⚙️ Inspector")
 
             var = gr.Textbox(label="Variable")
             value = gr.Textbox(label="Value")
             prompt = gr.Textbox(label="Prompt")
-            system = gr.Textbox(label="System")
+            system = gr.Textbox(label="System Prompt")
             output = gr.Textbox(label="Output", interactive=False)
 
-    # -----------------------------
-    # ▶ RUN + LOGS
-    # -----------------------------
+    # ---------------- RUN ----------------
     run_btn = gr.Button("▶ Run Flow")
 
     variables_out = gr.JSON(label="Variables")
     logs = gr.Textbox(label="Logs")
 
-    # -----------------------------
-    # EVENTS
-    # -----------------------------
-
-    # Add blocks
-    add_text.click(lambda b: add_block(b, "text"), blocks_state, [blocks_state, block_list])
-    add_ai.click(lambda b: add_block(b, "ai"), blocks_state, [blocks_state, block_list])
-
-    # Update block list display
-    def refresh_list(blocks):
-        return [f"{i}: {b['title']}" for i, b in enumerate(blocks)]
-
-    blocks_state.change(refresh_list, blocks_state, block_list)
-
-    # Select block
-    def load_block(blocks, idx):
-        if idx is None:
-            return "", "", "", "", ""
-        b = blocks[int(idx.split(":")[0])]
-        return b["var"], b["value"], b["prompt"], b["system"], b["output"]
-
-    block_list.change(load_block, [blocks_state, block_list], [var, value, prompt, system, output])
-
-    # Update block fields
-    def save_block(blocks, idx, v, val, p, s):
-        if idx is None:
-            return blocks
-        i = int(idx.split(":")[0])
-        blocks[i].update({
-            "var": v,
-            "value": val,
-            "prompt": p,
-            "system": s
-        })
-        return blocks
+    # ---------------- EVENTS ----------------
 
-    var.change(save_block, [blocks_state, block_list, var, value, prompt, system], blocks_state)
-    value.change(save_block, [blocks_state, block_list, var, value, prompt, system], blocks_state)
-    prompt.change(save_block, [blocks_state, block_list, var, value, prompt, system], blocks_state)
-    system.change(save_block, [blocks_state, block_list, var, value, prompt, system], blocks_state)
+    add_text.click(
+        fn=lambda b: add_block(b, "text"),
+        inputs=blocks_state,
+        outputs=[blocks_state, block_list]
+    )
 
-    # Run flow
-    run_btn.click(run_flow, blocks_state, [blocks_state, variables_out, logs])
+    add_ai.click(
+        fn=lambda b: add_block(b, "ai"),
+        inputs=blocks_state,
+        outputs=[blocks_state, block_list]
+    )
+
+    block_list.change(
+        fn=load_block,
+        inputs=[blocks_state, block_list],
+        outputs=[var, value, prompt, system, output]
+    )
+
+    var.change(
+        fn=save_block,
+        inputs=[blocks_state, block_list, var, value, prompt, system],
+        outputs=blocks_state
+    )
+
+    value.change(
+        fn=save_block,
+        inputs=[blocks_state, block_list, var, value, prompt, system],
+        outputs=blocks_state
+    )
+
+    prompt.change(
+        fn=save_block,
+        inputs=[blocks_state, block_list, var, value, prompt, system],
+        outputs=blocks_state
+    )
+
+    system.change(
+        fn=save_block,
+        inputs=[blocks_state, block_list, var, value, prompt, system],
+        outputs=blocks_state
+    )
+
+    run_btn.click(
+        fn=run_flow,
+        inputs=blocks_state,
+        outputs=[blocks_state, variables_out, logs]
+    )
 
-demo.launch()
+# ✅ FIXED CSS placement (Gradio 6+)
+demo.launch(css="""
+body { background-color: #0f172a; color: white; }
+""")