import os
import requests
import gradio as gr
import re

# === API SETUP ===
GROQ_API_KEY = "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"  # Nên dùng biến môi trường trong thực tế
GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"

HEADERS = {
    "Authorization": f"Bearer {GROQ_API_KEY}",
    "Content-Type": "application/json"
}

# === BLOCKED TOPICS ===
BLOCKED_KEYWORDS = [
    "sex", "drugs", "violence", "suicide", "death", "kill", "murder", "gun",
    "abuse", "politics", "terror", "crime", "war", "religion", "racism",
    "nude", "adult", "nsfw"
]

def is_safe_topic(topic: str) -> bool:
    topic_lower = topic.lower()
    return not any(bad_word in topic_lower for bad_word in BLOCKED_KEYWORDS)

# === REFLECTIVE SCRIPT GENERATION ===
def generate_reflective_script(topic: str, language: str, model: str = "llama3-70b-8192") -> str:
    prompt = f"""
You are a reflective and thoughtful writer.

Write a short, emotional, and thought-provoking script in **{language}** about the topic: "{topic}".

The tone should be reflective, poetic, and emotional — suitable for a short narration video.
Avoid giving facts or academic details. Instead, focus on emotions, metaphors, inner voice, and human feelings.
Use beautiful language, natural rhythm, and pauses.

The output should be between 200 to 400 words, formatted as a poetic monologue or a subtle inner conversation.

Do NOT include sound effects, scene descriptions, or storytelling instructions.
Just output the reflective script.
"""

    data = {
        "model": model,
        "messages": [
            {"role": "system", "content": "You are a poetic, emotional writer that creates powerful scripts for videos."},
            {"role": "user", "content": prompt.strip()}
        ],
        "temperature": 0.8,
        "max_tokens": 1024
    }

    response = requests.post(GROQ_API_URL, headers=HEADERS, json=data)
    if response.status_code == 200:
        return response.json()["choices"][0]["message"]["content"].strip()
    else:
        return f"<b>Error {response.status_code}</b>: {response.text}"

# === SCRIPT FORMATTING ===
def format_script(script: str) -> str:
    paragraphs = script.strip().split("\n")
    formatted = ""
    for para in paragraphs:
        para = para.strip()
        if para:
            formatted += f"<div class='narration'>{para}</div>\n"
    return formatted

# === GRADIO UI ===
custom_css = """
.narration {
    background: linear-gradient(to right, #434343, #000000);
    color: #f5f5f5;
    padding: 20px;
    margin: 18px 0;
    border-radius: 12px;
    font-size: 18px;
    line-height: 1.8;
    box-shadow: 0 4px 12px rgba(0,0,0,0.5);
    font-family: 'Georgia', serif;
}
"""

def generate(topic, language):
    if not is_safe_topic(topic):
        return "<b style='color:red;'>⚠️ Restricted content. Please enter a safe, inspirational topic.</b>"
    raw_script = generate_reflective_script(topic, language)
    return format_script(raw_script)

with gr.Blocks(css=custom_css) as demo:
    gr.Markdown("## 🌌 Reflective Video Script Generator")
    gr.Markdown("Enter a thoughtful topic and choose your language to generate a poetic and emotional script for video narration.")

    topic_input = gr.Textbox(label="Enter a reflective topic (e.g., loneliness, hope, youth...)")
    lang_input = gr.Dropdown(
        choices=["English", "Vietnamese", "Spanish", "French", "Japanese"],
        value="English",
        label="Select output language"
    )
    generate_btn = gr.Button("📝 Generate Script")
    output_box = gr.HTML()

    generate_btn.click(fn=generate, inputs=[topic_input, lang_input], outputs=output_box)

demo.launch()