Update app.py

app.py

@@ -2,19 +2,19 @@ import os
 import re
 import json
 import urllib.parse
+import time
 from pathlib import Path
 import gradio as gr
 from llama_cpp import Llama
 
 # ================== CONFIG ==================
-# Use a tiny 1.1B model for speed and memory
 MODEL_REPO = "TheBloke/TinyLlama-1.1B-Chat-v1.0-GGUF"
 GGUF_FILE = "tinyllama-1.1b-chat-v1.0.Q4_K_M.gguf"   # ~700 MB
 MAX_TURNS = 50
 HISTORY_WINDOW = 10             
 LEADERBOARD_FILE = Path("/tmp/leaderboard.json")
 
-# Initialize leaderboard
+# Initialize leaderboard safely
 if not LEADERBOARD_FILE.exists():
     LEADERBOARD_FILE.write_text(json.dumps([
         {"Rank": "1", "Genre": "Dark Fantasy", "Actions": 24, "Fate": "Devoured by Mimic"},
@@ -31,7 +31,7 @@ try:
         n_ctx=1024,
         n_threads=2,
         use_mmap=True,
-        chat_format="llama-2",       # suitable for TinyLlama
+        chat_format="llama-2",       
         verbose=False,
     )
     model_ready = True
@@ -52,8 +52,7 @@ def generate_response(messages, max_tokens=200, temperature=0.9):
             top_p=0.9,
         )
         raw_output = out["choices"][0]["message"]["content"].strip()
-        # Clean out any leftover Llama instruction tokens if they leak into text string
-        raw_output = re.sub(r"\\[/?INST\\]|\\[/?SYS\\]|<<SYS>>|<</SYS>>", "", raw_output).strip()
+        raw_output = re.sub(r"\[/?INST\]|\[/?SYS\]|<<SYS>>|<</SYS>>", "", raw_output).strip()
         return raw_output
     except Exception as e:
         print("LLM error:", e)
@@ -94,7 +93,6 @@ def new_game(selected_genre="Dark Fantasy"):
         "hp": 100,
         "genre": selected_genre,
         "status_msg": f"🎲 Setting up theme: {selected_genre}...",
-        "processing": False,
     }
 
 def _ctx(history, system, speaker):
@@ -119,23 +117,49 @@ def get_leaderboard_df():
     data = json.loads(LEADERBOARD_FILE.read_text())
     return [[r["Rank"], r["Genre"], r["Actions"], r["Fate"]] for r in data]
 
-def advance(state):
-    if not state or not state["running"] or state["turn"] >= MAX_TURNS * 2 or state["hp"] <= 0:
-        if state and state["hp"] <= 0 and state["running"]:
-            state["running"] = False
-            state["status_msg"] = "💀 GAME OVER! The player has died."
-            update_leaderboard(state["genre"], state["turn"], "Flatlined at 0 HP")
-        return state
+# ================== RENDER UTILITIES ==================
+def _dice(text):
+    text = re.sub(r"\[roll\s+d(\d+)\]", r'<span class="dice">🎲 d\1</span>', text)
+    text = re.sub(r"\[HP\s*([\+\-]\d+)\]", r'<span class="hp-change">💔 \1 HP</span>', text)
+    return text
 
-    if state.get("processing", False):
-        return state
-    state["processing"] = True
+def render(state):
+    if not state:
+        return []
+    msgs = []
+    for e in state["history"]:
+        role = "assistant" if e["role"] == "gm" else "user"
+        prefix = "🔮 **GM**: " if e["role"] == "gm" else "⚔️ **Player**: "
+        msgs.append({"role": role, "content": prefix + _dice(e["content"])})
+    return msgs
 
-    try:
+def make_share_link(state):
+    if not state or not state["history"]:
+        return gr.update(visible=False)
+    base_text = f"🍿 Survival Level: {state['turn']} actions inside the unhinged '{state['genre']}' AI simulation! Can you survive longer? 👇"
+    encoded_text = urllib.parse.quote(base_text)
+    encoded_url = urllib.parse.quote("https://huggingface.co")
+    twitter_url = f"https://twitter.com{encoded_text}&url={encoded_url}"
+    return gr.update(value=f'<a href="{twitter_url}" target="_blank" class="share-btn">🐦 Broadcast Score to X</a>', visible=True)
+
+# ================== MAIN GAME GENERATOR ENGINE ==================
+def run_simulation_loop(state):
+    """
+    Fixed native generation engine loop. 
+    Bypasses broken background clocks by yielding updates directly.
+    """
+    if not state or not state["running"]:
+        yield state, render(state), "⏸️ Simulation stopped.", state["hp"] if state else 100, get_leaderboard_df()
+        return
+
+    while state["running"] and state["turn"] < MAX_TURNS * 2 and state["hp"] > 0:
         hist = state["history"]
         genre_cfg = GENRES.get(state["genre"], GENRES["Dark Fantasy"])
 
         if state["next"] == "gm":
+            state["status_msg"] = "🔮 GM is writing the next scenario turn..."
+            yield state, render(state), state["status_msg"], state["hp"], get_leaderboard_df()
+            
             if not hist:
                 ctx = [{"role": "system", "content": genre_cfg["gm"]},
                        {"role": "user", "content": genre_cfg["start"]}]
@@ -146,7 +170,7 @@ def advance(state):
             response = generate_response(ctx, max_tokens=220)
             hist.append({"role": "gm", "content": response})
             
-            hp_match = re.search(r"\\[HP\\s*([\\+\\-])\\s*(\\d+)\\]", response)
+            hp_match = re.search(r"\[HP\s*([\+\-])\s*(\d+)\]", response)
             if hp_match:
                 op, val = hp_match.groups()
                 delta = int(val) if op == "+" else -int(val)
@@ -154,119 +178,37 @@ def advance(state):
                 
             state["next"] = "player"
         else:
+            state["status_msg"] = "⚔️ Player is formulating an action..."
+            yield state, render(state), state["status_msg"], state["hp"], get_leaderboard_df()
+
             if state["intervention"]:
                 player = state["intervention"]
                 state["intervention"] = None
+                state["status_msg"] = "🔮 Neural Hack Override Successful!"
             else:
                 ctx = _ctx(hist, genre_cfg["player"], "player")
                 ctx.append({"role": "user", "content": "Declare what action you take next."})
                 player = generate_response(ctx, max_tokens=180)
+                
             hist.append({"role": "player", "content": player})
             state["next"] = "gm"
 
         state["turn"] += 1
-    finally:
-        state["processing"] = False
-
-    return state
-
-# ================== RENDER UTILITIES ==================
-def _dice(text):
-    text = re.sub(r"\\[roll\\s+d(\\d+)\\]", r'<span class="dice">🎲 d\\1</span>', text)
-    text = re.sub(r"\\[HP\\s*([\\+\\-Outside]\\d+)\\]", r'<span class="hp-change">💔 \\1 HP</span>', text)
-    return text
-
-def render(state):
-    if not state:
-        return []
-    msgs = []
-    for e in state["history"]:
-        role = "assistant" if e["role"] == "gm" else "user"
-        prefix = "🔮 **GM**: " if e["role"] == "gm" else "⚔️ **Player**: "
-        msgs.append({"role": role, "content": prefix + _dice(e["content"])})
-    return msgs
-
-def make_share_link(state):
-    if not state or not state["history"]:
-        return gr.update(visible=False)
-    base_text = f"🍿 Survival Level: {state['turn']} actions inside the unhinged '{state['genre']}' AI simulation! Can you survive longer? 👇"
-    encoded_text = urllib.parse.quote(base_text)
-    encoded_url = urllib.parse.quote("https://huggingface.co/spaces/King3Djbl/dual-gm-simulator")
-    twitter_url = f"https://twitter.com/intent/tweet?text={encoded_text}&url={encoded_url}"
-    return gr.update(value=f'<a href="{twitter_url}" target="_blank" class="share-btn">🐦 Broadcast Score to X</a>', visible=True)
+        time.sleep(1) # Visual pacing padding gap
+        yield state, render(state), state["status_msg"], state["hp"], get_leaderboard_df()
+
+    # Game Over validation parameters
+    if state["hp"] <= 0:
+        state["running"] = False
+        state["status_msg"] = "💀 GAME OVER! The player has died."
+        update_leaderboard(state["genre"], state["turn"], "Flatlined at 0 HP")
+    elif state["turn"] >= MAX_TURNS * 2:
+        state["running"] = False
+        state["status_msg"] = "🏆 VICTORY! You survived the entire timeline arena!"
+        update_leaderboard(state["genre"], state["turn"], "Escaped the Arena Alive")
+        
+    yield state, render(state), state["status_msg"], state["hp"], get_leaderboard_df()
 
 # ================== APP LAYOUT & CORE THEME ==================
 custom_css = """
 .dice { display:inline-block; background:#1e1e2e; border-radius:6px; padding:2px 8px; font-weight:bold; color:#f9e2af; border:1px solid #f38ba8; }
-.hp-change { display:inline-block; background:#2a1414; border-radius:6px; padding:2px 8px; font-weight:bold; color:#f38ba8; border:1px solid red; }
-.share-btn { display: block; text-align: center; background: #1da1f2; color: white !important; font-weight: bold; padding: 12px; border-radius: 8px; text-decoration: none; margin-top: 10px; box-shadow: 0 4px 15px rgba(29, 161, 242, 0.3); }
-"""
-
-with gr.Blocks(title="🎭 Tiny Mythos Arena", theme=gr.themes.Default(primary_hue="rose"), css=custom_css) as demo:
-    gr.Markdown("# 🎭 Unhinged Dual-AI Simulation Arena\n### *Two instances of TinyLlama-1.1B cross-talk endless storylines under competing game parameters.*")
-    
-    state = gr.State(new_game())
-
-    with gr.Row():
-        with gr.Column(scale=2):
-            genre_dd = gr.Dropdown(choices=list(GENRES.keys()), value="Dark Fantasy", label="🚀 1. Choose Adventure Dimension (Genre)")
-            hp_bar = gr.Slider(label="❤️ Player Health Pool", value=100, minimum=0, maximum=100, interactive=False)
-            status = gr.Textbox(label="🎬 Arena Director Feed", value="✅ System Ready", interactive=False)
-        with gr.Column(scale=1):
-            start_btn = gr.Button("🎬 Boot Simulation", variant="primary", size="lg")
-            stop_btn = gr.Button("⏹️ Halt Simulation", variant="stop")
-            share_html = gr.HTML(visible=False)
-
-    with gr.Tabs():
-        with gr.TabItem("📺 Live Stream Monitor"):
-            chatbot = gr.Chatbot(label="Interactive Thread Logs", height=500, type="messages")
-            with gr.Row():
-                intervene_box = gr.Textbox(label="💥 Inject Neural Hack (Player Mind Intervention)", placeholder="Force the player to execute a specific move...", lines=2, scale=4)
-                send_intervene = gr.Button("⚡ Inject Move", variant="secondary", scale=1)
-        
-        with gr.TabItem("🏆 Global Hall of Fame (Leaderboard)"):
-            gr.Markdown("### 💀 Top Simulations Survived Across Active Container Lifespan")
-            leaderboard_view = gr.Dataframe(
-                headers=["Rank", "World Dimension", "Actions", "Ending Fate"],
-                value=get_leaderboard_df(),
-                interactive=False
-            )
-
-    # Hidden background runner clock loop setup
-    timer = gr.Timer(value=2, active=False)
-
-    def start_action(chosen_genre):
-        s = new_game(chosen_genre)
-        s["running"] = True
-        return s, f"🎬 Running dimension: '{chosen_genre}'... Generating introduction...", gr.Timer(active=True), s["hp"]
-
-    def stop_action(s):
-        if s: s["running"] = False
-        return s, "⏹️ Simulation frozen manually.", gr.Timer(active=False), s["hp"] if s else 100
-
-    def intervene_action(msg, s):
-        if s and s.get("running") and msg.strip():
-            s["intervention"] = msg.strip()
-            s["next"] = "player"
-            return s, f"🔮 Mind Control Override queued: '{msg.strip()}'", ""
-        return s, "Boot a simulation before hacking the neural stream!", ""
-
-    def tick(s):
-        s = advance(s)
-        timer_update = gr.Timer(active=s["running"])
-        return s, render(s), s["status_msg"], s["hp"], timer_update, get_leaderboard_df()
-
-    # Event Connections
-    start_btn.click(start_action, inputs=genre_dd, outputs=[state, status, timer, hp_bar])
-    stop_btn.click(stop_action, inputs=state, outputs=[state, status, timer, hp_bar])
-    
-    send_intervene.click(intervene_action, inputs=[intervene_box, state], outputs=[state, status, intervene_box])
-    intervene_box.submit(intervene_action, inputs=[intervene_box, state], outputs=[state, status, intervene_box])
-    
-    timer.tick(tick, inputs=state, outputs=[state, chatbot, status, hp_bar, timer, leaderboard_view]).then(
-        make_share_link, inputs=state, outputs=share_html
-    )
-
-if __name__ == "__main__":
-    demo.queue(default_concurrency_limit=1).launch(server_name="0.0.0.0", server_port=7860)
-