Update app.py

app.py

@@ -2,18 +2,32 @@ import streamlit as st
 import requests
 from PIL import Image
 from io import BytesIO
+import time
 
 st.title('CreativeAI')
+st.description('This is AI image generation based on KVIImager 2.0 API by KVI Kontent')
 
-prompt = st.text_input("Enter prompt")
+# Pre-defined prompts
+example_prompts = ["A cat sitting on a keyboard", "A futuristic cityscape", "Magical forest at night"]
 
-if prompt:
-    url = 'https://ee1f-217-71-237-228.ngrok-free.app/generate_image'
-    params = {'prompt': prompt}
-    response = requests.get(url, params=params)
+# Select box for choosing examples
+selected_prompt = st.selectbox("Select an example prompt", example_prompts)
 
-    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.")
+# 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...'):
+            time.sleep(3)  # Simulate some processing time
+            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.")