Upload 2 files

app.py

@@ -0,0 +1,195 @@
+import os
+import random
+import requests
+import gradio as gr
+from PIL import Image
+from io import BytesIO
+from gradio_client import Client
+
+
+sponsor_html = """
+<div style="display:flex; padding: 0em; justify-content: center; gap: 1em; border-radius: 2em;">
+  <img src="https://static-00.iconduck.com/assets.00/google-cloud-icon-2048x1288-h9qynww8.png"
+       style="height:1em; width:auto; object-fit:contain;"
+       title="Google Cloud for Startups"/>
+  <img src="https://upload.wikimedia.org/wikipedia/commons/thumb/9/93/Amazon_Web_Services_Logo.svg/2560px-Amazon_Web_Services_Logo.svg.png"
+       style="height:1em; width:auto; object-fit:contain;"
+       title="AWS Activate"/>
+  <img src="https://ageyetech.com/wp-content/uploads/2020/07/AgEye_nvidia_inception_logo_new.png"
+       style="height:1em; width:auto; object-fit:contain;"
+       title="NVIDIA Inception"/>
+  <img src="https://azurecomcdn.azureedge.net/cvt-8310f955fa0c7812bd316a20d46a917e5b94170e9e9da481ca3045acae446bb5/svg/logo.svg"
+       style="height:1em; width:auto; object-fit:contain;"
+       title="Azure for Startups"/>
+  <img src="https://upload.wikimedia.org/wikipedia/commons/thumb/4/4b/Cloudflare_Logo.svg/2560px-Cloudflare_Logo.svg.png"
+       style="height:1em; width:auto; object-fit:contain;"
+       title="Cloudflare"/>
+  <img src="https://scaleway.com/cdn-cgi/image/width=640/https://www-uploads.scaleway.com/Scaleway_3_D_Logo_57e7fb833f.png"
+       style="height:1em; width:auto; object-fit:contain;"
+       title="Scaleway"/>
+  <img src="https://cdn.prod.website-files.com/63e26df0d6659968e46142f7/63e27b40e661321d5278519b_logotype-bb8cd083.svg"
+       style="height:1em; width:auto; object-fit:contain;"
+       title="Modal"/>
+  <img src="https://pollinations.ai/favicon.ico"
+       style="height:1em; width:auto; object-fit:contain;"
+       title="Pollination.ai"/>
+</div>
+"""
+
+# more servers coming soon...
+
+
+SERVER_NAMES = {
+    "google_us": "Google US Server",
+    "azure_lite": "Azure Lite Supercomputer Server",
+    "artemis" : "Artemis GPU Super cluster",
+    "nb_dr" : "NebulaDrive Tensor Server",
+    "pixelnet" : "PixelNet NPU Server",
+    "nsfw_core" : "NSFW-Core: Uncensored Server",
+    "nsfw_core_2" : "NSFW-Core: Uncensored Server 2",
+    "nsfw_core_3" : "NSFW-Core: Uncensored Server 3",
+    "nsfw_core_4" : "NSFW-Core: Uncensored Server 4",
+}
+
+
+SERVER_SOCKETS = {
+    "google_us": None,
+    "azure_lite": "FLUX-Pro-SERVER1",
+    "artemis" : "FLUX-Pro-Artemis-GPU",
+    "nb_dr" : "FLUX-Pro-NEBULADRIVE",
+    "pixelnet" : "FLUX-Pro-PIXELNET",
+    "nsfw_core": "FLUX-Pro-NSFW-LocalCoreProcessor",
+    "nsfw_core_2" : "FLUX-Pro-NSFW-LocalCoreProcessor-v2",
+    "nsfw_core_3" : "FLUX-Pro-NSFW-LocalCoreProcessor-v3",
+    "nsfw_core_4" : "FLUX-Pro-NSFW-LocalCoreProcessor-v4",
+}
+
+HF_TOKEN = os.environ.get("HF_TOKEN")
+FLUX_URL  = os.environ.get("FLUX_URL")
+
+
+def _open_image_from_str(s: str):
+    # base64 decoding
+    if s.startswith("http"):
+        r = requests.get(s); return Image.open(BytesIO(r.content))
+    if os.path.exists(s):
+        return Image.open(s)
+    # try base64 blob
+    try:
+        import base64
+        _, b64 = s.split(",", 1)
+        data = base64.b64decode(b64)
+        return Image.open(BytesIO(data))
+    except:
+        raise ValueError(f"Can't parse image string: {s[:30]}…")
+
+
+def generate_image(prompt, width, height, seed, randomize, server_choice):
+
+    print(prompt+"\n\n\n\n")
+    # determine seed
+    if randomize:
+        seed = random.randint(0, 9_999_999)
+    used_seed = seed
+
+    # pick server key and socket
+    key = next(k for k, v in SERVER_NAMES.items() if v == server_choice)
+    socket = SERVER_SOCKETS.get(key)
+
+    # generate image via FLUX or HF space
+    if socket is None:
+        if not FLUX_URL:
+            return "Error: FLUX_URL not set.", used_seed
+        url = (
+            FLUX_URL
+            .replace("[prompt]", prompt)
+            .replace("[w]", str(width))
+            .replace("[h]", str(height))
+            .replace("[seed]", str(seed))
+        )
+        r = requests.get(url)
+        img = Image.open(BytesIO(r.content)) if r.ok else f"FLUX-Pro failed ({r.status_code})"
+    else:
+        space_id = f"NihalGazi/{socket}"
+        client = Client(space_id, hf_token=HF_TOKEN)
+        res = client.predict(
+            prompt=prompt,
+            width=width,
+            height=height,
+            seed=seed,
+            randomize=randomize,
+            api_name="/predict"
+        )
+        if isinstance(res, dict):
+            if res.get("path"):
+                img = Image.open(res["path"])
+            elif res.get("url"):
+                img = _open_image_from_str(res["url"])
+            else:
+                img = "No image found in response."
+        elif isinstance(res, str):
+            img = _open_image_from_str(res)
+        else:
+            img = f"Unexpected response type: {type(res)}"
+
+    # return both image and used seed
+    return img, used_seed
+
+# ─── GRADIO INTERFACE ─────────────────────────────────────────────────────
+with gr.Blocks(theme=gr.themes.Default()) as demo:
+    gr.Markdown(
+        """
+# Unlimited FLUX-Pro
+
+**Enter a prompt and tweak your settings:**  
+- **Width & Height** – choose your canvas size  
+- **Seed** – pick a number or check **Randomize Seed**  
+- **Server** – switch between servers if one is slow or fails:
+  - **Google US Server**  
+  - **Azure Lite Supercomputer Server**  
+  - **Artemis GPU Super cluster**  
+  - **NebulaDrive Tensor Server**  
+  - **PixelNet NPU Server**  
+  - **NSFW‑Core: Uncensored Servers** (for explicit content; use responsibly)
+- **Suggestions** – have ideas? I’m open to them!
+
+⚠️ **Caution:**  
+The **NSFW‑Core** server can generate adult‑only content. You must be of legal age in your jurisdiction and comply with all local laws and platform policies. Developer is not liable for misuse.
+
+
+> ⚡ 4 NSFW Servers available 
+
+
+Click **Generate** and enjoy unlimited AI art!
+
+❤️ **Like & follow** for more AI projects:  
+• Instagram: [@nihal_gazi_io](https://www.instagram.com/nihal_gazi_io/)  
+• Discord: nihal_gazi_io  
+"""
+    )
+
+    # Inputs
+    prompt = gr.Textbox(label="Prompt", placeholder="Enter your image prompt…", lines=4)
+    width  = gr.Slider(512, 2048, step=16, value=1280, label="Width")
+    height = gr.Slider(512, 2048, step=16, value=1280, label="Height")
+    seed   = gr.Number(label="Seed", value=0)
+    rand   = gr.Checkbox(label="Randomize Seed", value=True)
+    server = gr.Dropdown(label="Server", choices=list(SERVER_NAMES.values()),
+                         value=list(SERVER_NAMES.values())[0])
+
+    generate_btn = gr.Button("Generate")
+
+    # Outputs: image and seed display
+    output = gr.Image(type="pil", label="Generated Image")
+    seed_display = gr.Textbox(label="Used Seed", interactive=False)
+
+    generate_btn.click(
+        generate_image,
+        inputs=[prompt, width, height, seed, rand, server],
+        outputs=[output, seed_display]
+    )
+
+    # Sponsor wall
+    gr.HTML(sponsor_html)
+
+demo.launch()

requirements.txt

@@ -0,0 +1 @@
+requests