Update app.py

app.py

@@ -14,8 +14,12 @@ def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.content
 
+# Check for 'text' parameter in URL
+query_params = st.query_params
+prompt_from_url = query_params.get('text')
+
 def generate_image(prompt):
-    # Generate a random seed to ensure a new image each time
+    # Generate a random seed to force a new image
     random_seed = random.randint(0, 999999)
     payload = {
         "inputs": prompt,
@@ -26,29 +30,22 @@ def generate_image(prompt):
     image_bytes = query(payload)
     return Image.open(io.BytesIO(image_bytes))
 
-# Default prompt (used since we're not using a query parameter anymore)
-default_prompt = "Astronaut riding a horse"
-
-# Use session_state to avoid regenerating the image on every rerun
-if "image" not in st.session_state:
-    st.session_state.image = generate_image(default_prompt)
-
-if st.button("Generate"):
-    st.session_state.image = generate_image(default_prompt)
-
-# Display the image using the updated parameter
-image = st.session_state.image
-st.image(image, caption="Result", use_container_width=True)
-
-# Prepare image for download
-img_buffer = io.BytesIO()
-image.save(img_buffer, format="PNG")
-img_buffer.seek(0)
-st.download_button(
-    label="Download",
-    data=img_buffer,
-    file_name="generated_image.png",
-    mime="image/png"
-)
-
-st.write("---")
+if prompt_from_url:
+    # Generate and display image when 'text' parameter exists
+    prompt = prompt_from_url
+    image = generate_image(prompt)
+    st.image(image, caption="Generated Image", use_container_width=True)
+
+    # Provide download link
+    img_buffer = io.BytesIO()
+    image.save(img_buffer, format="PNG")
+    img_buffer.seek(0)
+    st.download_button(
+        label="Download Image 📥",
+        data=img_buffer,
+        file_name="generated_image.png",
+        mime="image/png"
+    )
+else:
+    # Show instructions when no parameter is provided
+    st.info("Add a 'text' parameter to the URL to generate an image. Example: ?text=astronaut+riding+a+horse")