Update main.py

main.py

@@ -239,6 +239,39 @@ def generate_master_blueprint_task(subject, flattened_data, uid, epiphany_id):
         logger.error(f"Master Blueprint Thread Error: {e}")
         return None
 
+#Prepare hi fidelity images to proper scale
+def prepare_vision_bytes(image_bytes, max_size_mb=9.5):
+    """
+    Surgically compresses/resizes image bytes to stay under the 10MB API limit.
+    Target: 9.5MB to allow a safe margin for the rest of the request payload.
+    """
+    size_mb = len(image_bytes) / (1024 * 1024)
+    if size_mb <= max_size_mb:
+        return image_bytes
+
+    logger.info(f"Vision Scaler: Image is {size_mb:.2f}MB. Optimizing for API...")
+    try:
+        img = Image.open(io.BytesIO(image_bytes))
+        
+        # Target resolution: 2048px on the long side is optimal for Gemini detection
+        max_dim = 2048
+        w, h = img.size
+        if max(w, h) > max_dim:
+            scale = max_dim / max(w, h)
+            img = img.resize((int(w * scale), int(h * scale)), Image.Resampling.LANCZOS)
+
+        output = io.BytesIO()
+        # Convert to RGB (removes Alpha channel if present to save space)
+        # Save as optimized JPEG at 85% quality
+        img.convert('RGB').save(output, format="JPEG", quality=85, optimize=True)
+        new_bytes = output.getvalue()
+        
+        new_size_mb = len(new_bytes)/(1024*1024)
+        logger.info(f"Vision Scaler: Optimization complete. New size: {new_size_mb:.2f}MB")
+        return new_bytes
+    except Exception as e:
+        logger.error(f"Vision Scaler Failure: {e}")
+        return image_bytes # Fallback to original and hope for the best
 # -----------------------------------------------------------------------------
 # 4. PRIMARY ENDPOINTS: GENERATE & THEIA SWEEP
 # -----------------------------------------------------------------------------
@@ -377,7 +410,7 @@ def generate_epiphany():
 
 @app.route('/api/epiphany/theia', methods=['POST'])
 def theia_sweep():
-    """Standalone Theia Mode: Bounding Box Annotations with DB Persistence."""
+    """Standalone Theia Mode: Bounding Box Annotations with Vision Scaling."""
     logger.info(">>> THEIA SWEEP INITIATED")
     uid = verify_token(request.headers.get('Authorization'))
     if not uid: return jsonify({'error': 'Unauthorized'}), 401
@@ -397,7 +430,16 @@ def theia_sweep():
     if user_data.get('credits', 0) < 4:
         return jsonify({'error': 'Need 4 Sparks for a Theia Sweep.'}), 402
 
+    # --- IMAGE PRE-PROCESSING ---
+    if 'image' not in request.files:
+        return jsonify({'error': 'image file is required.'}), 400
+        
     image_file = request.files['image']
+    raw_bytes = image_file.read()
+    
+    # Surgical Fix: Stay under 10MB limit
+    image_bytes = prepare_vision_bytes(raw_bytes)
+    
     subject = existing_data.get('subject', 'Complex System')
 
     sweep_prompt = f"""
@@ -411,7 +453,7 @@ def theia_sweep():
     """
 
     try:
-        pil_image = Image.open(io.BytesIO(image_file.read())).convert('RGB')
+        pil_image = Image.open(io.BytesIO(image_bytes)).convert('RGB')
         res = client.models.generate_content(
             model=ATHENA_FLASH,
             contents=[sweep_prompt, pil_image],
@@ -422,8 +464,9 @@ def theia_sweep():
         )
         
         raw_json = res.text.strip()
-        if "```json" in raw_json:
-            raw_json = re.search(r'```json\n(.*?)\n```', raw_json, re.DOTALL).group(1)
+        # Use helper to strip fences if present
+        if "```" in raw_json:
+            raw_json = _strip_json_fences(raw_json)
             
         annotations = json.loads(raw_json)