update

app.py

@@ -22,6 +22,26 @@ DEFAULT_NEGATIVE_PROMPT = "bad quality, worst quality, low resolution, blur, dis
 # Cache for loaded pipe
 pipe_cache = None
 
+# Função auxiliar para redimensionar mantendo aspect ratio
+def smart_resize(image, target_long_edge=1024):
+    width, height = image.size
+    
+    # Calcular nova proporção mantendo o aspect ratio
+    if width > height:
+        new_width = target_long_edge
+        new_height = int(height * (target_long_edge / width))
+    else:
+        new_height = target_long_edge
+        new_width = int(width * (target_long_edge / height))
+    
+    # Arredondar para múltiplos de 32 (necessário para o modelo)
+    new_width = (new_width // 32) * 32
+    new_height = (new_height // 32) * 32
+    
+    # Redimensionar usando LANCZOS para alta qualidade
+    resized_image = image.resize((new_width, new_height), Image.Resampling.LANCZOS)
+    return resized_image, new_width, new_height
+
 @spaces.GPU(duration=300)
 def face_swap(
     body_image,
@@ -127,8 +147,19 @@ def face_swap(
             print(f"Active LoRAs: {adapters} with weights {adapter_weights}")
         
         # Prepare images
-        body_img = Image.fromarray(body_image).convert("RGB")
-        face_img = Image.fromarray(face_image).convert("RGB")
+        body_img_pil = Image.fromarray(body_image).convert("RGB")
+        face_img_pil = Image.fromarray(face_image).convert("RGB")
+        
+        # --- LÓGICA DE REDIMENSIONAMENTO INTELIGENTE ---
+        # Define o tamanho alvo baseado no maior lado (1024 é um bom equilíbrio, pode subir para 1280)
+        # Isso corrige a distorção mantendo o aspect ratio correto
+        TARGET_RESOLUTION = 1024 
+        
+        body_resized, target_w, target_h = smart_resize(body_img_pil, target_long_edge=TARGET_RESOLUTION)
+        # Opcional: redimensionar a face para não ficar gigante ou minúscula comparada ao corpo
+        face_resized, _, _ = smart_resize(face_img_pil, target_long_edge=TARGET_RESOLUTION)
+        
+        print(f"Original size: {body_img_pil.size} | Generation Target: {target_w}x{target_h}")
         
         # Combine fixed prompt with any additional instructions
         final_prompt = FIXED_PROMPT
@@ -137,20 +168,18 @@ def face_swap(
         
         print(f"Using prompt: {final_prompt[:100]}...")
         
-        # --- CORREÇÃO PRINCIPAL AQUI ---
-        # Qwen Image Edit aceita uma lista de imagens.
-        # Index 0 = Picture 1 (Corpo), Index 1 = Picture 2 (Rosto)
-        input_images_list = [body_img, face_img]
+        # Qwen Image Edit uses a list for inputs: [body, face]
+        input_images_list = [body_resized, face_resized]
         
         # Generate the head swap
         result = pipe(
-            image=input_images_list,  # Passamos a lista aqui
+            image=input_images_list,
             prompt=final_prompt,
             negative_prompt=DEFAULT_NEGATIVE_PROMPT, 
-            true_cfg_scale=guidance_scale, 
-            # input_image removido pois não existe neste pipeline
+            true_cfg_scale=guidance_scale,
+            height=target_h,  # FORÇA A ALTURA CORRETA
+            width=target_w,   # FORÇA A LARGURA CORRETA
             num_inference_steps=num_inference_steps,
-            #guidance_scale=guidance_scale,
             generator=generator
         ).images[0]
         
@@ -163,13 +192,12 @@ def face_swap(
         if enable_skin_lora and skin_lora_scale > 0:
             active_loras.append(f"Skin({skin_lora_scale:.2f})")
         
-        status = f"✅ Head swap completed | Active LoRAs: {', '.join(active_loras) if active_loras else 'None'}"
+        status = f"✅ Head swap completed ({target_w}x{target_h}) | Active LoRAs: {', '.join(active_loras) if active_loras else 'None'}"
         
         return result, status
         
     except Exception as e:
         print(f"Error: {str(e)}")
-        # Return a blank red image on error so UI doesn't break
         error_img = Image.new('RGB', (512, 512), color=(200, 50, 50))
         return error_img, f"❌ Error: {str(e)}"
 
@@ -177,8 +205,10 @@ def face_swap(
 with gr.Blocks(title="BFS-Best Face Swap with Qwen", theme=gr.themes.Soft(), css="""
     .container {max-width: 1200px; margin: auto;}
     .image-container {border-radius: 10px; border: 2px dashed #ccc;}
-    .fixed-prompt {background-color: #000000; padding: 10px; border-radius: 5px; font-family: monospace;}
-    .lora-info {background-color: #000000; padding: 8px; border-radius: 5px; margin: 5px 0; font-size: 0.9em;}
+    .fixed-prompt {background-color: #000000; padding: 10px; border-radius: 5px; font-family: monospace; color: #00ff00;}
+    .lora-info {background-color: #000000; padding: 8px; border-radius: 5px; margin: 5px 0; font-size: 0.9em; color: white;}
+    .footer-link {text-decoration: none !important; color: #5865F2 !important; font-weight: bold;}
+    .footer-link:hover {text-decoration: underline !important;}
 """) as demo:
     gr.Markdown(
         """
@@ -274,7 +304,7 @@ with gr.Blocks(title="BFS-Best Face Swap with Qwen", theme=gr.themes.Soft(), css
                     minimum=10,
                     maximum=100,
                     step=5,
-                    value=20,
+                    value=30,
                     label="Inference Steps",
                     info="Higher = better quality but slower"
                 )
@@ -283,8 +313,8 @@ with gr.Blocks(title="BFS-Best Face Swap with Qwen", theme=gr.themes.Soft(), css
                     minimum=1.0,
                     maximum=20.0,
                     step=0.5,
-                    value=2.5,
-                    label="Guidance Scale",
+                    value=5.0,
+                    label="Guidance Scale (CFG)",
                     info="How closely to follow the prompt"
                 )
                 
@@ -342,67 +372,6 @@ with gr.Blocks(title="BFS-Best Face Swap with Qwen", theme=gr.themes.Soft(), css
         outputs=skin_lora_scale
     )
     
-    # # Examples
-    # gr.Examples(
-    #     examples=[
-    #         [
-    #             None,  # body_image
-    #             None,  # face_image
-    #             "",  # custom_prompt_addon
-    #             1.0,  # bfs_lora_scale
-    #             0.7,  # angles_lora_scale
-    #             0.6,  # skin_lora_scale
-    #             True,  # enable_angles_lora
-    #             True,  # enable_skin_lora
-    #             30,  # num_inference_steps
-    #             7.5,  # guidance_scale
-    #             42  # seed
-    #         ],
-    #         [
-    #             None,
-    #             None,
-    #             "professional lighting, high resolution",
-    #             1.2,
-    #             0.8,
-    #             0.5,
-    #             True,
-    #             True,
-    #             40,
-    #             8.0,
-    #             123
-    #         ],
-    #         [
-    #             None,
-    #             None,
-    #             "",
-    #             0.9,
-    #             0.0,
-    #             0.0,
-    #             False,
-    #             False,
-    #             25,
-    #             7.0,
-    #             -1
-    #         ]
-    #     ],
-    #     inputs=[
-    #         body_image,
-    #         face_image,
-    #         custom_prompt_addon,
-    #         bfs_lora_scale,
-    #         angles_lora_scale,
-    #         skin_lora_scale,
-    #         enable_angles_lora,
-    #         enable_skin_lora,
-    #         num_inference_steps,
-    #         guidance_scale,
-    #         seed
-    #     ],
-    #     outputs=[output_image, status_text],
-    #     fn=face_swap,
-    #     cache_examples=False
-    # )
-    
     # Event handlers
     generate_btn.click(
         fn=face_swap,