Spaces:

Sachin5112
/

Automation

Paused

App Files Files Community

Sachin5112 commited on Apr 28

Commit

41903df

verified ·

1 Parent(s): eda9ebd

Update app.py

Browse files

Files changed (1) hide show

app.py +93 -52

app.py CHANGED Viewed

@@ -5,22 +5,27 @@ import time
 import threading
 from gradio_client import Client
-# Configuration
-SPACE_ID = "Sachin5112/Bihgfgh"
-STORAGE_PATH = "/data/evolution_state.json" if os.path.exists("/data") else "evolution_state.json"
 class ArchitectState:
     def __init__(self):
-        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>Evolution loop starting...</p></div></body></html>"""
         self.gen = 0
         self.is_running = False
         self.logs = []
         self.load()
     def save(self):
-        data = {"code": self.code, "gen": self.gen, "is_running": self.is_running}
-        with open(STORAGE_PATH, "w") as f:
-            json.dump(data, f)
     def load(self):
         if os.path.exists(STORAGE_PATH):
@@ -30,15 +35,16 @@ class ArchitectState:
                     self.code = data.get("code", self.code)
                     self.gen = data.get("gen", self.gen)
                     self.is_running = data.get("is_running", False)
-            except:
-                pass
     def add_log(self, text):
         timestamp = time.strftime("%H:%M:%S")
         self.logs.append(f"[{timestamp}] {text}")
-        if len(self.logs) > 20: self.logs.pop(0)
 state = ArchitectState()
 def extract_html(text):
     start = text.lower().find("<html")
@@ -47,77 +53,112 @@ def extract_html(text):
         return text[start:end+7]
     return text
-def evolution_loop():
     while True:
         if state.is_running:
             try:
-                state.add_log(f"Starting Gen {state.gen + 1}...")
                 client = Client(SPACE_ID)
-                prompt = f"Task: Improved 3D Minecraft Mobile Clone using Three.js. Gen: {state.gen+1}. Output ONLY full <html> code. Current: {state.code[:500]}..."
-                # Connect to the AI Space
-                result = client.predict(
-                    message=prompt,
-                    api_name="/chat"
                 )
-                raw_response = result if isinstance(result, str) else result[0]
-                new_code = extract_html(raw_response)
-                if len(new_code) > 100:
-                    state.code = new_code
                     state.gen += 1
-                    state.add_log(f"Gen {state.gen} success. Saved to persistent storage.")
                     state.save()
                 else:
-                    state.add_log("Error: AI returned invalid code.")
             except Exception as e:
-                state.add_log(f"Loop Error: {str(e)}")
-            time.sleep(15) # Safety pause
         else:
             time.sleep(2)
-# Start background thread
-thread = threading.Thread(target=evolution_loop, daemon=True)
-thread.start()
-# Gradio Interface
-with gr.Blocks(theme=gr.themes.Soft(), css="footer {display:none !important}") as demo:
-    gr.Markdown("# 🏗️ ARCHITECT V15 - BACKGROUND EVOLUTION")
     with gr.Row():
         with gr.Column(scale=3):
             preview = gr.HTML(value=state.code)
-            status_label = gr.Markdown(f"**Current Gen:** {state.gen}")
-        with gr.Column(scale=1):
-            run_btn = gr.Button("START EVOLUTION", variant="primary")
-            stop_btn = gr.Button("STOP")
-            log_display = gr.Textbox(label="System Logs", lines=10, value="\n".join(state.logs), interactive=False)
-    with gr.Accordion("View Source Code", open=False):
-        code_view = gr.Code(value=state.code, language="html", interactive=False)
-    def start_loop():
         state.is_running = True
         state.save()
-        return gr.update(value="### 🟢 STATUS: RUNNING"), "\n".join(state.logs)
-    def stop_loop():
         state.is_running = False
         state.save()
-        return gr.update(value="### 🔴 STATUS: STOPPED"), "\n".join(state.logs)
-    def update_ui():
-        return state.code, f"**Current Gen:** {state.gen}", "\n".join(state.logs), state.code
-    run_btn.click(start_loop, None, [status_label, log_display])
-    stop_btn.click(stop_loop, None, [status_label, log_display])
-    # Auto-refresh UI every 5 seconds
-    timer = gr.Timer(5)
-    timer.tick(update_ui, None, [preview, status_label, log_display, code_view])
 demo.launch()

 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"
 class ArchitectState:
     def __init__(self):
+        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()
     def save(self):
+        try:
+            data = {"code": self.code, "gen": self.gen, "is_running": self.is_running}
+            with open(STORAGE_PATH, "w") as f:
+                json.dump(data, f)
+        except Exception as e:
+            print(f"Save error: {e}")
     def load(self):
         if os.path.exists(STORAGE_PATH):
                     self.code = data.get("code", self.code)
                     self.gen = data.get("gen", self.gen)
                     self.is_running = data.get("is_running", False)
+            except Exception as e:
+                print(f"Load error: {e}")
     def add_log(self, text):
         timestamp = time.strftime("%H:%M:%S")
         self.logs.append(f"[{timestamp}] {text}")
+        if len(self.logs) > 15: self.logs.pop(0)
 state = ArchitectState()
+SPACE_ID = "Sachin5112/Bihgfgh"
 def extract_html(text):
     start = text.lower().find("<html")
         return text[start:end+7]
     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 = (
+                    f"Task: Improve the 3D Minecraft Mobile Clone using Three.js. "
+                    f"Current Generation: {state.gen}. "
+                    f"STRICT: Output ONLY the full <html> source code. No conversational text. "
+                    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
         state.save()
+        return "### STATUS: RUNNING", "\n".join(state.logs)
+    def toggle_stop():
         state.is_running = False
         state.save()
+        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,
+            state.code
+        )
+    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])
 demo.launch()