Update app.py

app.py

@@ -6,6 +6,37 @@ 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...
 
 
@@ -13,8 +44,10 @@ SERVER_NAMES = {
     "google_us": "Google US Server",
     "azure_lite": "Azure Lite Supercomputer Server",
     "artemis" : "Artemis GPU Super cluster",
+    "nb_dr" : "NebulaDrive Tensor Server",
     "nsfw_core" : "NSFW-Core: Uncensored Server",
     "nsfw_core_2" : "NSFW-Core: Uncensored Server 2",
+    "nsfw_core_3" : "NSFW-Core: Uncensored Server 3",
     
 }
 
@@ -23,8 +56,10 @@ SERVER_SOCKETS = {
     "google_us": None,
     "azure_lite": "FLUX-Pro-SERVER1",
     "artemis" : "FLUX-Pro-Artemis-GPU",
+    "nb_dr" : "FLUX-Pro-NEBULADRIVE",
     "nsfw_core": "FLUX-Pro-NSFW-LocalCoreProcessor",
     "nsfw_core_2" : "FLUX-Pro-NSFW-LocalCoreProcessor-v2",
+    "nsfw_core_3" : "FLUX-Pro-NSFW-LocalCoreProcessor-v3",
     
 }
 
@@ -101,23 +136,11 @@ def generate_image(prompt, width, height, seed, randomize, server_choice):
 
 # ─── GRADIO INTERFACE ─────────────────────────────────────────────────────
 
-iface = gr.Interface(
-    fn=generate_image,
-    inputs=[
-        gr.Textbox(label="Prompt", placeholder="Enter your image prompt…"),
-        gr.Slider(512, 1280, step=16, value=1280, label="Width"),
-        gr.Slider(512, 1280, step=16, value=1280, label="Height"),
-        gr.Number(label="Seed", value=0),
-        gr.Checkbox(label="Randomize Seed", value=True),
-        gr.Dropdown(
-            label="Server",
-            choices=list(SERVER_NAMES.values()),
-            value=SERVER_NAMES["google_us"]
-        ),
-    ],
-    outputs=gr.Image(type="pil"),
-    title="Unlimited FLUX‑Pro",
-    description = """
+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**  
@@ -125,6 +148,7 @@ iface = gr.Interface(
   - **Google US Server**  
   - **Azure Lite Supercomputer Server**  
   - **Artemis GPU Super cluster**  
+  - **NebulaDrive Tensor Server**
   - **NSFW‑Core: Uncensored Servers** (for explicit content; use responsibly)
 - **Suggestions** – have ideas? I’m open to them!
 
@@ -132,7 +156,7 @@ iface = gr.Interface(
 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.
 
 
-> ⚡ 2 NSFW Servers available 
+> ⚡ 3 NSFW Servers available 
 
 
 Click **Generate** and enjoy unlimited AI art!
@@ -141,7 +165,31 @@ Click **Generate** and enjoy unlimited AI art!
 • Instagram: [@nihal_gazi_io](https://www.instagram.com/nihal_gazi_io/)  
 • Discord: nihal_gazi_io  
 """
-)
+    )
+    # Inputs / Outputs as before
+    
+    prompt = gr.Textbox(label="Prompt", placeholder="Enter your image prompt…", lines=4)
+    width  = gr.Slider(512, 1280, step=16, value=1280, label="Width")
+    height = gr.Slider(512, 1280, 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")
+
+    output = gr.Image(type="pil")
+    
+    generate_btn.click(
+        generate_image,
+        inputs=[prompt, width, height, seed, rand, server],
+        outputs=output
+    )
+
+    
+
+    # Now inject your raw HTML sponsor wall
+    gr.HTML(sponsor_html)
 
-if __name__ == "__main__":
-    iface.launch()
+demo.launch()