Update app.py

app.py

@@ -1,84 +1,293 @@
-import streamlit as st
-import re
-from pygmyclaw import PygmyClaw
+#!/usr/bin/env python3
+"""
+PygmyClaw Dashboard UI
+Shows queue, running jobs, results and allows safe code execution.
+"""
+
+import gradio as gr
+import json
+import time
+import traceback
+from pathlib import Path
 
-st.set_page_config(page_title="PygmyClaw AI", layout="wide")
+from pygmyclaw import PygmyClaw
 
-st.title("🧠 PygmyClaw AI Agent")
+# ---------------------------------------------------------
+# Initialize Agent
 
 agent = PygmyClaw()
 
-# -------------------
-# Prompt Input
-# -------------------
+# ---------------------------------------------------------
+# Storage
+
+WORKSPACE = Path("./workspace")
+WORKSPACE.mkdir(exist_ok=True)
+
+TASK_FILE = WORKSPACE / "tasks.json"
+
+
+def load_tasks():
+    if TASK_FILE.exists():
+        return json.loads(TASK_FILE.read_text())
+    return {}
+
+
+def save_tasks(tasks):
+    TASK_FILE.write_text(json.dumps(tasks, indent=2))
+
+
+tasks = load_tasks()
+
+# ---------------------------------------------------------
+# Queue system
+
+queue = []
+running_task = None
+
+
+# ---------------------------------------------------------
+# Submit Prompt
+
+
+def submit_prompt(prompt, tool):
+
+    global queue
+
+    task_id = str(time.time())
+
+    task = {
+        "task_id": task_id,
+        "prompt": prompt,
+        "tool": tool,
+        "status": "queued",
+        "code": "",
+        "explanation": "",
+        "output": "",
+        "timestamp": time.time()
+    }
+
+    tasks[task_id] = task
+    queue.append(task_id)
+
+    save_tasks(tasks)
+
+    return f"Task {task_id} added to queue."
+
+
+# ---------------------------------------------------------
+# Process Queue
+
+
+def process_queue():
+
+    global running_task
+
+    if running_task:
+        return
+
+    if not queue:
+        return
+
+    task_id = queue.pop(0)
+    running_task = task_id
+
+    task = tasks[task_id]
+    task["status"] = "running"
+
+    try:
+
+        result = agent.run(task["prompt"], tool=task["tool"])
+
+        task["code"] = result.get("code", "")
+        task["explanation"] = result.get("explanation", "")
+        task["status"] = "completed"
+
+    except Exception as e:
+
+        task["status"] = "failed"
+        task["output"] = str(e)
+
+    running_task = None
+    save_tasks(tasks)
+
+
+# ---------------------------------------------------------
+# Run Generated Code
+
+
+def run_code(task_id):
+
+    task = tasks.get(task_id)
+
+    if not task:
+        return "Task not found"
+
+    code = task.get("code")
+
+    if not code:
+        return "No code to run"
+
+    try:
+
+        local_env = {}
+
+        exec(code, {}, local_env)
+
+        output = str(local_env)
+
+        task["output"] = output
+        save_tasks(tasks)
+
+        return output
+
+    except Exception:
+
+        err = traceback.format_exc()
+        task["output"] = err
+        save_tasks(tasks)
+
+        return err
+
+
+# ---------------------------------------------------------
+# Dashboard Data
+
+
+def get_dashboard():
+
+    process_queue()
+
+    queue_view = []
+    completed_view = []
+
+    for tid, t in tasks.items():
+
+        row = [
+            tid,
+            t["prompt"],
+            t["status"],
+            t["tool"]
+        ]
+
+        if t["status"] in ["queued", "running"]:
+            queue_view.append(row)
+        else:
+            completed_view.append(row)
+
+    running = running_task if running_task else "None"
+
+    return queue_view, running, completed_view
+
+
+# ---------------------------------------------------------
+# Task Details
+
+
+def get_task_details(task_id):
+
+    task = tasks.get(task_id)
+
+    if not task:
+        return "", "", ""
+
+    return task["explanation"], task["code"], task["output"]
+
+
+# ---------------------------------------------------------
+# UI
+
+
+with gr.Blocks(title="PygmyClaw AI Dashboard") as demo:
+
+    gr.Markdown("# 🧠 PygmyClaw AI Agent Dashboard")
+
+    # -------------------------------------------------
+    # Prompt
+
+    with gr.Row():
 
-prompt_text = st.text_area("Enter Prompt")
+        prompt = gr.Textbox(
+            label="Prompt",
+            placeholder="Write a python function to add numbers..."
+        )
 
-tool_list = ["AI Agent"] + list(agent.tools_data.keys())
+        tool = gr.Dropdown(
+            choices=["AI Agent", "sys_info", "echo"],
+            value="AI Agent",
+            label="Tool"
+        )
 
-selected_tool = st.selectbox("Select Tool", tool_list)
+        submit = gr.Button("Submit Task")
 
-# -------------------
-# Generate
-# -------------------
 
-if st.button("Generate Code"):
+    submit_status = gr.Textbox(label="Status")
 
-    result = agent.submit_prompt(prompt_text, selected_tool)
+    submit.click(
+        submit_prompt,
+        inputs=[prompt, tool],
+        outputs=submit_status
+    )
 
-    st.markdown("### AI Response")
-    st.markdown(result)
+    # -------------------------------------------------
+    # Queue
 
-    match = re.search(r"```python(.*?)```", result, re.S)
+    gr.Markdown("## Queue")
 
-    if match:
+    queue_table = gr.Dataframe(
+        headers=["Task ID", "Prompt", "Status", "Tool"],
+        interactive=False
+    )
 
-        code = match.group(1).strip()
+    running_box = gr.Textbox(label="Running Task")
 
-        st.markdown("### Extracted Code")
-        st.code(code, language="python")
+    refresh = gr.Button("Refresh Dashboard")
 
-        if st.button("Run Code"):
+    # -------------------------------------------------
+    # Completed
 
-            try:
+    gr.Markdown("## Completed Jobs")
 
-                local_vars = {}
-                exec(code, {}, local_vars)
+    completed_table = gr.Dataframe(
+        headers=["Task ID", "Prompt", "Status", "Tool"],
+        interactive=False
+    )
 
-                st.success("Execution Result")
-                st.write(local_vars)
+    refresh.click(
+        get_dashboard,
+        outputs=[queue_table, running_box, completed_table]
+    )
 
-            except Exception as e:
+    # -------------------------------------------------
+    # Task Details
 
-                st.error(str(e))
+    gr.Markdown("## Task Details")
 
+    task_id_input = gr.Textbox(label="Enter Task ID")
 
-# -------------------
-# Tools Section
-# -------------------
+    explanation = gr.Textbox(label="Explanation", lines=6)
 
-st.divider()
-st.header("🛠 Tools")
+    code = gr.Code(label="Generated Code", language="python")
 
-for name, tool in agent.tools_data.items():
+    output = gr.Textbox(label="Execution Output", lines=6)
 
-    st.subheader(name)
-    st.write(tool["description"])
-    st.code(tool["code"], language="python")
+    load_task = gr.Button("Load Task")
 
-# -------------------
-# Create Tool
-# -------------------
+    load_task.click(
+        get_task_details,
+        inputs=task_id_input,
+        outputs=[explanation, code, output]
+    )
 
-st.divider()
-st.header("Create Tool")
+    # -------------------------------------------------
+    # Run Code
 
-tool_name = st.text_input("Tool Name")
-tool_desc = st.text_area("Description")
-tool_code = st.text_area("Python Code")
+    run = gr.Button("Run Code")
 
-if st.button("Save Tool"):
+    run.click(
+        run_code,
+        inputs=task_id_input,
+        outputs=output
+    )
 
-    agent.add_tool(tool_name, tool_desc, tool_code)
 
-    st.success("Tool saved and uploaded to HF")
+demo.launch(server_name="0.0.0.0", server_port=7860)