Update app.py

app.py

@@ -46,6 +46,37 @@ aspect_ratios = {
 }
 
 
+
+def sanitize_seed(seed):
+    """
+    Validate and clamp a seed to int32 max. Returns 0 if invalid.
+
+    Rules:
+    - Accept int-like values (ints, numeric strings).
+    - Must be an integer >= 0 and <= MAX_SEED.
+    - Otherwise return 0.
+    """
+    # Try to coerce from strings/floats that represent integers
+    try:
+        # Handle strings or floats that are integer-valued
+        if isinstance(seed, str):
+            seed = seed.strip()
+            if seed == "":
+                return 0
+            seed_int = int(seed, 10)
+        elif isinstance(seed, (int, np.integer)):
+            seed_int = int(seed)
+        elif isinstance(seed, float) and seed.is_integer():
+            seed_int = int(seed)
+        else:
+            return 0
+    except (ValueError, TypeError):
+        return 0
+
+    if 0 <= seed_int <= MAX_SEED:
+        return seed_int
+    return 0
+
 def polish_prompt_en(original_prompt):
     
     SYSTEM_PROMPT = open("improve_prompt.txt").read()
@@ -85,7 +116,8 @@ def infer(
         
     width, height = aspect_ratios[aspect_ratio]
     
-    print("Generating for prompt:", prompt)
+    print(f"Generating for prompt: \n\t{prompt}\n\t{seed}\n\t{aspect_ratio}\n\t{num_inference_steps}")
+    
     image = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
@@ -93,11 +125,9 @@ def infer(
         height=height,
         num_inference_steps=num_inference_steps,
         true_cfg_scale=guidance_scale,
-        generator=torch.Generator(device="cuda").manual_seed(seed)
+        generator=torch.Generator(device="cuda").manual_seed(sanitize_seed(seed))
     ).images[0]
 
-    #image.save("example.png")
-
     return image, seed
 
 
@@ -106,16 +136,16 @@ def infer(
 css = """
 #col-container {
     margin: 0 auto;
-    max-width: 1024px;
+    max-width: 1920px;
 }
 """
 
 
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        # gr.Markdown('<div style="text-align: center;"><a href="https://huggingface.co/Qwen/Qwen-Image"><img src="https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-Image/qwen_image_logo.png" width="400"/></a></div>')
-        gr.Markdown('<img src="https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-Image/qwen_image_logo.png" alt="your_alt_text" width="400" style="display: block; margin: 0 auto;">')
-        gr.Markdown("[Learn more](https://github.com/QwenLM/Qwen-Image) about the Qwen-Image series. Try on [Qwen Chat](https://chat.qwen.ai/), or [download model](https://huggingface.co/Qwen/Qwen-Image) to run locally with ComfyUI or diffusers.")
+        with gr.Row():
+            gr.Markdown("HINT: Use smaller image size for testing, will consume less of your free GPU time!")
+        
         with gr.Row():
             prompt = gr.Text(
                 label="Prompt",
@@ -137,12 +167,10 @@ with gr.Blocks(css=css) as demo:
                 visible=True,
             )
 
-            seed = gr.Slider(
-                label="Seed",
-                minimum=0,
-                maximum=MAX_SEED,
-                step=1,
-                value=0,
+            seed = gr.Textbox(
+                lines=1
+                label="Manual seed",
+                info="Manual seed, if not a valid 32 bit integer, will be 0."
             )
 
             randomize_seed = gr.Checkbox(label="Randomize seed", value=True)