| 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, flavourful NPCs that fit the world's tone. " |
| "Include name, appearance, personality, role, a secret, and a plot hook. " |
| "Write in English unless asked otherwise." |
| ) |
|
|
|
|
| def generate_npc(location: str, role: str, traits: str): |
| if not location and not role: |
| return "Fill in at least a location or role." |
|
|
| prompt_parts = ["Generate an NPC"] |
| if location: |
| prompt_parts.append(f"from {location}") |
| if role: |
| prompt_parts.append(f"who is a {role}") |
| if traits: |
| prompt_parts.append(f"with the following traits: {traits}") |
| prompt = " ".join(prompt_parts) + "." |
|
|
| content, _ = infer_model(prompt, messages=[{"role": "system", "content": SYSTEM_PROMPT}]) |
| return content |
|
|
|
|
| def create_tab(): |
| with gr.Tab("NPC Generator"): |
| gr.Markdown("## NPC Generator\n> Generate NPCs consistent with the Ekalia lore.") |
| with gr.Row(): |
| location = gr.Textbox(label="Location", placeholder="e.g. Glückslott, Costas Piratas") |
| role = gr.Textbox(label="Role / Type", placeholder="e.g. merchant, guard captain, spy") |
| traits = gr.Textbox(label="Extra traits (optional)", placeholder="e.g. nervous, hides a scar, fond of birds") |
| generate_btn = gr.Button("Generate NPC", variant="primary") |
| output = gr.Markdown() |
|
|
| generate_btn.click(fn=generate_npc, inputs=[location, role, traits], outputs=output) |
|
|
