Update app.py

app.py

@@ -36,7 +36,6 @@ sponsor_html = """
 </div>
 """
 
-
 # more servers coming soon...
 
 
@@ -50,8 +49,6 @@ SERVER_NAMES = {
     "nsfw_core_2" : "NSFW-Core: Uncensored Server 2",
     "nsfw_core_3" : "NSFW-Core: Uncensored Server 3",
     "nsfw_core_4" : "NSFW-Core: Uncensored Server 4",
-    
-    
 }
 
 
@@ -65,14 +62,12 @@ SERVER_SOCKETS = {
     "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"):
@@ -89,22 +84,20 @@ def _open_image_from_str(s: str):
         raise ValueError(f"Can't parse image string: {s[:30]}…")
 
 
-
 def generate_image(prompt, width, height, seed, randomize, server_choice):
-
-    print("\f\n\n\n\n"+prompt+"\n\n\n\n")
-    
+    # determine seed
     if randomize:
         seed = random.randint(0, 9_999_999)
+    used_seed = seed
 
-    # dict key for this display name
+    # 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."
+            return "Error: FLUX_URL not set.", used_seed
         url = (
             FLUX_URL
             .replace("[prompt]", prompt)
@@ -113,34 +106,34 @@ def generate_image(prompt, width, height, seed, randomize, server_choice):
             .replace("[seed]", str(seed))
         )
         r = requests.get(url)
-        return Image.open(BytesIO(r.content)) if r.ok else f"FLUX-Pro failed ({r.status_code})"
-
-    
-    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"
-    )
-
-    # handle dict or str
-    if isinstance(res, dict):
-        if res.get("path"):
-            return Image.open(res["path"])
-        if res.get("url"):
-            return _open_image_from_str(res["url"])
-        return "No image found in response."
-    elif isinstance(res, str):
-        return _open_image_from_str(res)
+        img = Image.open(BytesIO(r.content)) if r.ok else f"FLUX-Pro failed ({r.status_code})"
     else:
-        return f"Unexpected response type: {type(res)}"
+        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(
         """
@@ -153,8 +146,8 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
   - **Google US Server**  
   - **Azure Lite Supercomputer Server**  
   - **Artemis GPU Super cluster**  
-  - **NebulaDrive Tensor Server**
-  - **PixelNet NPU Server**
+  - **NebulaDrive Tensor Server**  
+  - **PixelNet NPU Server**  
   - **NSFW‑Core: Uncensored Servers** (for explicit content; use responsibly)
 - **Suggestions** – have ideas? I’m open to them!
 
@@ -172,8 +165,8 @@ Click **Generate** and enjoy unlimited AI art!
 • Discord: nihal_gazi_io  
 """
     )
-    # Inputs / Outputs as before
-    
+
+    # Inputs
     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")
@@ -182,20 +175,19 @@ Click **Generate** and enjoy unlimited AI art!
     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")
-    
+    # 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
+        outputs=[output, seed_display]
     )
 
-    
-
-    # Now inject your raw HTML sponsor wall
+    # Sponsor wall
     gr.HTML(sponsor_html)
 
-demo.launch()
+demo.launch()