Create app.py

app.py

@@ -0,0 +1,46 @@
+import os
+import random
+import requests
+import gradio as gr
+from io import BytesIO
+
+def generate_image(prompt, width, height, seed, randomize):
+    # If randomized seed is checked, generate a random seed.
+    if randomize:
+        seed = random.randint(0, 99999)
+    # Fetch the secret URL from environment variables.
+    # The URL should be stored in an env variable named "FLUX_URL".
+    base_url = os.environ.get("FLUX_URL")
+    if not base_url:
+        return "Error: FLUX_URL environment variable not set."
+    # Replace placeholders in the URL.
+    url = (
+        base_url.replace("[prompt]", prompt)
+        .replace("[w]", str(width))
+        .replace("[h]", str(height))
+        .replace("[seed]", str(seed))
+    )
+    # Fetch the image from the URL.
+    response = requests.get(url)
+    return BytesIO(response.content)
+
+# Gradio Interface setup with title and description.
+title = "Unlimited FLUX-Pro"
+description = "Here you can use the FLUX-Pro as much as you want without limits"
+
+iface = gr.Interface(
+    fn=generate_image,
+    inputs=[
+        gr.inputs.Textbox(label="Prompt", placeholder="Enter your image prompt..."),
+        gr.inputs.Slider(minimum=100, maximum=1024, step=1, default=512, label="Width"),
+        gr.inputs.Slider(minimum=100, maximum=1024, step=1, default=512, label="Height"),
+        gr.inputs.Number(label="Seed", default=0),
+        gr.inputs.Checkbox(label="Randomize Seed")
+    ],
+    outputs=gr.outputs.Image(type="pil"),
+    title=title,
+    description=description,
+)
+
+if __name__ == "__main__":
+    iface.launch()