Update app.py

app.py

@@ -309,13 +309,7 @@ loras = [
         "weights": "50s_western_z_100.safetensors",
         "trigger_word": "50s_western"    
     },
-    {
-        "image": "https://huggingface.co/Sumitc13/Z-image-Turbo_LogC4_lora/resolve/main/images/1764464517272__000005000_1.jpg",
-        "title": "LogC4",
-        "repo": "Sumitc13/Z-image-Turbo_LogC4_lora", #30
-        "weights": "z-image-logc4_000005000.safetensors",
-        "trigger_word": "LogC4"    
-    },
+
     {
         "image": "https://huggingface.co/neph1/80s_scifi_lora_zit/resolve/main/images/ComfyUI_10288_.png",
         "title": "80s Scifi",
@@ -587,20 +581,27 @@ def update_selection(evt: gr.SelectData, width, height):
         updated_text = f"### Selected: Local LoRA - {selected_lora['title']} ✅"
     else:
         updated_text = f"### Selected: [{lora_repo}](https://huggingface.co/{lora_repo}) ✅"
+    
+    # Default aspect ratio
+    aspect = "1:1 (Instagram Square)"
+    width = 1024
+    height = 1024
+    
     if "aspect" in selected_lora:
         if selected_lora["aspect"] == "portrait":
+            aspect = "9:16 (Instagram Reels/TikTok/Shorts)"
             width = 768
-            height = 1024
+            height = 1344
         elif selected_lora["aspect"] == "landscape":
-            width = 1024
+            aspect = "16:9 (YouTube/Twitter/X)"
+            width = 1344
             height = 768
-        else:
-            width = 1024
-            height = 1024
+    
     return (
         gr.update(placeholder=new_placeholder),
         updated_text,
         evt.index,
+        aspect,
         width,
         height,
     )
@@ -806,24 +807,61 @@ with gr.Blocks(title="Z-IMAGE GEN/LORA", delete_cache=(60, 60)) as demo:
             progress_bar = gr.Markdown(elem_id="progress",visible=False)
             result = gr.Image(label="Generated Image", format="png", height=630)
 
+    # SNS Aspect Ratio Presets
+    ASPECT_RATIOS = {
+        "1:1 (Instagram Square)": (1024, 1024),
+        "9:16 (Instagram Reels/TikTok/Shorts)": (768, 1344),
+        "16:9 (YouTube/Twitter/X)": (1344, 768),
+        "4:5 (Instagram Portrait)": (896, 1120),
+        "5:4 (Instagram Landscape)": (1120, 896),
+        "3:4 (Portrait Photo)": (896, 1152),
+        "4:3 (Landscape Photo)": (1152, 896),
+        "2:3 (Pinterest)": (832, 1248),
+        "3:2 (Classic Photo)": (1248, 832),
+        "21:9 (Cinematic Ultra-wide)": (1344, 576),
+        "9:21 (Tall Banner)": (576, 1344),
+    }
+    
+    def update_size(aspect_ratio):
+        width, height = ASPECT_RATIOS.get(aspect_ratio, (1024, 1024))
+        return width, height
+    
     with gr.Row():
-        with gr.Accordion("Advanced Settings", open=False):
+        with gr.Accordion("⚙️ Advanced Settings", open=True):
             with gr.Row():
                 input_image = gr.Image(label="Input image (Ignored for Z-Image-Turbo)", type="filepath", visible=False)
                 image_strength = gr.Slider(label="Denoise Strength", info="Ignored for Z-Image-Turbo", minimum=0.1, maximum=1.0, step=0.01, value=0.75, visible=False)
+            
+            gr.HTML('<div class="info-box">📐 <b>Image Size</b> - Select aspect ratio for different platforms</div>')
+            
+            with gr.Row():
+                aspect_ratio = gr.Dropdown(
+                    choices=list(ASPECT_RATIOS.keys()),
+                    value="1:1 (Instagram Square)",
+                    label="📱 Aspect Ratio (SNS Presets)",
+                    info="Choose the best ratio for your target platform"
+                )
+            
+            with gr.Row():
+                width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
+                height = gr.Slider(label="Height", minimum=256, maximum=1536, step=64, value=1024)
+            
             with gr.Column():
                 with gr.Row():
                     cfg_scale = gr.Slider(label="CFG Scale", info="Forced to 0.0 for Turbo", minimum=0, maximum=20, step=0.5, value=0.0, interactive=False)
                     steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=25)
                 
-                with gr.Row():
-                    width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1536)
-                    height = gr.Slider(label="Height", minimum=256, maximum=1536, step=64, value=1536)
-                
                 with gr.Row():
                     randomize_seed = gr.Checkbox(True, label="Randomize seed")
                     seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
                     lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=3, step=0.01, value=0.95)
+    
+    # Connect aspect ratio dropdown to width/height sliders
+    aspect_ratio.change(
+        fn=update_size,
+        inputs=[aspect_ratio],
+        outputs=[width, height]
+    )
 
     # Footer
     gr.HTML("""
@@ -838,7 +876,7 @@ with gr.Blocks(title="Z-IMAGE GEN/LORA", delete_cache=(60, 60)) as demo:
     gallery.select(
         update_selection,
         inputs=[width, height],
-        outputs=[prompt, selected_info, selected_index, width, height]
+        outputs=[prompt, selected_info, selected_index, aspect_ratio, width, height]
     )
     custom_lora.input(
         add_custom_lora,