Upload 3 files

app.py

@@ -125,46 +125,8 @@ def draw_kps(image_pil, kps, color_list=[(255,0,0), (0,255,0), (0,0,255), (255,2
 
     return Image.fromarray(out_img.astype(np.uint8))
 
-def apply_logo_overlay(generated_image, logo_image, style_name):
-    """
-    Applica il logo sull'immagine generata come una "spilla" o "toppa",
-    adattandone l'aspetto in base allo stile scelto.
-    """
-    logo = logo_image.copy().convert("RGBA")
-
-    # Ridimensiona il logo per farlo sembrare una spilla (circa 15% dell'immagine)
-    logo_size = int(generated_image.width * 0.15)
-    logo = logo.resize((logo_size, logo_size), Image.LANCZOS)
-
-    # Applica filtri stilistici al logo in base allo stile
-    logo_np = np.array(logo)
-
-    if "Bit-Builder" in style_name:
-        # Effetto pixelato: rimpicciolisci e riallarga
-        small = cv2.resize(logo_np, (16, 16), interpolation=cv2.INTER_NEAREST)
-        logo_np = cv2.resize(small, (logo_size, logo_size), interpolation=cv2.INTER_NEAREST)
-    elif "Rinascimentale" in style_name:
-        # Effetto dipinto a olio: sfocatura leggera
-        logo_np[:, :, :3] = cv2.bilateralFilter(logo_np[:, :, :3], 9, 75, 75)
-    elif "Architetto" in style_name:
-        # Effetto geometrico: aumento contrasto
-        logo_np[:, :, :3] = np.clip(logo_np[:, :, :3].astype(np.float32) * 1.3, 0, 255).astype(np.uint8)
-    # Pop-Art: logo invariato (già vivace di suo)
-
-    logo_styled = Image.fromarray(logo_np)
-
-    # Posiziona il logo in basso a sinistra (zona petto)
-    x_pos = int(generated_image.width * 0.15)
-    y_pos = int(generated_image.height * 0.70)
-
-    # Componi sul risultato finale
-    result = generated_image.convert("RGBA")
-    result.paste(logo_styled, (x_pos, y_pos), logo_styled)
-
-    return result.convert("RGB")
-
 @spaces.GPU(duration=120)
-def generate_avatar(user_image, logo_image, selected_style):
+def generate_avatar(user_image, selected_style):
     if user_image is None:
         raise gr.Error("Per favore, carica un'immagine del volto.")
 
@@ -219,9 +181,7 @@ def generate_avatar(user_image, logo_image, selected_style):
         except:
             pass
 
-    # 4. Sovrapposizione del logo stilizzato (se fornito)
-    if logo_image is not None:
-        image = apply_logo_overlay(image, logo_image, selected_style)
+
 
     return image
 
@@ -250,12 +210,10 @@ with gr.Blocks(css=custom_css, theme=theme) as demo:
         with gr.Row():
             with gr.Column(scale=1):
                 input_image = gr.Image(type="pil", label="1. Carica il tuo Selfie")
-                logo_image = gr.Image(type="pil", label="2. Carica il Logo (opzionale, PNG a colori)")
-
             with gr.Column(scale=1):
                 style_selector = gr.Radio(
                     choices=list(STYLES.keys()),
-                    label="3. Scegli il tuo Stile",
+                    label="2. Scegli il tuo Stile",
                     value=list(STYLES.keys())[0]
                 )
                 gr.Markdown("<br>")
@@ -266,7 +224,7 @@ with gr.Blocks(css=custom_css, theme=theme) as demo:
 
         generate_button.click(
             fn=generate_avatar,
-            inputs=[input_image, logo_image, style_selector],
+            inputs=[input_image, style_selector],
             outputs=output_image
         )