Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,152 @@
+import os
+import io
+import random
+import gradio as gr
+from huggingface_hub import InferenceClient
+from PIL import Image
+
+# --- Configuration -----------------------------------------------------------
+HF_TOKEN = os.environ.get("HF_TOKEN", None)
+MODEL_ID = "black-forest-labs/FLUX.1-schnell"
+
+client = InferenceClient(model=MODEL_ID, token=HF_TOKEN)
+
+# Dark theme colors
+BG_COLOR = "#0A0F1A"
+ACCENT = "#1FE0E6"
+
+# Aspect ratio -> (width, height)
+ASPECT_RATIOS = {
+    "9:16 (Phone)": (720, 1280),
+    "1:1 (Square)": (1024, 1024),
+    "16:9 (Wide)": (1280, 720),
+}
+
+# Presets: name -> prompt
+PRESETS = {
+    "UAE Sunset": "A breathtaking UAE desert sunset, golden sand dunes rolling under a vibrant orange and purple sky, silhouette of a lone camel, cinematic lighting, ultra detailed, 4k wallpaper",
+    "Cyberpunk City": "A neon-soaked cyberpunk city at night, towering skyscrapers with glowing cyan and magenta lights, flying cars, rain-slicked streets reflecting neon, blade runner aesthetic, ultra detailed wallpaper",
+    "Minimal Dark": "Minimalist dark abstract wallpaper, deep black background with subtle geometric cyan lines, smooth gradients, modern elegant design, 4k",
+    "Arabic Calligraphy": "Beautiful Arabic calligraphy art wallpaper, elegant flowing script in gold on dark textured background, intricate decorative borders, luxurious design, 4k",
+    "Desert Stars": "A serene Arabian desert at night under a sky full of stars, Milky Way galaxy visible, soft sand dunes in foreground, peaceful cosmic atmosphere, ultra detailed wallpaper",
+}
+
+
+# --- Core function -----------------------------------------------------------
+def generate_wallpaper(prompt, aspect, preset):
+    """Generate a wallpaper via FLUX.1-schnell."""
+    # If a preset is selected, use its prompt (override user prompt if empty)
+    if preset and preset in PRESETS:
+        final_prompt = PRESETS[preset]
+    elif prompt and prompt.strip():
+        final_prompt = prompt.strip()
+    else:
+        final_prompt = PRESETS["UAE Sunset"]
+
+    w, h = ASPECT_RATIOS.get(aspect, (1024, 1024))
+
+    try:
+        # FLUX.1-schnell supports image generation via inference
+        image = client.text_to_image(
+            final_prompt,
+            width=w,
+            height=h,
+        )
+        # Ensure PIL Image
+        if not isinstance(image, Image.Image):
+            image = Image.open(image) if hasattr(image, "read") else Image.frombytes("RGB", image.size, image.tobytes()) if hasattr(image, "size") else Image.open(io.BytesIO(image))
+        return image, "✅ Wallpaper generated successfully!"
+    except Exception as e:
+        # Return a placeholder error image
+        img = Image.new("RGB", (w, h), BG_COLOR)
+        return img, f"❌ Error: {str(e)}"
+
+
+def download_image(img):
+    """Provide downloadable file path."""
+    if img is None:
+        return None
+    path = os.path.join(os.getcwd(), "wallpaper_output.png")
+    img.save(path)
+    return path
+
+
+def pick_preset_prompt(preset_name):
+    """Load preset prompt into the text box."""
+    return PRESETS.get(preset_name, "")
+
+
+# --- UI -----------------------------------------------------------------------
+CUSTOM_CSS = f"""
+#main {{background-color: {BG_COLOR};}}
+.gradio-container {{background-color: {BG_COLOR} !important;}}
+h1, h2, h3 {{color: {ACCENT} !important;}}
+"""
+
+with gr.Blocks(
+    title="Dispatch AI Wallpaper Engine",
+    theme=gr.themes.Base(
+        primary_hue="cyan",
+        neutral_hue="slate",
+        font=("Inter", "system-ui", "sans-serif"),
+    ),
+    css=CUSTOM_CSS,
+) as demo:
+    gr.Markdown(
+        f"""
+        <div style="text-align:center;">
+        <h1>Dispatch AI Wallpaper Engine</h1>
+        <p style="color:{ACCENT};">Generate stunning phone wallpapers with FLUX.1-schnell · Powered by Dispatch AI (FZE)</p>
+        </div>
+        """
+    )
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            prompt_input = gr.Textbox(
+                label="Prompt",
+                placeholder="Describe your wallpaper... or pick a preset below",
+                lines=3,
+            )
+            aspect_select = gr.Radio(
+                list(ASPECT_RATIOS.keys()),
+                label="Aspect Ratio",
+                value="9:16 (Phone)",
+            )
+            preset_select = gr.Dropdown(
+                list(PRESETS.keys()),
+                label="Preset",
+                value=None,
+                info="Select a preset to auto-fill the prompt",
+            )
+            load_preset_btn = gr.Button("Load Preset Prompt", variant="secondary")
+            generate_btn = gr.Button("🎨 Generate Wallpaper", variant="primary")
+        with gr.Column(scale=2):
+            output_image = gr.Image(
+                label="Generated Wallpaper",
+                type="pil",
+                show_download_button=True,
+            )
+            status_box = gr.Textbox(label="Status", interactive=False)
+            download_btn = gr.Button("⬇️ Download PNG", variant="secondary")
+            download_file = gr.File(label="Download")
+
+    # Events
+    load_preset_btn.click(pick_preset_prompt, inputs=preset_select, outputs=prompt_input)
+    generate_btn.click(
+        generate_wallpaper,
+        inputs=[prompt_input, aspect_select, preset_select],
+        outputs=[output_image, status_box],
+    )
+    download_btn.click(download_image, inputs=output_image, outputs=download_file)
+
+    gr.Markdown(
+        """
+        <div style="text-align:center; opacity:0.6; padding-top:20px;">
+        <small>Dispatch AI (FZE) · UAE · Model: FLUX.1-schnell by Black Forest Labs</small>
+        </div>
+        """
+    )
+
+if __name__ == "__main__":
+    demo.launch()