from transformers import pipeline
import gradio as gr

generator = pipeline("text-generation", model="toloka/gpt2-large-supervised-prompt-writing")

def generate_prompt(topic):
    prompt = f"Write a creative story about {topic}. Here is a prompt to get you started:"
    result = generator(prompt, max_length=100, num_return_sequences=1)
    return result[0]['generated_text']

iface = gr.Interface(
    fn=generate_prompt,
    inputs="text",
    outputs="text",
    title="Creative Writing Prompt Generation"
)

iface.launch()