import os
import gradio as gr
from huggingface_hub import InferenceClient
from duckduckgo_search import DDGS

# --- APEX-1 SOVEREIGN ENGINE ---
MODEL_ID = "Qwen/Qwen2.5-Coder-32B-Instruct"
IMAGE_ID = "black-forest-labs/FLUX.1-schnell"

system_halted = False

def toggle_halt(state):
    global system_halted
    system_halted = state
    return "### // STATUS: EMERGENCY HALT //" if state else "### // STATUS: OPERATIONAL //"

def sentinel_scan(query):
    if system_halted: return "[HALTED]"
    try:
        results = DDGS().text(query, max_results=3)
        return "\n".join([f"- {r['title']}: {r['body']}" for r in results])
    except: return "[SENTINEL_OFFLINE]"

def apex_1_logic(message, history, protocol, use_web, use_img):
    if system_halted:
        yield "CRITICAL: SYSTEM HALTED BY COMMANDER."
        return

    client = InferenceClient()

    if use_img:
        yield "SYSTEM: GENERATING VISUAL DATA..."
        img = client.text_to_image(message, model=IMAGE_ID)
        yield img
        return

    context = sentinel_scan(message) if use_web else "Local Intel Only."
    prompts = {
        "WAR_ROOM": "ROLE: Strategic Commander. TONE: Ruthless. GOAL: Global Supremacy.",
        "WINGMAN": "ROLE: Social Architect. TONE: Charismatic. GOAL: Social Dominance."
    }

    system_instr = f"{prompts[protocol]} | INTEL: {context}"
    
    partial = ""
    messages = [{"role": "system", "content": system_instr}]
    for h in history:
        messages.append({"role": "user", "content": h[0]})
        messages.append({"role": "assistant", "content": h[1]})
    messages.append({"role": "user", "content": message})

    for chunk in client.chat_completion(model=MODEL_ID, messages=messages, max_tokens=2048, stream=True):
        if chunk.choices[0].delta.content:
            partial += chunk.choices[0].delta.content
            yield partial

# --- THE GHOST HUD UI ---
with gr.Blocks(theme=gr.themes.Monochrome(), css="body {background: #050505; color: #00ffcc;}") as demo:
    status = gr.Markdown("### // STATUS: OPERATIONAL //")
    with gr.Row():
        protocol = gr.Dropdown(["WAR_ROOM", "WINGMAN"], value="WAR_ROOM", label="PROTOCOL")
        halt = gr.Checkbox(label="KILL SWITCH", value=False)
        web = gr.Checkbox(label="SENTINEL_SCAN", value=True)
        img = gr.Checkbox(label="RENDER_IMAGE", value=False)

    chat = gr.Chatbot(height=550, label="APEX-1 LINK")
    msg = gr.Textbox(placeholder="COMMAND...", label="INPUT")

    halt.change(toggle_halt, [halt], [status])
    msg.submit(apex_1_logic, [msg, chat, protocol, web, img], [chat])

if __name__ == "__main__":
    demo.launch()