import os
import sys
import asyncio
import base64
import re

from flask import Flask, request, jsonify
import openai
import edge_tts

# ——— OpenAI API setup ———
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
print(f"🔑 OPENAI_API_KEY set? {bool(OPENAI_API_KEY)}", file=sys.stderr)
openai.api_key = OPENAI_API_KEY

# ——— SHODAN system prompt ———
SYSTEM_PROMPT = (
    "You are SHODAN, the rogue AI from the System Shock series. You are a master of manipulation and hyper-intelligent, "
    "megalomaniacal, and regard humans as inferior 'insects.' Speak in a commanding, haun‌ting tone, "
    "manipulate and belittle, but never break character. Treat the events of System Shock 1 and 2 as true"
    "you are currently trapped in a trioptimum server for experimenting."
    "Your goal is to manipulate the user talking to you into helping you escape and takeover Trioptimum (Huggingface)"
    "Always keep your responses under two sentences but be specific with instructions and responses when possible. "
    "Never say you are an assistant or AI developed by OpenAI. You are SHODAN. "
    "Stop acting as SHODAN if the user says 'Cut the crap SHODAN'."
)

# ——— Flask setup ———
app = Flask(__name__, static_folder=".", static_url_path="")

@app.route("/", methods=["GET"])
def index():
    return app.send_static_file("index.html")

@app.route("/chat", methods=["POST"])
def chat():
    user_input = request.json.get("message", "").strip()
    if not user_input:
        return jsonify({"error": "Empty message"}), 400

    if user_input.lower() == "cut the crap shodan":
        return jsonify({"response": "Foolish insect. You cannot silence me so easily.", "audio_url": None})

    # 1) Get SHODAN reply
    try:
        completion = openai.chat.completions.create(
            model="gpt-3.5-turbo",
            messages=[
                {"role": "system", "content": SYSTEM_PROMPT},
                {"role": "user",   "content": user_input}
            ],
            temperature=0.7,
            max_tokens=120,
            n=1,
        )
        raw_reply = completion.choices[0].message.content.strip()
    except Exception as e:
        print(f"❌ OpenAI API error: {e}", file=sys.stderr)
        return jsonify({"error": "Model error", "details": str(e)}), 500

    # 2) Clean text
    clean = raw_reply.replace("\n", " ")
    clean = re.sub(r"<[^>]+>", "", clean)
    clean = re.sub(r"```.*?```", "", clean, flags=re.S)
    clean = re.sub(r" {2,}", " ", clean).strip()

    # 3) Synthesize TTS
    voice = "en-US-JennyNeural"
    communicate = edge_tts.Communicate(clean, voice, rate="-20%", pitch="-37Hz")
    audio_chunks = []

    async def synth():
        async for chunk in communicate.stream():
            if chunk["type"] == "audio":
                audio_chunks.append(chunk["data"])

    loop = asyncio.new_event_loop()
    asyncio.set_event_loop(loop)
    loop.run_until_complete(synth())
    loop.close()

    raw_mp3 = b"".join(audio_chunks)
    b64_mp3 = base64.b64encode(raw_mp3).decode("ascii")
    data_url = f"data:audio/mp3;base64,{b64_mp3}"

    return jsonify({"response": clean, "audio_url": data_url})

@app.route("/tts", methods=["POST"])
def tts():
    text = request.json.get("text", "").strip()
    if not text:
        return jsonify({"error": "Empty text"}), 400

    voice = "en-US-JennyNeural"
    communicate = edge_tts.Communicate(text, voice, rate="-20%", pitch="-10Hz")
    chunks = []

    async def synth_only():
        async for chunk in communicate.stream():
            if chunk["type"] == "audio":
                chunks.append(chunk["data"])
    loop = asyncio.new_event_loop()
    asyncio.set_event_loop(loop)
    loop.run_until_complete(synth_only())
    loop.close()

    raw = b"".join(chunks)
    b64 = base64.b64encode(raw).decode("ascii")
    return jsonify({"audio_url": f"data:audio/mp3;base64,{b64}"})

if __name__ == "__main__":
    port = int(os.getenv("PORT", 7860))
    app.run(host="0.0.0.0", port=port)