Update app.py

app.py

@@ -13,37 +13,59 @@ def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.content
 
-# Streamlit UI
-st.title("AI Image Generator 🎨🚀")
-st.write("Enter a prompt below and generate an AI-generated image using Hugging Face!")
-
-# User Input
-prompt = st.text_input("Enter your prompt:", "Astronaut riding a horse")
-
-if st.button("Generate Image"):
-    if prompt:
-        st.write("Generating image... Please wait ⏳")
-        image_bytes = query({"inputs": 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
-        img_buffer = io.BytesIO()
-        image.save(img_buffer, format="PNG")
-        img_buffer.seek(0)
-
-        # Download Button
-        st.download_button(
-            label="Download Image 📥",
-            data=img_buffer,
-            file_name="generated_image.png",
-            mime="image/png"
-        )
-    else:
-        st.warning("Please enter a prompt before generating an image.")
-
-# Footer
-st.write("---")
-st.write("Powered by [Hugging Face](https://huggingface.co/) 🚀")
+# Check for 'text' parameter in URL
+query_params = st.experimental_get_query_params()
+prompt_from_url = query_params.get('text', None)
+
+if prompt_from_url:
+    # If 'text' parameter is present, generate and display the image only
+    prompt = prompt_from_url[0]
+    image_bytes = query({"inputs": prompt})
+    image = Image.open(io.BytesIO(image_bytes))
+    st.image(image, caption="Generated Image", use_column_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:
+    # Standard interface with prompt input
+    st.title("AI Image Generator 🎨🚀")
+    st.write("Enter a prompt below and generate an AI-generated image using Hugging Face!")
+
+    # User Input
+    prompt = st.text_input("Enter your prompt:", "Astronaut riding a horse")
+
+    if st.button("Generate Image"):
+        if prompt:
+            st.write("Generating image... Please wait ⏳")
+            image_bytes = query({"inputs": prompt})
+
+            # Display Image
+            image = Image.open(io.BytesIO(image_bytes))
+            st.image(image, caption="Generated Image", use_column_width=True)
+
+            # Convert Image to Bytes for Download
+            img_buffer = io.BytesIO()
+            image.save(img_buffer, format="PNG")
+            img_buffer.seek(0)
+
+            # Download Button
+            st.download_button(
+                label="Download Image 📥",
+                data=img_buffer,
+                file_name="generated_image.png",
+                mime="image/png"
+            )
+        else:
+            st.warning("Please enter a prompt before generating an image.")
+
+    # Footer
+    st.write("---")
+    st.write("Powered by [Hugging Face](https://huggingface.co/) 🚀")