app.py

import streamlit as st
from transformers import DiffusionPipeline

# Load the Diffusion pipeline
pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-3-medium")

def generate_prompt(prompt_text):
    # Generate response using the Diffusion model
    response = pipeline(prompt=prompt_text, top_p=0.9, num_return_sequences=1)[0]['generated_text']
    return response

# Streamlit app UI
st.title('Diffusion Model Prompt Generator')

# Input prompt from user
prompt_input = st.text_area('Enter your prompt here:', height=100)

# Generate button
if st.button('Generate'):
    if prompt_input:
        with st.spinner('Generating...'):
            generated_text = generate_prompt(prompt_input)
        st.success('Generation complete!')
        st.text_area('Generated Text:', value=generated_text, height=200)
    else:
        st.warning('Please enter a prompt to generate.')