Update main.py

main.py

@@ -339,20 +339,19 @@ def generate_epiphany():
 
 @app.route('/api/epiphany/theia', methods=['POST'])
 def theia_sweep():
-    """Standalone Theia Mode: Bounding Box Annotations via Code Execution."""
+    """Standalone Theia Mode: Bounding Box Annotations with DB Persistence."""
     logger.info(">>> THEIA SWEEP INITIATED")
     uid = verify_token(request.headers.get('Authorization'))
     if not uid: return jsonify({'error': 'Unauthorized'}), 401
     
-    # 1. Defend against missing parameters (Fixed 400 logic)
     epiphany_id = request.form.get('epiphanyId')
-    if not epiphany_id:
-        logger.error("Theia Sweep failed: epiphanyId missing from form.")
-        return jsonify({'error': 'epiphanyId is required.'}), 400
+    if not epiphany_id: return jsonify({'error': 'epiphanyId is required.'}), 400
 
-    if 'image' not in request.files:
-        logger.error("Theia Sweep failed: Image file missing.")
-        return jsonify({'error': 'image file is required.'}), 400
+    # 1. Check if annotations already exist to prevent double-charging
+    epiphany_ref = db_ref.child(f'epiphanies/{epiphany_id}')
+    existing_data = epiphany_ref.get() or {}
+    if 'annotations' in existing_data:
+        return jsonify({"annotations": existing_data['annotations'], "status": "already_stored"}), 200
 
     # 2. Check Sparks
     user_ref = db_ref.child(f'users/{uid}')
@@ -360,27 +359,21 @@ def theia_sweep():
     if user_data.get('credits', 0) < 4:
         return jsonify({'error': 'Need 4 Sparks for a Theia Sweep.'}), 402
 
-    # 3. Fetch Context from the Epiphany
-    epiphany_context = db_ref.child(f'epiphanies/{epiphany_id}').get() or {}
-    subject = epiphany_context.get('subject', 'Complex System')
-
     image_file = request.files['image']
-    image_bytes = image_file.read()
-    
+    subject = existing_data.get('subject', 'Complex System')
+
     sweep_prompt = f"""
-    Activate Theia Mode for: {subject}.
-    Using Python Code Execution, perform a spatial deconstruction of this image.
-    Identify every functional, biological, or structural component.
-    For each component detected, return a JSON object:
-    1. label: The name of the part.
+    Theia Mode Activation: {subject}. 
+    Use Python Code Execution for spatial deconstruction.
+    Identify every functional component. For each return:
+    1. label: name of the part.
     2. coordinates: [ymin, xmin, ymax, xmax] (normalized 0-1000).
-    3. micro_epiphany: A 20-word Feynman-style secret about this part's role.
-    
-    Return a JSON LIST ONLY.
+    3. micro_epiphany: 20-word Feynman secret.
+    Return JSON list ONLY.
     """
 
     try:
-        pil_image = Image.open(io.BytesIO(image_bytes)).convert('RGB')
+        pil_image = Image.open(io.BytesIO(image_file.read())).convert('RGB')
         res = client.models.generate_content(
             model=ATHENA_FLASH,
             contents=[sweep_prompt, pil_image],
@@ -395,13 +388,14 @@ def theia_sweep():
             raw_json = re.search(r'```json\n(.*?)\n```', raw_json, re.DOTALL).group(1)
             
         annotations = json.loads(raw_json)
-        # Charge 4 sparks for Code Execution
+        
+        # 3. STORE IN DB & CHARGE
+        epiphany_ref.update({"annotations": annotations})
         user_ref.update({'credits': user_data.get('credits', 0) - 4})
         
-        logger.info(f"THEIA SUCCESS: {len(annotations)} components annotated.")
         return jsonify({"annotations": annotations}), 200
     except Exception as e:
-        logger.error(f"Theia Sweep Engine Error: {e}")
+        logger.error(f"Theia Sweep Error: {e}")
         return jsonify({'error': str(e)}), 500
 
 @app.route('/api/epiphany/deep-dive', methods=['POST'])