Update app.py

app.py

@@ -4,25 +4,16 @@ import io
 import random
 from PIL import Image
 import os
-import threading
-
-# Import FastAPI-related classes
-from fastapi import FastAPI, HTTPException
-from fastapi.responses import StreamingResponse
-import uvicorn
 
 # API Configuration for Hugging Face
 API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
 headers = {"Authorization": f"Bearer {os.getenv('HF')}"}
 
-# Define your base URL here (adjust this for your deployment)
-BASE_URL = "https://bowozzz-gen.hf.space"
-
-# Function to query the Hugging Face API and generate an image
 def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
     if response.status_code != 200:
-        raise HTTPException(status_code=response.status_code, detail="API call failed")
+        st.error("Error calling the Hugging Face API.")
+        st.stop()
     return response.content
 
 def generate_image(prompt: str) -> Image.Image:
@@ -35,71 +26,29 @@ def generate_image(prompt: str) -> Image.Image:
     image_bytes = query(payload)
     return Image.open(io.BytesIO(image_bytes))
 
-# --- FastAPI App for API endpoint ---
-api_app = FastAPI()
-
-@api_app.get("/api")
-def get_generated_image(text: str):
-    """
-    API endpoint that returns the generated image directly.
-    Example: https://bowozzz-gen.hf.space/api?text=jokowi
-    """
-    try:
-        image = generate_image(text)
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
-    img_buffer = io.BytesIO()
-    image.save(img_buffer, format="PNG")
-    img_buffer.seek(0)
-    return StreamingResponse(img_buffer, media_type="image/png")
-
-# Run FastAPI in a separate thread so it runs alongside Streamlit
-def run_api():
-    uvicorn.run(api_app, host="0.0.0.0", port=8000)
-
-threading.Thread(target=run_api, daemon=True).start()
-
-# --- Streamlit UI ---
-
-# Use st.query_params to check for a query parameter.
-query_params = st.query_params
-prompt_from_url = query_params.get('text')
-
-if prompt_from_url:
-    # If a query parameter "text" is present, assume this is an API request.
-    # Provide the API endpoint URL that returns raw image data.
-    st.write("A request with a text query parameter was detected. To get the raw image data, use the API endpoint below:")
-    api_endpoint = f"{BASE_URL}/api?text={prompt_from_url}"
-    st.write("API Endpoint:", api_endpoint)
-    
-    # Optionally, display the generated image in the UI:
-    image = generate_image(prompt_from_url)
-    st.image(image, caption="Generated Image", use_container_width=True)
-else:
-    # Standard interactive UI
-    st.title("AI Image Generator 🎨🚀")
-    st.write("Enter a prompt below and generate an AI-generated image using Hugging Face!")
-    
-    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 = generate_image(prompt)
-            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)
-            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.")
-    
-    st.write("---")
-    st.write("Powered by [Hugging Face](https://huggingface.co/) 🚀")
+st.title("AI Image Generator 🎨🚀")
+st.write("Enter a prompt below and generate an AI-generated image using Hugging Face!")
+
+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 = generate_image(prompt)
+        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)
+        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.")
+
+st.write("---")
+st.write("Powered by [Hugging Face](https://huggingface.co/) 🚀")