Update app.py

app.py

@@ -4,14 +4,14 @@ import json
 from PIL import Image, ImageDraw, ImageFont, ImageColor
 import google.generativeai as genai
 
-# --- 1. SETUP API KEY (HARDCODED) ---
-# ⚠️ WARNING: Never share this file publicly with the key inside!
+# --- 1. SETUP API KEY ---
+# We use the key you provided. 
+# (Note: For long-term safety, use Settings > Secrets, but this works for now!)
 api_key = "AIzaSyABaidsygD73gbaSlMHlkrhhiDT8NDzrjE"
 
 genai.configure(api_key=api_key)
 
 # --- 2. CONFIGURATION ---
-# We use Gemini 1.5 Flash for fast object detection
 generation_config = {
     "temperature": 0.5,
     "response_mime_type": "application/json"
@@ -33,7 +33,7 @@ model = genai.GenerativeModel(
 def parse_json(json_output):
     """Safely parse JSON even if the model adds text."""
     try:
-        # Remove code blocks if present
+        # Clean potential markdown
         json_output = json_output.replace("```json", "").replace("```", "")
         return json.loads(json_output)
     except Exception as e:
@@ -57,22 +57,4 @@ def plot_bounding_boxes(im, boxes):
             font = ImageFont.load_default()
 
         for i, box in enumerate(boxes):
-            color = colors[i % len(colors)]
-            
-            # Get coords
-            ymin, xmin, ymax, xmax = box.get("box_2d", [0,0,0,0])
-            label = box.get("label", "Object")
-
-            # Convert 0-1000 scale to pixels
-            left = int((xmin / 1000) * width)
-            top = int((ymin / 1000) * height)
-            right = int((xmax / 1000) * width)
-            bottom = int((ymax / 1000) * height)
-
-            # Draw Box
-            draw.rectangle([left, top, right, bottom], outline=color, width=4)
-            
-            # Draw Label with background
-            text_bbox = draw.textbbox((left, top), label, font=font)
-            draw.rectangle(text_bbox, fill=color)
-            draw.text((left, top), label, fill="white", font=font)
+            color = colors[i % len(colors)]