Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,153 @@
+import os
+import io
+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)
+
+BG_COLOR = "#0A0F1A"
+ACCENT = "#1FE0E6"
+
+# Calligraphy styles
+STYLES = {
+    "Thuluth": "Thuluth script, elegant large sweeping curves, ornate flowing letters, classical",
+    "Naskh": "Naskh script, clean readable traditional book script, rounded letters, classic",
+    "Diwani": "Diwani script, decorative curved ornamental Ottoman style, intricate overlapping letters",
+    "Kufi": "Kufic script, geometric angular blocky letters, strict grid pattern, ancient architectural style",
+    "Modern": "Modern Arabic calligraphy, contemporary stylized abstract elegant fusion of traditional and modern art",
+}
+
+# Color schemes: name -> (foreground desc, background desc)
+COLOR_SCHEMES = {
+    "Gold on Black": ("gold metallic letters", "deep black textured background"),
+    "White on Blue": ("pure white letters", "rich deep blue gradient background"),
+    "Silver on Dark": ("silver metallic letters", "dark charcoal gradient background"),
+}
+
+# Example prompts in Arabic
+EXAMPLES_AR = [
+    "بسم الله الرحمن الرحيم",
+    "السلام عليكم ورحمة الله وبركاته",
+    "الحب للوطن من اليمان",
+    "العلم نور",
+    "صبر جميل",
+    "توكل على الله",
+    " Dispatch AI ",
+]
+
+
+# --- Core function -----------------------------------------------------------
+def build_prompt(arabic_text, style, color_scheme):
+    """Construct a detailed prompt for FLUX."""
+    style_desc = STYLES.get(style, STYLES["Thuluth"])
+    fg, bg = COLOR_SCHEMES.get(color_scheme, COLOR_SCHEMES["Gold on Black"])
+
+    text = arabic_text.strip() if arabic_text.strip() else "بسم الله"
+
+    prompt = (
+        f"Arabic calligraphy art of the text '{text}', "
+        f"rendered in {style_desc}, "
+        f"with {fg} on {bg}, "
+        f"centered composition, elegant ornamental frame, "
+        f"high detail, ultra sharp, 4k, luxurious Islamic art style"
+    )
+    return prompt
+
+
+def generate_calligraphy(arabic_text, style, color_scheme):
+    """Generate Arabic calligraphy art via FLUX.1-schnell."""
+    prompt = build_prompt(arabic_text, style, color_scheme)
+    w, h = 1024, 1024
+
+    try:
+        image = client.text_to_image(prompt, width=w, height=h)
+        if not isinstance(image, Image.Image):
+            image = Image.open(io.BytesIO(image)) if hasattr(image, "read") else Image.open(image)
+        return image, "✅ Calligraphy generated!", prompt
+    except Exception as e:
+        img = Image.new("RGB", (w, h), BG_COLOR)
+        return img, f"❌ Error: {str(e)}", prompt
+
+
+# --- UI -----------------------------------------------------------------------
+CUSTOM_CSS = f"""
+.gradio-container {{background-color: {BG_COLOR} !important;}}
+h1, h2, h3 {{color: {ACCENT} !important;}}
+"""
+
+with gr.Blocks(
+    title="Dispatch AI Arabic Calligraphy",
+    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 Arabic Calligraphy</h1>
+        <p style="color:{ACCENT};">Generate beautiful Arabic calligraphy art with FLUX.1-schnell · Dispatch AI (FZE)</p>
+        </div>
+        """
+    )
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            arabic_input = gr.Textbox(
+                label="Arabic Text",
+                placeholder="اكتب النص هنا",
+                value="بسم الله الرحمن الرحيم",
+                lines=2,
+                rtl=True,
+            )
+            style_select = gr.Dropdown(
+                list(STYLES.keys()),
+                label="Calligraphy Style",
+                value="Thuluth",
+            )
+            color_select = gr.Dropdown(
+                list(COLOR_SCHEMES.keys()),
+                label="Color Scheme",
+                value="Gold on Black",
+            )
+            generate_btn = gr.Button("✨ Generate Calligraphy", variant="primary")
+        with gr.Column(scale=2):
+            output_image = gr.Image(
+                label="Generated Calligraphy",
+                type="pil",
+                show_download_button=True,
+            )
+            status_box = gr.Textbox(label="Status", interactive=False)
+            prompt_box = gr.Textbox(label="Generated Prompt (for debugging)", interactive=False, lines=3)
+
+    gr.Markdown("### Example Arabic Prompts")
+    gr.Examples(
+        examples=[[ex] for ex in EXAMPLES_AR],
+        inputs=arabic_input,
+        label="Click an example to load it into the Arabic text box",
+    )
+
+    # Events
+    generate_btn.click(
+        generate_calligraphy,
+        inputs=[arabic_input, style_select, color_select],
+        outputs=[output_image, status_box, prompt_box],
+    )
+
+    gr.Markdown(
+        """
+        <div style="text-align:center; opacity:0.6; padding-top:20px;">
+        <small>Dispatch AI (FZE) · UAE · Model: FLUX.1-schnell · Arabic Calligraphy Engine</small>
+        </div>
+        """
+    )
+
+if __name__ == "__main__":
+    demo.launch()