import streamlit as st
from transformers import pipeline
model_name = "gpt2"
text_generator = pipeline("text-generation", model=model_name)
def generate_blog(topic, word_count):
    try:
        word_count = int(word_count)
        prompt = f"Write a detailed blog about: {topic}\n\n"
        generated_text = text_generator(prompt, max_length=word_count, num_return_sequences=1)
        return generated_text[0]["generated_text"]
    except Exception as e:
        return f"Error: {str(e)}"
st.title("Blog Generator")
st.write("Provide a topic and word count to generate a blog.")
topic = st.text_input("Blog Topic", placeholder="Enter the topic here")
word_count = st.text_input("Word Count", placeholder="Enter the desired word count")
if st.button("Generate Blog"):
    if topic and word_count.isdigit():
        with st.spinner("Generating blog..."):
            blog_content = generate_blog(topic, word_count)
            st.text_area("Generated Blog Content", value=blog_content, height=300)
    else:
        st.error("Please enter a valid topic and numerical word count.")