Update app.py

app.py

@@ -10,28 +10,19 @@ fa = face_alignment.FaceAlignment('2D', device='cpu', flip_input=False)
 
 def process_mask(mask_img):
     """Automatically remove white/transparent background from mask image"""
-    # Convert to RGBA if not already
     mask_img = mask_img.convert("RGBA")
-    
-    # Get image data
     datas = mask_img.getdata()
     new_data = []
-    
-    # Define white threshold (can adjust if needed)
     white_threshold = 240
     
     for item in datas:
-        # If pixel is mostly white (all channels > threshold) or already transparent
         if (item[0] > white_threshold and 
             item[1] > white_threshold and 
             item[2] > white_threshold) or item[3] == 0:
-            # Make fully transparent
             new_data.append((255, 255, 255, 0))
         else:
-            # Keep the pixel as is
             new_data.append(item)
     
-    # Update image data
     mask_img.putdata(new_data)
     return mask_img
 
@@ -45,7 +36,7 @@ def fit_mask_to_face(image):
     landmarks = fa.get_landmarks(img_np)
 
     if not landmarks:
-        return image  # Return original image if no face detected
+        return image
 
     image_rgb = cv2.cvtColor(img_np, cv2.COLOR_BGR2RGB)
 
@@ -73,39 +64,26 @@ def fit_mask_to_face(image):
         return background_pil.convert("RGB")
     return image
 
-# Rest of the code remains the same...
-
 def capture_and_process():
     """Capture image from webcam and process it"""
-    # Create a temporary file
     temp_file = tempfile.NamedTemporaryFile(suffix=".jpg", delete=False)
     temp_path = temp_file.name
     
-    # Open camera
     cap = cv2.VideoCapture(0)
-    
-    # Capture frame
     ret, frame = cap.read()
     
     if ret:
-        # Save captured frame
         cv2.imwrite(temp_path, frame)
-        
-        # Convert to PIL Image
         pil_image = Image.open(temp_path)
-        
-        # Process the image
         result = fit_mask_to_face(pil_image)
-        
-        # Release camera
         cap.release()
-        
         return result
     else:
         cap.release()
         return None
 
-with gr.Blocks(css=css) as demo:
+# Create interface without CSS
+with gr.Blocks() as demo:
     gr.Markdown("# 🎭 Party Face Mask App")
     gr.Markdown("Upload an image or capture from your webcam to try on our party mask!")
     
@@ -113,19 +91,18 @@ with gr.Blocks(css=css) as demo:
         with gr.TabItem("📁 Upload Image"):
             with gr.Row():
                 with gr.Column():
-                    upload_input = gr.Image(label="Upload your photo", type="pil", elem_classes="image-container")
+                    upload_input = gr.Image(label="Upload your photo", type="pil")
                     upload_button = gr.Button("Apply Mask")
                 with gr.Column():
-                    upload_output = gr.Image(label="Result", type="pil", elem_classes="image-container")
+                    upload_output = gr.Image(label="Result", type="pil")
             upload_button.click(fn=fit_mask_to_face, inputs=upload_input, outputs=upload_output)
             
         with gr.TabItem("📸 Webcam"):
             with gr.Row():
                 with gr.Column():
                     webcam_button = gr.Button("Capture Photo from Webcam")
-                    webcam_status = gr.Textbox(label="Status", visible=False)
                 with gr.Column():
-                    webcam_output = gr.Image(label="Result", type="pil", elem_classes="image-container")
+                    webcam_output = gr.Image(label="Result", type="pil")
             webcam_button.click(fn=capture_and_process, outputs=webcam_output)
 
 demo.launch()