import streamlit as st
from diffusers import DiffusionPipeline
from PIL import Image
import requests
from io import BytesIO

st.title('Image Generation App')
pipeline = DiffusionPipeline.from_pretrained("dataautogpt3/OpenDalleV1.1")

prompt = st.text_input('Enter prompt', '')

if st.button('Generate'):
    results = []
    for _ in range(4):
        response = requests.post(pipeline, json={"prompt": prompt})
        image_bytes = BytesIO(response.content)
        img = Image.open(image_bytes)
        results.append(img)

    for img in results:
        st.image(img, use_column_width=True, caption='Generated Image')