from transformers import pipeline
import streamlit as st

# Text generation pipeline
generator = pipeline("text-generation", model="mosaicml/mpt-7b-storywriter", trust_remote_code=True)

# Summarization pipeline
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")

# Input text
prompt = st.text_input("Enter a prompt to expand on:")

if prompt:
    # Generate text based on the prompt
    generated_text = generator(prompt, max_length=300, num_return_sequences=1)
    detailed_text = generated_text[0]['generated_text']
    st.write("Generated Text:")
    st.write(detailed_text)

    # Summarize the generated text
    summary = summarizer(detailed_text, max_length=130, min_length=30, do_sample=False)
    summarized_text = summary[0]['summary_text']
    st.write("Summarized Text:")
    st.write(summarized_text)