Create app.py

app.py

@@ -0,0 +1,109 @@
+import os
+import base64
+import io
+import requests
+from PIL import Image
+import gradio as gr
+
+# ------------------------------------------------------------------------------
+# Make sure you export your API token before running this script:
+#   export CHUTES_API_TOKEN="your_real_token_here"
+# ------------------------------------------------------------------------------
+
+API_TOKEN = os.getenv("CHUTES_API_TOKEN")
+if API_TOKEN is None:
+    raise ValueError("Please set the environment variable CHUTES_API_TOKEN before running.")
+
+def generate_image(prompt: str,
+                   resolution: str,
+                   guidance_scale: float,
+                   num_inference_steps: int,
+                   shift: int) -> Image.Image:
+    """
+    Calls the HiDream (Chutes) API to generate an image from the given prompt
+    and parameters. Returns a PIL.Image.
+    """
+    headers = {
+        "Authorization": f"Bearer {API_TOKEN}",
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "seed": None,                 # always random
+        "shift": shift,
+        "prompt": prompt,
+        "resolution": resolution,
+        "guidance_scale": guidance_scale,
+        "num_inference_steps": num_inference_steps
+    }
+
+    response = requests.post(
+        "https://chutes-hidream.chutes.ai/generate",
+        headers=headers,
+        json=payload
+    )
+
+    if response.status_code != 200:
+        # you can customize error handling as needed
+        raise RuntimeError(f"API returned {response.status_code}: {response.text}")
+
+    data = response.json()
+    # Typically, the API returns a base64‐encoded image under the key "image".
+    # If yours returns a URL (e.g. under "url"), you can fetch that instead.
+    b64_img = data.get("image")
+    if b64_img:
+        decoded = base64.b64decode(b64_img)
+        img = Image.open(io.BytesIO(decoded)).convert("RGB")
+        return img
+
+    # Fallback: if the API returns a direct URL
+    img_url = data.get("url") or data.get("image_url")
+    if img_url:
+        img_response = requests.get(img_url)
+        img = Image.open(io.BytesIO(img_response.content)).convert("RGB")
+        return img
+
+    raise RuntimeError("No image field found in API response.")
+
+
+# Build the Gradio interface
+with gr.Blocks(title="HiDream Unlimited") as demo:
+    gr.Markdown("## HiDream Unlimited\nGenerate unlimited AI‐driven images powered by HiDream/Chutes.\n\nEnter your prompt and tweak the parameters, then click **Generate**.")
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            prompt_in = gr.Textbox(
+                label="Prompt",
+                placeholder="e.g. a serene sunset over a coral reef, digital painting",
+                lines=2
+            )
+            resolution_in = gr.Dropdown(
+                choices=["512x512", "768x768", "1024x1024"],
+                value="1024x1024",
+                label="Resolution"
+            )
+            guidance_in = gr.Slider(
+                minimum=0.0, maximum=20.0, step=0.5,
+                value=5.0, label="Guidance Scale"
+            )
+            steps_in = gr.Slider(
+                minimum=1, maximum=100, step=1,
+                value=50, label="Num Inference Steps"
+            )
+            shift_in = gr.Slider(
+                minimum=0, maximum=10, step=1,
+                value=3, label="Shift"
+            )
+            generate_btn = gr.Button("Generate Image", variant="primary")
+
+        with gr.Column(scale=1):
+            output_img = gr.Image(label="Generated Image", interactive=False)
+
+    # Wire up the button
+    generate_btn.click(
+        fn=generate_image,
+        inputs=[prompt_in, resolution_in, guidance_in, steps_in, shift_in],
+        outputs=output_img
+    )
+
+if __name__ == "__main__":
+    demo.launch()