Spaces:

Sachin5112
/

Automation

Paused

App Files Files Community

Sachin5112 commited on Apr 28

Commit

fd35eee

verified ·

1 Parent(s): 41903df

Update app.py

Browse files

Files changed (1) hide show

app.py +35 -42

app.py CHANGED Viewed

@@ -6,7 +6,6 @@ import threading
 from gradio_client import Client
 # Configuration for HF Persistent Storage
-# The /data directory is the standard mount point for HF persistent buckets
 STORAGE_DIR = "/data"
 STORAGE_PATH = os.path.join(STORAGE_DIR, "evolution_state.json") if os.path.exists(STORAGE_DIR) else "evolution_state.json"
@@ -15,7 +14,7 @@ class ArchitectState:
         self.code = """<!DOCTYPE html><html><body style="background:#000;color:#555;display:flex;align-items:center;justify-content:center;height:100vh;margin:0;font-family:sans-serif;"><div><h1>SYSTEM READY</h1><p>Waiting for evolution loop to start...</p></div></body></html>"""
         self.gen = 0
         self.is_running = False
-        self.current_stream = "" # For live typing effect
         self.logs = []
         self.load()
@@ -54,11 +53,11 @@ def extract_html(text):
     return text
 def evolution_worker():
-    """Background thread that runs forever, streaming tokens and saving to disk."""
     while True:
         if state.is_running:
             try:
-                state.add_log(f"Gen {state.gen + 1}: Connecting to {SPACE_ID}...")
                 client = Client(SPACE_ID)
                 prompt = (
@@ -68,69 +67,67 @@ def evolution_worker():
                     f"Current Code: {state.code[:1000]}..."
                 )
-                # Use submit for streaming
                 job = client.submit(message=prompt, api_name="/chat")
-                temp_output = ""
-                state.add_log("Streaming started...")
-                while not job.done():
-                    # Get current updates from the stream
-                    updates = job.communicator.get_updates()
-                    if updates:
-                        # Grab the latest data chunk
-                        latest_data = updates[-1].data
-                        if latest_data and len(latest_data) > 0:
-                            temp_output = latest_data[0]
-                            # Update global state live so UI can pick it up
-                            state.current_stream = temp_output
-                    time.sleep(0.1)
-                # Finalize after stream ends
-                final_raw = job.outputs()[-1][0] if job.outputs() else temp_output
                 processed_code = extract_html(final_raw)
                 if len(processed_code) > 200:
                     state.code = processed_code
                     state.gen += 1
-                    state.current_stream = "" # Clear stream
                     state.add_log(f"Success! Gen {state.gen} saved to bucket.")
                     state.save()
                 else:
-                    state.add_log("Warning: Received code too short. Retrying...")
             except Exception as e:
                 state.add_log(f"Stream Error: {str(e)}")
-            # Wait 10 seconds before starting next gen
-            time.sleep(10)
         else:
             time.sleep(2)
-# Start the worker thread once
 worker = threading.Thread(target=evolution_worker, daemon=True)
 worker.start()
-# --- UI Setup ---
 with gr.Blocks(theme=gr.themes.Monochrome(), css=".log-box textarea { font-family: monospace; font-size: 11px; }") as demo:
-    gr.Markdown("# 🧱 ARCHITECT V16 - BACKGROUND STREAMER")
     with gr.Row():
         with gr.Column(scale=3):
-            # The Live Preview
             preview = gr.HTML(value=state.code)
         with gr.Column(scale=1):
             status = gr.Markdown(f"### STATUS: {'RUNNING' if state.is_running else 'STOPPED'}\n**Generation:** {state.gen}")
-            start_btn = gr.Button("🚀 START LOOP", variant="primary")
-            stop_btn = gr.Button("🛑 STOP LOOP")
-            logs = gr.Textbox(label="Live Logs", value="\n".join(state.logs), lines=8, interactive=False, elem_classes="log-box")
     with gr.Tabs():
-        with gr.Tab("Live Stream (Raw)"):
-            live_code = gr.Code(label="Streaming View", value=state.current_stream, language="html", interactive=False)
-        with gr.Tab("Last Success (Full)"):
-            full_code = gr.Code(label="Stable Code", value=state.code, language="html", interactive=False)
     def toggle_start():
         state.is_running = True
@@ -143,11 +140,8 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=".log-box textarea { font-famil
         return "### STATUS: STOPPED", "\n".join(state.logs)
     def refresh_ui():
-        # This function updates the UI with the latest state from the background thread
-        # If streaming is happening, show the stream. Otherwise show the saved code.
-        display_code = state.current_stream if state.current_stream else state.code
         return (
-            state.code, # update preview
             f"### STATUS: {'RUNNING' if state.is_running else 'STOPPED'}\n**Generation:** {state.gen}",
             "\n".join(state.logs),
             state.current_stream,
@@ -157,7 +151,6 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=".log-box textarea { font-famil
     start_btn.click(toggle_start, None, [status, logs])
     stop_btn.click(toggle_stop, None, [status, logs])
-    # Refresh every 2 seconds to show live typing
     timer = gr.Timer(2)
     timer.tick(refresh_ui, None, [preview, status, logs, live_code, full_code])

 from gradio_client import Client
 # Configuration for HF Persistent Storage
 STORAGE_DIR = "/data"
 STORAGE_PATH = os.path.join(STORAGE_DIR, "evolution_state.json") if os.path.exists(STORAGE_DIR) else "evolution_state.json"
         self.code = """<!DOCTYPE html><html><body style="background:#000;color:#555;display:flex;align-items:center;justify-content:center;height:100vh;margin:0;font-family:sans-serif;"><div><h1>SYSTEM READY</h1><p>Waiting for evolution loop to start...</p></div></body></html>"""
         self.gen = 0
         self.is_running = False
+        self.current_stream = ""
         self.logs = []
         self.load()
     return text
 def evolution_worker():
+    """Background thread using the corrected Gradio iterator for live streaming."""
     while True:
         if state.is_running:
             try:
+                state.add_log(f"Gen {state.gen + 1}: Connecting...")
                 client = Client(SPACE_ID)
                 prompt = (
                     f"Current Code: {state.code[:1000]}..."
                 )
+                # Use submit to get a job
                 job = client.submit(message=prompt, api_name="/chat")
+                state.add_log("Stream started...")
+                # The correct way to stream in recent Gradio versions is iterating over the job
+                for output in job:
+                    if not state.active_check(): break
+                    # Gradio chat usually returns a list or tuple; we want the string part
+                    if isinstance(output, (list, tuple)):
+                        chunk = output[0]
+                    else:
+                        chunk = output
+                    state.current_stream = chunk
+                # Final processing
+                final_raw = state.current_stream
                 processed_code = extract_html(final_raw)
                 if len(processed_code) > 200:
                     state.code = processed_code
                     state.gen += 1
+                    state.current_stream = ""
                     state.add_log(f"Success! Gen {state.gen} saved to bucket.")
                     state.save()
                 else:
+                    state.add_log("Warning: Invalid code received. Retrying...")
             except Exception as e:
                 state.add_log(f"Stream Error: {str(e)}")
+            time.sleep(12) # Cooldown
         else:
             time.sleep(2)
+state.active_check = lambda: state.is_running
+# Start worker
 worker = threading.Thread(target=evolution_worker, daemon=True)
 worker.start()
+# --- UI ---
 with gr.Blocks(theme=gr.themes.Monochrome(), css=".log-box textarea { font-family: monospace; font-size: 11px; }") as demo:
+    gr.Markdown("# 🧱 ARCHITECT V17 - FIXED STREAMER")
     with gr.Row():
         with gr.Column(scale=3):
             preview = gr.HTML(value=state.code)
         with gr.Column(scale=1):
             status = gr.Markdown(f"### STATUS: {'RUNNING' if state.is_running else 'STOPPED'}\n**Generation:** {state.gen}")
+            start_btn = gr.Button("🚀 START", variant="primary")
+            stop_btn = gr.Button("🛑 STOP")
+            logs = gr.Textbox(label="Logs", value="\n".join(state.logs), lines=8, interactive=False, elem_classes="log-box")
     with gr.Tabs():
+        with gr.Tab("Live Stream"):
+            live_code = gr.Code(label="Streaming...", value=state.current_stream, language="html", interactive=False)
+        with gr.Tab("Last Stable"):
+            full_code = gr.Code(label="Current Saved Code", value=state.code, language="html", interactive=False)
     def toggle_start():
         state.is_running = True
         return "### STATUS: STOPPED", "\n".join(state.logs)
     def refresh_ui():
         return (
+            state.code,
             f"### STATUS: {'RUNNING' if state.is_running else 'STOPPED'}\n**Generation:** {state.gen}",
             "\n".join(state.logs),
             state.current_stream,
     start_btn.click(toggle_start, None, [status, logs])
     stop_btn.click(toggle_stop, None, [status, logs])
     timer = gr.Timer(2)
     timer.tick(refresh_ui, None, [preview, status, logs, live_code, full_code])