Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import requests
 import io
+import random
 from PIL import Image
 import os
 
@@ -17,11 +18,22 @@ def query(payload):
 query_params = st.query_params
 prompt_from_url = query_params.get('text')
 
+def generate_image(prompt):
+    # Generate a random seed to force a new image
+    random_seed = random.randint(0, 999999)
+    payload = {
+        "inputs": prompt,
+        "parameters": {
+            "seed": random_seed
+        }
+    }
+    image_bytes = query(payload)
+    return Image.open(io.BytesIO(image_bytes))
+
 if prompt_from_url:
     # If 'text' parameter is present, generate and display the image only
     prompt = prompt_from_url
-    image_bytes = query({"inputs": prompt})
-    image = Image.open(io.BytesIO(image_bytes))
+    image = generate_image(prompt)
     st.image(image, caption="Generated Image", use_container_width=True)
 
     # Provide download link
@@ -45,10 +57,9 @@ else:
     if st.button("Generate Image"):
         if prompt:
             st.write("Generating image... Please wait ⏳")
-            image_bytes = query({"inputs": prompt})
+            image = generate_image(prompt)
 
             # Display Image
-            image = Image.open(io.BytesIO(image_bytes))
             st.image(image, caption="Generated Image", use_container_width=True)
 
             # Convert Image to Bytes for Download