app.py

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

st.title('CreativeAI')

# Description
st.write("""
This app uses HuggingFace to generate an image based on the provided prompt. You can either select an example prompt from the dropdown list or enter your own prompt.
Click the "Generate Image" button to initiate the image generation process. Based on KVIImager API by KVI Kontent.
""")

custom_prompt = st.text_input("Enter your own prompt")

prompt = custom_prompt

if st.button('Generate Image'):
    if prompt:
        with st.spinner('Generating your image...'):
            url = 'https://ee1f-217-71-237-228.ngrok-free.app/generate_image'
            params = {'prompt': prompt}
            response = requests.get(url, params=params)

            if response.status_code == 200:
                image = Image.open(BytesIO(response.content))
                st.image(image, caption='Generated Image', use_column_width=True)
            else:
                st.error("Failed to generate the image. Please try again.")