Update app.py

app.py

@@ -13,7 +13,7 @@ def parse_list_boxes(text):
     matches = re.findall(pattern, text)
     return [[float(m) for m in match] for match in matches]
 
-def draw_bounding_boxes(image, boxes):
+'''def draw_bounding_boxes(image, boxes):
     """Zeichnet Bounding Boxes auf das Bild"""
     draw = ImageDraw.Draw(image)
     width, height = image.size
@@ -25,8 +25,29 @@ def draw_bounding_boxes(image, boxes):
             xmax * width,
             ymax * height
         ], outline="red", width=3)
+    return image'''
+
+def draw_bounding_boxes(image, boxes):
+    """Zeichnet Bounding Boxes auf das Bild"""
+    draw = ImageDraw.Draw(image)
+    width, height = image.size
+    
+    for box in boxes:
+        # Sicherstellen, dass alle Werte zwischen 0-1 liegen
+        ymin = max(0.0, min(1.0, box[0]))
+        xmin = max(0.0, min(1.0, box[1]))
+        ymax = max(0.0, min(1.0, box[2]))
+        xmax = max(0.0, min(1.0, box[3]))
+
+        draw.rectangle([
+            xmin * width,
+            ymin * height,
+            xmax * width,
+            ymax * height
+        ], outline="red", width=5)  # Dickere Linien
     return image
 
+    
 # Streamlit UI
 st.title("Bildanalyse mit Gemini")
 col1, col2 = st.columns(2)
@@ -68,6 +89,7 @@ with col1:
                         model="gemini-2.0-flash-exp",
                         contents=[detection_prompt, image_part]
                     )
+                    st.write("Raw API Response:", box_response.text)
 
                     # Verarbeitung
                     boxes = parse_list_boxes(box_response.text)