remove comment

app.py

@@ -100,11 +100,7 @@ def create_prompt(params):
 
 
 def call_client(
-    prompt,
-    seed,
-    randomize_seed,
-    aspect_ratio,
-    num_inference_steps,
+    prompt, seed, randomize_seed, aspect_ratio, num_inference_steps, hf_token=None
 ):
     """
     Call the gradio client for image generation
@@ -124,11 +120,11 @@ def call_client(
         resolution_map = {
             "1:1": "1024x1024 ( 1:1 )",
             "16:9": "1280x720 ( 16:9 )",
-            "4:3": "1152x864 ( 4:3 )",
+            "9:16": "720x1280 ( 9:16 )",
         }
         resolution = resolution_map.get(aspect_ratio, "1024x1024 ( 1:1 )")
 
-        client = Client("Tongyi-MAI/Z-Image-Turbo")
+        client = Client("Tongyi-MAI/Z-Image-Turbo", hf_token=hf_token)
         result = client.predict(
             prompt=prompt,
             resolution=resolution,
@@ -158,27 +154,33 @@ def run_inference_engine(
     aspect_ratio,
     guidance_scale,
     num_inference_steps,
-    hf_token: gr.Header = None,
+    request: gr.Request,
 ):
     """
     generate UI button images
 
     Args:
-        label (str): Text to display on the button
+        label (str): Text to display on the button,allow empty
         detail (str): Detailed design prompt
         shape (str): Button shape ("box", "rounded", "oval", "free")
-        layout (str): Layout arrangement ("horizontal_3", "vertical_2", "box_2x2")
+        layout (str): Layout arrangement ("horizontal_3:3x1", "vertical_2:1x2", "box_2x2:2x2")
         background (str): Background setting ("natural", "white", "black")
         seed (int): Random seed value
         randomize_seed (bool): Whether to randomize the seed
         aspect_ratio (str): Image aspect ratio (use 1:1)
-        guidance_scale (float): Guidance scale (use 1)
-        num_inference_steps (int): Number of inference steps (use 8)
+        guidance_scale (float): Guidance scale (use 1),no need to change
+        num_inference_steps (int): Number of inference steps (use 8),no need to change
 
     Yields:
         tuple: (image, seed, status_message) - Generated image, used seed, status message
     """
-    print(hf_token)
+    hf_token = HF_TOKEN
+    if request:
+        if hasattr(request, "headers"):
+            if hasattr(request.headers, "authorization"):
+                hf_token = request.headers.authorization
+        # print(hf_token)
+
     yield None, seed, "Generating..."
     prompt_params = {
         "label": label,
@@ -190,16 +192,14 @@ def run_inference_engine(
     prompt = create_prompt(prompt_params)
 
     # Debug: Print the generated prompt
-    print(f"Generated prompt: {prompt}")
+    # print(f"Generated prompt: {prompt}")
 
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
     image, generated_seed, error = call_client(
-        prompt, seed, randomize_seed, aspect_ratio, num_inference_steps
+        prompt, seed, randomize_seed, aspect_ratio, num_inference_steps, hf_token
     )
-    print(type(image))
-    print(image)
 
     if image is None:
         gr.Warning(f"Error: {error}")
@@ -302,7 +302,7 @@ example_data = [
     ],
 ]
 
-with gr.Blocks(css=css, theme=theme, title="UI Button Generator") as demo:
+with gr.Blocks(css=css, theme=theme, title="UI Button Generator MCP") as demo:
     gr.Markdown("# 🎨 AI UI Button Generator")
     gr.Markdown(
         "<div class='subtitle'>UI button material and design concept generation tool using Z-Image-Turbo</div>"
@@ -358,7 +358,7 @@ with gr.Blocks(css=css, theme=theme, title="UI Button Generator") as demo:
         randomize_seed = gr.Checkbox(label="Randomize seed", value=True, render=False)
         aspect_ratio = gr.Radio(
             label="Aspect Ratio",
-            choices=["1:1", "16:9", "4:3"],
+            choices=["1:1", "16:9", "9:16"],
             value="1:1",
             render=False,
         )