Upload app.py with huggingface_hub

app.py

@@ -1019,10 +1019,31 @@ def do_polish_prompt(prompt: str, style: str, do_polish: bool, mode: str = "gene
     return final_prompt, polished
 
 def do_polish_transform_prompt(prompt: str, style: str, do_polish: bool) -> Tuple[str, str]:
-    """Polish prompt for transformation (style-focused)."""
+    """Polish prompt for transformation (style-focused).
+    
+    When a style is selected without a prompt, the style suffix contains all the 
+    information needed to guide the transformation. No prompt is required.
+    """
+    style_suffix = STYLE_SUFFIXES.get(style, "")
+    has_prompt = prompt and prompt.strip()
+    has_style = style and style != "None" and style_suffix
+    
+    # Style-only transformation: use style as the complete transformation guide
+    if not has_prompt and has_style:
+        # Remove leading comma and space from suffix to make it a primary prompt
+        style_prompt = style_suffix.lstrip(", ").strip()
+        display = f"[{style} Style] {style_prompt[:100]}..." if len(style_prompt) > 100 else f"[{style} Style] {style_prompt}"
+        logger.info(f"do_polish_transform_prompt: Style-only transform with {style}")
+        return style_prompt, display
+    
+    # No prompt and no style - use generic enhancement
+    if not has_prompt and not has_style:
+        return "high quality image, enhanced details, professional quality", ""
+    
+    # Has prompt - proceed with normal flow
     if not do_polish:
-        base = prompt.strip() if prompt else "high quality image"
-        final = base + STYLE_SUFFIXES.get(style, "")
+        base = prompt.strip()
+        final = base + style_suffix
         return final, ""
 
     return do_polish_prompt(prompt, style, True, mode="transform")
@@ -1075,8 +1096,13 @@ def generate_with_polish(prompt: str, style: str, do_polish: bool, ratio: str, s
 def transform_with_polish(input_image: Optional[Image.Image], prompt: str, style: str, do_polish: bool, strength: float, steps: int, seed: int, randomize: bool):
     """Unified transform with progress feedback using generator.
     Yields intermediate status updates with timer so user knows what's happening.
+    
+    Style-only transformation: When a style is selected without a prompt, the style 
+    suffix contains all the information needed to guide the transformation.
     """
-    logger.info(f"transform_with_polish: do_polish={do_polish}, style={style}, prompt_len={len(prompt) if prompt else 0}")
+    has_prompt = prompt and prompt.strip()
+    has_style = style and style != "None"
+    logger.info(f"transform_with_polish: do_polish={do_polish}, style={style}, has_prompt={has_prompt}, has_style={has_style}")
     
     # Start timer
     timer = GenerationTimer()
@@ -1086,8 +1112,11 @@ def transform_with_polish(input_image: Optional[Image.Image], prompt: str, style
         yield None, create_status_html("Please upload an image first", timer.format(), is_generating=False).replace("✅", "❌"), 0
         return
     
-    # Always yield initial status with animation
-    if do_polish:
+    # Show appropriate initial status based on transformation type
+    if not has_prompt and has_style:
+        # Style-only transformation
+        yield None, create_status_html(f"Applying {style} style transformation", timer.format()), 0
+    elif do_polish and has_prompt:
         yield None, create_status_html("Enhancing prompt with DeepSeek Reasoner", timer.format()), 0
     else:
         yield None, create_status_html("Preparing transformation", timer.format()), 0
@@ -2053,7 +2082,7 @@ with gr.Blocks(title="Z Image Turbo", css=css, theme=dark_theme) as demo:
             with gr.Row():
                 with gr.Column(scale=2):
                     trans_input = gr.Image(label="Upload Image", type="pil", height=300)
-                    trans_prompt = gr.Textbox(label="Transformation Prompt", placeholder="e.g., 'oil painting style, vibrant colors'", lines=3)
+                    trans_prompt = gr.Textbox(label="Transformation Prompt (optional if style selected)", placeholder="Optional: describe changes, or just select a Style below (Van Gogh, Picasso, etc.)", lines=3)
                     trans_polish = gr.Checkbox(label="Prompt+ by deepseek-reasoner", value=False)
                     with gr.Row():
                         trans_style = gr.Dropdown(choices=STYLES, value="None", label="Style")