Update app.py

app.py

@@ -15,6 +15,9 @@ import uvicorn
 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)
@@ -39,7 +42,7 @@ api_app = FastAPI()
 def get_generated_image(text: str):
     """
     API endpoint that returns the generated image directly.
-    Example: https://your-domain/?text=jokowi will eventually call this endpoint.
+    Example: https://bowozzz-gen.hf.space/api?text=jokowi
     """
     try:
         image = generate_image(text)
@@ -58,21 +61,18 @@ threading.Thread(target=run_api, daemon=True).start()
 
 # --- Streamlit UI ---
 
-# Use st.query_params (a property) to check for a query parameter.
+# 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.
-    # For API clients, we don't want to show UI elements like buttons.
-    # Instead, we offer a clickable link or instructions so that they know
-    # they can fetch the image directly from our FastAPI endpoint.
-    st.write("A request with a text query parameter was detected. To get the raw image data, please use the API endpoint:")
-    # Construct the API URL (adjust the host/domain and port as needed)
-    api_endpoint = f"{st.experimental_get_url()[:-1]}/api?text={prompt_from_url}"
+    # 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, you can also show the generated image in the UI:
+    # 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: