Update app.py

app.py

@@ -1,15 +1,10 @@
-
-import sys
-if 'google.colab' in sys.modules:
-  !pip install deepface nudenet opencv-python
-
 from PIL import Image, ImageDraw, ImageFont
 import numpy as np
 import gradio as gr
 import os
-import cv2
-from deepface import DeepFace
-from nudenet import NudeDetector
+import cv2 # Explicitly adding cv2 import
+from deepface import DeepFace # Explicitly adding DeepFace import
+from nudenet import NudeDetector # Explicitly adding NudeDetector import
 import io
 import tempfile
 
@@ -202,7 +197,7 @@ def estimate_age(crop_img):
         return "Alter nicht bestimmbar"
 
 # Initialize NudeDetector
-detector = NudeDetector()
+detector = NudeDetector() # NudeDetector instantiation
 
 def analyze_image_with_gradio(input_image_path):
     print(f"[INFO] analyze_image_with_gradio called with path: {input_image_path}")
@@ -366,7 +361,7 @@ def analyze_image_with_gradio(input_image_path):
             bbox_label_global = FONT_LABEL_SIMPLE.getbbox(label_global_simple)
             text_width_global = bbox_label_global[2] - bbox_label_global[0]
             text_height_global = bbox_label_global[3] - bbox_label_global[1]
-            draw_annotated.rectangle([(x1, y1 - text_height_global - 2), (x1 + text_width_global, y1)], fill=(255, 255, 0))
+            draw_annotated.rectangle([(x1, y1 - text_height_global - 2), (x1 + text_width_global, y1)], fill=BACKGROUND_COLOR)
             draw_annotated.text((x1, y1 - text_height_global - 2), label_global_simple, font=FONT_LABEL_SIMPLE, fill=BACKGROUND_COLOR)
 
             crop = original_img.crop((x1, y1, x2, y2))
@@ -403,10 +398,10 @@ def analyze_image_with_gradio(input_image_path):
                         crop = original_img.crop(b_data['bbox'])
                         desc = describe_breast_precise(crop) + f" · Alter: {b_data['age_str']}"
                         current_image_crops_with_details.append((f"Global Brust (einzeln)", resize_crop(crop, TARGET_MAX_DIMENSION), desc))
-            else:
-                merged_breast_crop = original_img.crop((x1_final, y1_final, x2_final, y2_final))
-                desc = describe_breast_precise(merged_breast_crop) + f" · Alter: {global_breasts_for_merging[0]['age_str']}"
-                current_image_crops_with_details.append((f"Global Brüste (beide)", resize_crop(merged_breast_crop, TARGET_MAX_DIMENSION), desc))
+                else:
+                    merged_breast_crop = original_img.crop((x1_final, y1_final, x2_final, y2_final))
+                    desc = describe_breast_precise(merged_breast_crop) + f" · Alter: {global_breasts_for_merging[0]['age_str']}"
+                    current_image_crops_with_details.append((f"Global Brüste (beide)", resize_crop(merged_breast_crop, TARGET_MAX_DIMENSION), desc))
             else:
                 b_data = global_breasts_for_merging[0]
                 crop = original_img.crop(b_data['bbox'])
@@ -561,4 +556,4 @@ interface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    interface.launch(debug=False, share=True)
+    interface.launch(debug=False, share=True)