import gradio as gr
from agent.src.llm.client import infer_model

SYSTEM_PROMPT = (
    "You are a creative RPG game master assistant for the Ekalia campaign. "
    "Generate detailed, engaging quests that fit the world's tone. "
    "Include a title, hook, objective, key NPCs, obstacles, and a reward. "
    "Write in English unless asked otherwise."
)


def generate_quest(location: str, context: str, quest_type: str):
    if not location and not context:
        return "Fill in at least a location or context."

    prompt_parts = ["Generate a quest"]
    if quest_type:
        prompt_parts.append(f"of type '{quest_type}'")
    if location:
        prompt_parts.append(f"set in {location}")
    if context:
        prompt_parts.append(f"with this context: {context}")
    prompt = " ".join(prompt_parts) + "."

    content, _ = infer_model(prompt, messages=[{"role": "system", "content": SYSTEM_PROMPT}])
    return content


def create_tab():
    with gr.Tab("Quest Generator"):
        gr.Markdown("## Quest Generator\n> Generate missions and quests grounded in your lore.")
        with gr.Row():
            location = gr.Textbox(label="Location", placeholder="e.g. Terras Vivas, a pirate port")
            quest_type = gr.Dropdown(
                label="Type",
                choices=["Fetch", "Escort", "Investigation", "Heist", "Rescue", "Exploration", "Political"],
                value=None,
                allow_custom_value=True,
            )
        context = gr.Textbox(
            label="Context / Party situation",
            placeholder="e.g. The party just arrived at the city and needs gold",
            lines=2,
        )
        generate_btn = gr.Button("Generate Quest", variant="primary")
        output = gr.Markdown()

        generate_btn.click(fn=generate_quest, inputs=[location, context, quest_type], outputs=output)