app.py

import gradio as gr
from transformers import pipeline

# Load Hugging Face model once
generator = pipeline("text-generation", model="gpt2")

def generate_blog(topic: str, max_tokens: int = 250) -> str:
    prompt = f"Write a detailed blog on the topic: {topic}\n\n"
    result = generator(prompt, max_length=max_tokens, do_sample=True, temperature=0.7)
    return result[0]["generated_text"]

# Launch Gradio app
demo = gr.Interface(
    fn=generate_blog,
    inputs=[
        gr.Textbox(label="Enter Blog Topic", placeholder="e.g., The Future of AI"),
        gr.Slider(50, 500, value=250, label="Max Tokens")
    ],
    outputs=gr.Textbox(label="Generated Blog", lines=15),
    title="📝 AI Blog Writer",
    description="Enter a topic and generate a full-length blog using GPT-2!"
)

demo.launch()