Create app.py

app.py

@@ -0,0 +1,295 @@
+import os
+
+import gradio as gr
+from game_config import CLOUD_TICK_SECONDS, HF_MAX_NEW_TOKENS, HF_MODEL_ID
+from huggingface_hub import InferenceClient
+from state import (
+    buy_upgrade,
+    chat_temperature,
+    cloud_tick,
+    new_game,
+    status_lines,
+    toggle_cloud,
+    train_model,
+    unlock_cloud,
+    upgrade_status,
+)
+
+CSS = """
+<style>
+@import url('https://fonts.googleapis.com/css2?family=Fira+Code:wght@400;700&display=swap');
+body, .gradio-container {
+    background-color: #0a0f0d !important;
+    font-family: 'Fira Code', monospace !important;
+    color: #33ff33 !important;
+}
+#terminal-card {
+    background-color: #0d1410 !important;
+    border: 2px solid #22aa22 !important;
+    box-shadow: 0 0 20px rgba(51, 255, 51, 0.15);
+    border-radius: 4px;
+    padding: 15px;
+}
+.gr-button, button {
+    background: #142218 !important;
+    color: #33ff33 !important;
+    border: 1px solid #33ff33 !important;
+    font-family: 'Fira Code', monospace !important;
+    border-radius: 2px !important;
+    cursor: pointer;
+    transition: all 0.2s ease;
+}
+.gr-button:hover, button:hover {
+    background: #33ff33 !important;
+    color: #0a0f0d !important;
+    box-shadow: 0 0 10px #33ff33;
+}
+.gr-text-input, textarea, input[type="text"] {
+    background-color: #050806 !important;
+    border: 1px solid #22aa22 !important;
+    color: #33ff33 !important;
+    font-family: 'Fira Code', monospace !important;
+    border-radius: 2px !important;
+}
+.gr-text-input:focus, textarea:focus, input[type="text"]:focus {
+    border-color: #33ff33 !important;
+    outline: none !important;
+}
+#console-log-area {
+    max-height: 350px;
+    overflow-y: auto;
+}
+footer { display: none !important; }
+</style>
+"""
+
+
+def get_huggingface_response(prompt: str, temperature: float, token: str) -> str:
+    """
+    Queries Hugging Face using the player's active OAuth token.
+    Uses chat_completion to satisfy the conversational task requirement.
+    """
+    try:
+        # Securely explicitly initializes InferenceClient with the player's token
+        client = InferenceClient(model=HF_MODEL_ID, token=token)
+
+        messages = [{"role": "user", "content": prompt}]
+
+        response = client.chat_completion(
+            messages=messages,
+            max_tokens=HF_MAX_NEW_TOKENS,
+            temperature=max(0.01, temperature),
+        )
+        return response.choices[0].message.content
+
+    except Exception as e:
+        return f"[FATAL_ERR] Model Inference Failed: {str(e)}"
+
+
+def render_ui_panes(state: dict) -> tuple[str, str, str]:
+    stats = f"=== SYSTEM STATUS ===\n{status_lines(state)}\n====================="
+
+    upg_info = upgrade_status(state)
+    upg_lines = ["=== HARDWARE & ARCHITECTURE MARKETPLACE ==="]
+    for key, info in upg_info.items():
+        cost_str = f"${info['next_cost']:.0f}" if not info["maxed"] else "MAXED"
+        upg_lines.append(
+            f"{info['icon']} {info['label']} [{info['current_name']}]\n"
+            f"  Tier: {info['current_tier']}/{info['max_tier']} | Cost: {cost_str}\n"
+            f"  Desc: {info['description']}"
+        )
+    upg_lines.append("==========================================")
+
+    log_text = "\n".join(state.get("log", ["[SYS] Cluster Initialized."]))
+    return stats, "\n\n".join(upg_lines), log_text
+
+
+# --- Interface Core Action Mappers ---
+
+
+def handle_init():
+    state = new_game()
+    stats, upgs, logs = render_ui_panes(state)
+    return state, stats, upgs, logs, gr.update(choices=list(state["upgrades"].keys()))
+
+
+def handle_train(state: dict):
+    new_state, _ = train_model(state)
+    stats, upgs, logs = render_ui_panes(new_state)
+    return new_state, stats, upgs, logs
+
+
+def handle_upgrade(state: dict, upgrade_key: str):
+    if not upgrade_key:
+        return state, gr.update(), gr.update(), gr.update()
+    new_state, _ = buy_upgrade(state, upgrade_key)
+    stats, upgs, logs = render_ui_panes(new_state)
+    return new_state, stats, upgs, logs
+
+
+def handle_cloud_unlock(state: dict):
+    new_state, _ = unlock_cloud(state)
+    stats, upgs, logs = render_ui_panes(new_state)
+    return new_state, stats, upgs, logs
+
+
+def handle_cloud_toggle(state: dict):
+    new_state, _ = toggle_cloud(state)
+    stats, upgs, logs = render_ui_panes(new_state)
+    return new_state, stats, upgs, logs
+
+
+def handle_tick(state: dict):
+    if not state.get("cloud_active", False):
+        return state, gr.update(), gr.update(), gr.update()
+    new_state, revenue = cloud_tick(state)
+    stats, upgs, logs = render_ui_panes(new_state)
+    return new_state, stats, upgs, logs
+
+
+def handle_chat(
+    state: dict,
+    history: list,
+    message: str,
+    profile: gr.OAuthProfile | None,
+    oauth_token: gr.OAuthToken | None,
+):
+    if not message.strip():
+        return history, ""
+
+    if history is None:
+        history = []
+
+    # Block inference if the user isn't authenticated through the UI
+    if oauth_token is None:
+        history.append(
+            [
+                message,
+                "[SYS_ERR] Access Denied. Please authorize using the 'Sign in with Hugging Face' button.",
+            ]
+        )
+        return history, ""
+
+    temp = chat_temperature(state)
+
+    if temp > 1.4:
+        prompt = f"Produce dynamic fragmented word-salad, glitch texts, and chaotic tokens based loosely on: {message}"
+    else:
+        prompt = message
+
+    # Execute request using their injected token
+    response = get_huggingface_response(prompt, temp, oauth_token.token)
+
+    # Reverted to standard tuple arrays to fix TypeError in Chatbot init
+    history.append([message, response])
+
+    return history, ""
+
+
+# --- Layout Assembly ---
+
+with gr.Blocks(title="AI Training Simulator v2.026") as demo:
+    gr.HTML(CSS)
+    game_state = gr.State()
+    tick_trigger = gr.Button("tick_trigger", visible=False, elem_id="tick_trigger")
+
+    gr.Markdown("# 📟 THOUSAND TOKEN WOOD // STARTUP TERMINAL")
+
+    with gr.Column(elem_id="terminal-card"):
+        with gr.Row():
+            with gr.Column(scale=1):
+                status_pane = gr.Code(
+                    label="System Mon", language="markdown", interactive=False
+                )
+                train_btn = gr.Button("⚡ RUN TRAINING PASS", variant="primary")
+
+                with gr.Row():
+                    unlock_cloud_btn = gr.Button("🔓 UNLOCK CLOUD")
+                    toggle_cloud_btn = gr.Button("⏯️ TOGGLE CLOUD")
+
+            with gr.Column(scale=1):
+                market_pane = gr.Code(
+                    label="Stack Upgrade Hub", language="markdown", interactive=False
+                )
+                upgrade_dropdown = gr.Dropdown(
+                    label="Select Component to Upgrade", choices=[]
+                )
+                buy_btn = gr.Button("💳 PURCHASE UPGRADE")
+
+        gr.Markdown("### 📜 Console System Output Logs")
+        log_pane = gr.Textbox(
+            label="", interactive=False, lines=8, elem_id="console-log-area"
+        )
+
+        gr.Markdown("### 💬 Live Model Inference Playground")
+
+        # Injected Native Hugging Face Auth Components
+        with gr.Row():
+            gr.LoginButton()
+
+        chatbot = gr.Chatbot(label="Model Output Channel")
+        with gr.Row():
+            chat_input = gr.Textbox(
+                placeholder="Query model parameter space...", scale=4, show_label=False
+            )
+            send_btn = gr.Button("📡 INFERENCE", scale=1)
+
+    # --- Wire UI action flows ---
+
+    demo.load(
+        handle_init,
+        outputs=[game_state, status_pane, market_pane, log_pane, upgrade_dropdown],
+    )
+
+    demo.load(
+        None,
+        js=f"""
+        function() {{
+            setInterval(() => {{
+                let btn = document.querySelector('#tick_trigger');
+                if (btn) btn.click();
+            }}, {CLOUD_TICK_SECONDS * 1000});
+        }}
+    """,
+    )
+
+    train_btn.click(
+        handle_train,
+        inputs=[game_state],
+        outputs=[game_state, status_pane, market_pane, log_pane],
+    )
+    buy_btn.click(
+        handle_upgrade,
+        inputs=[game_state, upgrade_dropdown],
+        outputs=[game_state, status_pane, market_pane, log_pane],
+    )
+    unlock_cloud_btn.click(
+        handle_cloud_unlock,
+        inputs=[game_state],
+        outputs=[game_state, status_pane, market_pane, log_pane],
+    )
+    toggle_cloud_btn.click(
+        handle_cloud_toggle,
+        inputs=[game_state],
+        outputs=[game_state, status_pane, market_pane, log_pane],
+    )
+    tick_trigger.click(
+        handle_tick,
+        inputs=[game_state],
+        outputs=[game_state, status_pane, market_pane, log_pane],
+    )
+
+    # Gradio handles mapping profile/oauth_token automatically, so we don't put them in the inputs array
+    send_btn.click(
+        handle_chat,
+        inputs=[game_state, chatbot, chat_input],
+        outputs=[chatbot, chat_input],
+    )
+    chat_input.submit(
+        handle_chat,
+        inputs=[game_state, chatbot, chat_input],
+        outputs=[chatbot, chat_input],
+    )
+
+if __name__ == "__main__":
+    demo.launch()