Upload 2 files

app.py

@@ -0,0 +1,193 @@
+import gradio as gr
+import requests
+import io
+from PIL import Image
+
+# ── Models ──────────────────────────────────────────────────────────────────
+MODELS = {
+    "⚡ FLUX Schnell (fastest)":        "black-forest-labs/FLUX.1-schnell",
+    "🎨 SDXL Base (high quality)":      "stabilityai/stable-diffusion-xl-base-1.0",
+    "🖼  SD 1.5 (classic)":             "runwayml/stable-diffusion-v1-5",
+    "🌅 Openjourney (Midjourney style)": "prompthero/openjourney-v4",
+}
+
+# ── Style presets ────────────────────────────────────────────────────────────
+STYLES = {
+    "None":        "",
+    "Cinematic":   ", cinematic lighting, film grain, anamorphic lens, dramatic shadows",
+    "Anime":       ", anime style, vibrant colors, studio ghibli, detailed illustration",
+    "Oil Painting": ", oil painting, textured brushstrokes, classical art, museum quality",
+    "Neon Noir":   ", neon noir, rain-slicked streets, cyberpunk, moody atmosphere",
+    "Soft Dream":  ", soft focus, dreamy, pastel tones, ethereal, bokeh",
+    "Raw Photo":   ", photorealistic, RAW photo, DSLR, 85mm lens, sharp focus",
+}
+
+# ── Aspect ratios ────────────────────────────────────────────────────────────
+RATIOS = {
+    "1:1  (512×512)":  (512, 512),
+    "3:2  (768×512)":  (768, 512),
+    "2:3  (512×768)":  (512, 768),
+    "16:9 (896×512)":  (896, 512),
+}
+
+HF_API = "https://api-inference.huggingface.co/models/"
+
+# ─────────────────────────────────────────────────────────────────────────────
+
+def generate(
+    hf_token: str,
+    prompt: str,
+    negative_prompt: str,
+    model_label: str,
+    style_label: str,
+    ratio_label: str,
+    steps: int,
+    guidance: float,
+):
+    if not hf_token.strip():
+        raise gr.Error("🔑 Please enter your Hugging Face API token.")
+    if not prompt.strip():
+        raise gr.Error("✏️  Please enter a prompt.")
+
+    model_id    = MODELS[model_label]
+    style_tag   = STYLES[style_label]
+    width, height = RATIOS[ratio_label]
+    full_prompt = prompt + style_tag
+
+    headers  = {"Authorization": f"Bearer {hf_token.strip()}"}
+    payload  = {"inputs": full_prompt}
+
+    # FLUX Schnell doesn't support guidance_scale / negative_prompt
+    if "FLUX" in model_label:
+        payload["parameters"] = {
+            "num_inference_steps": steps,
+            "width": width,
+            "height": height,
+        }
+    else:
+        payload["parameters"] = {
+            "negative_prompt":    negative_prompt,
+            "num_inference_steps": steps,
+            "guidance_scale":     guidance,
+            "width":  width,
+            "height": height,
+        }
+
+    response = requests.post(HF_API + model_id, headers=headers, json=payload, timeout=120)
+
+    if response.status_code == 503:
+        raise gr.Error("⏳ Model is loading — wait ~20 seconds then try again.")
+    if not response.ok:
+        try:
+            detail = response.json().get("error", response.text)
+        except Exception:
+            detail = response.text
+        raise gr.Error(f"API error {response.status_code}: {detail}")
+
+    image = Image.open(io.BytesIO(response.content))
+    return image, full_prompt
+
+
+# ── UI ───────────────────────────────────────────────────────────────────────
+
+css = """
+#title  { text-align: center; margin-bottom: 4px; }
+#sub    { text-align: center; color: #888; margin-bottom: 24px; }
+footer  { display: none !important; }
+"""
+
+with gr.Blocks(title="dreamforge · AI Image Generator", css=css, theme=gr.themes.Default(
+    primary_hue="yellow",
+    neutral_hue="slate",
+    font=gr.themes.GoogleFont("Syne"),
+)) as demo:
+
+    gr.Markdown("# ✦ dreamforge", elem_id="title")
+    gr.Markdown("AI image generation powered by Hugging Face", elem_id="sub")
+
+    with gr.Row():
+        # ── Left column: controls ──────────────────────────────────────────
+        with gr.Column(scale=1):
+            hf_token = gr.Textbox(
+                label="🔑 Hugging Face Token",
+                placeholder="hf_xxxxxxxxxxxxxxxxxxxxxxxx",
+                type="password",
+                info="Get yours free at huggingface.co/settings/tokens",
+            )
+
+            prompt = gr.Textbox(
+                label="✏️  Prompt",
+                placeholder="A misty mountain at golden hour, oil on canvas…",
+                lines=4,
+            )
+
+            negative_prompt = gr.Textbox(
+                label="🚫 Negative Prompt",
+                value="blurry, low quality, watermark, text, ugly, deformed",
+                lines=2,
+            )
+
+            model_label = gr.Radio(
+                label="🤖 Model",
+                choices=list(MODELS.keys()),
+                value=list(MODELS.keys())[0],
+            )
+
+            with gr.Row():
+                style_label = gr.Dropdown(
+                    label="🎨 Style Preset",
+                    choices=list(STYLES.keys()),
+                    value="None",
+                )
+                ratio_label = gr.Dropdown(
+                    label="📐 Aspect Ratio",
+                    choices=list(RATIOS.keys()),
+                    value="1:1  (512×512)",
+                )
+
+            with gr.Row():
+                steps = gr.Slider(
+                    label="Steps",
+                    minimum=1, maximum=50, step=1, value=4,
+                    info="More steps = more detail (slower)"
+                )
+                guidance = gr.Slider(
+                    label="Guidance Scale",
+                    minimum=1.0, maximum=20.0, step=0.5, value=7.5,
+                    info="How closely to follow the prompt"
+                )
+
+            generate_btn = gr.Button("✦ Generate", variant="primary", size="lg")
+
+        # ── Right column: output ───────────────────────────────────────────
+        with gr.Column(scale=1):
+            output_image = gr.Image(
+                label="Generated Image",
+                type="pil",
+                height=520,
+            )
+            used_prompt = gr.Textbox(
+                label="Full prompt sent to model",
+                interactive=False,
+                lines=2,
+            )
+
+    # ── Examples ──────────────────────────────────────────────────────────
+    gr.Examples(
+        examples=[
+            ["", "A lone astronaut on a coral reef planet, bioluminescent sea creatures, epic scale", "", "⚡ FLUX Schnell (fastest)", "Cinematic", "1:1  (512×512)", 4, 7.5],
+            ["", "Portrait of a Victorian explorer, oil paint, dramatic candlelight", "",             "🎨 SDXL Base (high quality)", "Oil Painting", "2:3  (512×768)", 30, 8.0],
+            ["", "Neon-lit Tokyo alley at 3am, rain reflections, cat sitting on a barrel", "",        "🖼  SD 1.5 (classic)", "Neon Noir", "3:2  (768×512)", 20, 7.5],
+        ],
+        inputs=[hf_token, prompt, negative_prompt, model_label, style_label, ratio_label, steps, guidance],
+        label="✨ Example Prompts",
+    )
+
+    generate_btn.click(
+        fn=generate,
+        inputs=[hf_token, prompt, negative_prompt, model_label, style_label, ratio_label, steps, guidance],
+        outputs=[output_image, used_prompt],
+    )
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio>=4.0.0
+requests
+Pillow