Add Gradio image generator app

app.py

@@ -0,0 +1,138 @@
+import os
+import requests
+import gradio as gr
+
+API_BASE_URL = os.getenv("SOLRICKS_API_BASE_URL", "https://api.solricks.com")
+API_KEY = os.getenv("SOLRICKS_SPACE_API_KEY", "")
+
+API_ENDPOINT = f"{API_BASE_URL}/api/workflows/visual-ai/run"
+
+ALLOWED_RESOLUTIONS = ["1024x1024", "768x1344", "1344x768"]
+ALLOWED_STYLES = ["Portrait", "Realistic", "Cinematic", "Product"]
+
+
+def generate_image(prompt, style, resolution):
+    prompt = (prompt or "").strip()
+
+    if not prompt:
+        return None, "Please enter a prompt."
+
+    if len(prompt) > 500:
+        return None, "Prompt is too long. Please keep it under 500 characters."
+
+    if resolution not in ALLOWED_RESOLUTIONS:
+        return None, "Invalid resolution."
+
+    if style not in ALLOWED_STYLES:
+        return None, "Invalid style."
+
+    payload = {
+        "prompt": prompt,
+        "style": style,
+        "resolution": resolution,
+        "enhancePrompt": False,
+        "seedMode": "Random",
+        "source": "huggingface-space",
+    }
+
+    headers = {
+        "Content-Type": "application/json",
+    }
+
+    if API_KEY:
+        headers["Authorization"] = f"Bearer {API_KEY}"
+
+    try:
+        response = requests.post(
+            API_ENDPOINT,
+            json=payload,
+            headers=headers,
+            timeout=240,
+        )
+
+        try:
+            data = response.json()
+        except Exception:
+            return None, "Solricks backend returned an invalid response."
+
+        if response.status_code == 429:
+            return None, "Too many requests. Please wait a few minutes and try again."
+
+        if not response.ok:
+            return None, data.get("error") or data.get("detail") or "Generation failed."
+
+        image_url = data.get("image_url")
+
+        if not image_url:
+            return None, "No image was returned."
+
+        if image_url.startswith("/"):
+            image_url = f"{API_BASE_URL}{image_url}"
+
+        return image_url, "Done."
+
+    except requests.exceptions.Timeout:
+        return None, "Generation timed out. Please try again."
+    except Exception as error:
+        return None, f"Error: {error}"
+
+
+with gr.Blocks(title="SOLRICKS Image Generator") as demo:
+    gr.Markdown(
+        """
+        # SOLRICKS Image Generator
+
+        Free visual AI preview powered by the Solricks backend.
+
+        This public demo is rate-limited and intended for short preview generations.
+        """
+    )
+
+    prompt = gr.Textbox(
+        label="Prompt",
+        placeholder="Describe the image you want to generate...",
+        lines=4,
+        max_lines=6,
+    )
+
+    with gr.Row():
+        style = gr.Dropdown(
+            label="Style",
+            choices=ALLOWED_STYLES,
+            value="Portrait",
+        )
+
+        resolution = gr.Dropdown(
+            label="Resolution",
+            choices=ALLOWED_RESOLUTIONS,
+            value="1024x1024",
+        )
+
+    generate_button = gr.Button("Generate Image")
+
+    output_image = gr.Image(
+        label="Generated Image",
+        type="filepath",
+    )
+
+    status = gr.Textbox(
+        label="Status",
+        interactive=False,
+    )
+
+    gr.Examples(
+        examples=[
+            ["cinematic portrait of a futuristic woman, dramatic lighting, realistic details", "Portrait", "768x1344"],
+            ["luxury product photo of a perfume bottle, black background, studio lighting", "Product", "1024x1024"],
+            ["realistic AI robot in a dark premium technology lab, cinematic lighting", "Cinematic", "1344x768"],
+        ],
+        inputs=[prompt, style, resolution],
+    )
+
+    generate_button.click(
+        fn=generate_image,
+        inputs=[prompt, style, resolution],
+        outputs=[output_image, status],
+    )
+
+demo.queue(max_size=10).launch()