Create app.py

app.py

@@ -0,0 +1,161 @@
+import gradio as gr
+import json
+import boto3
+import uuid
+import re
+
+# -----------------------------
+# GLOBAL STATE HELPERS
+# -----------------------------
+
+def new_block(block_type):
+    return {
+        "id": str(uuid.uuid4())[:8],
+        "type": block_type,
+        "var": f"{block_type}_{str(uuid.uuid4())[:4]}",
+        "title": block_type.upper(),
+        "value": "",
+        "prompt": "",
+        "system": "",
+        "output": "",
+        "options": [],
+        "min": 0,
+        "max": 100,
+        "step": 1,
+    }
+
+def resolve_vars(text, variables):
+    def repl(match):
+        key = match.group(1)
+        return variables.get(key, f"{{{{{key}}}}}")
+    return re.sub(r"\{\{(\w+)\}\}", repl, text or "")
+
+# -----------------------------
+# AI CALL
+# -----------------------------
+
+def call_claude(prompt, system):
+    client = boto3.client("bedrock-runtime", region_name="us-east-1")
+
+    body = json.dumps({
+        "max_tokens": 1000,
+        "system": system,
+        "messages": [{"role": "user", "content": prompt}]
+    })
+
+    response = client.invoke_model(
+        modelId="anthropic.claude-3-sonnet-20240229-v1:0",
+        body=body
+    )
+
+    result = json.loads(response["body"].read())
+    return result["content"][0]["text"]
+
+# -----------------------------
+# BLOCK EXECUTION ENGINE
+# -----------------------------
+
+def run_blocks(blocks):
+    variables = {}
+    logs = []
+
+    for b in blocks:
+        try:
+            if b["type"] == "text-input":
+                variables[b["var"]] = b["value"]
+
+            elif b["type"] == "select-input":
+                variables[b["var"]] = b["value"]
+
+            elif b["type"] == "slider-input":
+                variables[b["var"]] = str(b["value"])
+
+            elif b["type"] in ["ai-text", "ai-transform"]:
+                prompt = resolve_vars(b["prompt"], variables)
+                system = b["system"] or "You are a helpful assistant."
+                output = call_claude(prompt, system)
+
+                b["output"] = output
+                variables[b["var"]] = output
+                logs.append(f"✅ {b['title']} ran")
+
+            elif b["type"] == "display":
+                ref = b.get("ref")
+                b["output"] = variables.get(ref, "")
+
+        except Exception as e:
+            logs.append(f"❌ {b['title']} failed: {str(e)}")
+            b["output"] = f"Error: {str(e)}"
+
+    return blocks, variables, "\n".join(logs)
+
+# -----------------------------
+# UI HELPERS
+# -----------------------------
+
+def add_block(blocks, block_type):
+    blocks.append(new_block(block_type))
+    return blocks, blocks
+
+def update_block(blocks, idx, field, value):
+    blocks[idx][field] = value
+    return blocks
+
+# -----------------------------
+# UI RENDERER
+# -----------------------------
+
+def render_blocks(blocks):
+    ui = []
+
+    for i, b in enumerate(blocks):
+        with gr.Group():
+            gr.Markdown(f"### {b['title']} ({b['type']})")
+
+            var = gr.Textbox(value=b["var"], label="Variable")
+            val = gr.Textbox(value=b["value"], label="Value")
+
+            prompt = gr.Textbox(value=b["prompt"], label="Prompt")
+            system = gr.Textbox(value=b["system"], label="System Prompt")
+
+            out = gr.Textbox(value=b.get("output", ""), label="Output")
+
+            var.change(lambda v, blocks=blocks, i=i: update_block(blocks, i, "var", v), var, None)
+            val.change(lambda v, blocks=blocks, i=i: update_block(blocks, i, "value", v), val, None)
+            prompt.change(lambda v, blocks=blocks, i=i: update_block(blocks, i, "prompt", v), prompt, None)
+            system.change(lambda v, blocks=blocks, i=i: update_block(blocks, i, "system", v), system, None)
+
+        ui.append(gr.Markdown("---"))
+
+    return ui
+
+# -----------------------------
+# MAIN UI
+# -----------------------------
+
+with gr.Blocks(title="Blocksmith AI") as demo:
+
+    blocks_state = gr.State([])
+
+    gr.Markdown("# ⚡ Blocksmith (Gradio Version)")
+
+    with gr.Row():
+        add_text = gr.Button("➕ Text Input")
+        add_ai = gr.Button("➕ AI Generate")
+        add_display = gr.Button("➕ Display")
+
+    blocks_ui = gr.Column()
+
+    run_btn = gr.Button("▶ Run All")
+    output_vars = gr.JSON(label="Variables")
+    logs = gr.Textbox(label="Run Logs")
+
+    # ADD BLOCKS
+    add_text.click(lambda b: add_block(b, "text-input"), blocks_state, [blocks_state, blocks_ui])
+    add_ai.click(lambda b: add_block(b, "ai-text"), blocks_state, [blocks_state, blocks_ui])
+    add_display.click(lambda b: add_block(b, "display"), blocks_state, [blocks_state, blocks_ui])
+
+    # RUN
+    run_btn.click(run_blocks, blocks_state, [blocks_state, output_vars, logs])
+
+demo.launch()