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.
""")

# Pre-defined prompts
example_prompts = ["A cat sitting on a keyboard", "A futuristic cityscape", "Magical forest at night"]

# Select box for choosing examples
selected_prompt = st.selectbox("Select an example prompt", example_prompts)

# Input field for custom prompt
custom_prompt = st.text_input("Enter your own prompt")

prompt = selected_prompt if not custom_prompt else custom_prompt

if st.button('Generate Image'):
    if prompt:
        with st.spinner('Generating your image...'):
            url = 'https://57b7-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.")