Update main.py

main.py

@@ -765,6 +765,89 @@ def generate_sozo_slides(project_id):
         db.reference(f'sozo_projects/{project_id}').update({'status': 'failed_slides', 'error': str(e)})
         return jsonify({'error': str(e)}), 500
 
+@app.route('/api/image-proxy', methods=['GET'])
+def image_proxy():
+    image_url = request.args.get('url')
+    logger.info(f"[IMAGE PROXY] Received URL: {image_url}")
+    
+    if not image_url:
+        logger.error("[IMAGE PROXY] ERROR: URL parameter is missing")
+        return jsonify({'error': 'URL parameter is missing.'}), 400
+
+    try:
+        # Parse Firebase Storage URL
+        # Expected format: https://storage.googleapis.com/bucket-name/path/to/file.ext
+        if 'storage.googleapis.com' not in image_url:
+            logger.error(f"[IMAGE PROXY] ERROR: Invalid Firebase Storage URL: {image_url}")
+            return jsonify({'error': 'Invalid Firebase Storage URL.'}), 400
+        
+        logger.info(f"[IMAGE PROXY] Parsing URL: {image_url}")
+        
+        # Extract bucket name and blob path from the URL
+        url_parts = image_url.split('storage.googleapis.com/')[1]
+        logger.info(f"[IMAGE PROXY] URL parts after split: {url_parts}")
+        
+        # Remove query parameters if present
+        url_parts = url_parts.split('?')[0]
+        logger.info(f"[IMAGE PROXY] URL parts after removing query params: {url_parts}")
+        
+        # Split into bucket name and blob path
+        path_components = url_parts.split('/', 1)
+        logger.info(f"[IMAGE PROXY] Path components: {path_components}")
+        
+        if len(path_components) < 2:
+            logger.error(f"[IMAGE PROXY] ERROR: Invalid URL format - path_components: {path_components}")
+            return jsonify({'error': 'Invalid URL format.'}), 400
+            
+        url_bucket_name = path_components[0]
+        blob_path = path_components[1]
+        
+        logger.info(f"[IMAGE PROXY] Extracted bucket name: {url_bucket_name}")
+        logger.info(f"[IMAGE PROXY] Extracted blob path: {blob_path}")
+        
+        # Verify bucket name matches (optional security check)
+        expected_bucket_name = bucket.name
+        logger.info(f"[IMAGE PROXY] Expected bucket name: {expected_bucket_name}")
+        
+        if url_bucket_name != expected_bucket_name:
+            logger.error(f"[IMAGE PROXY] ERROR: Bucket name mismatch - URL: {url_bucket_name}, Expected: {expected_bucket_name}")
+            return jsonify({'error': 'Bucket name mismatch.'}), 403
+        
+        logger.info(f"[IMAGE PROXY] Creating blob object for path: {blob_path}")
+        
+        # Get the blob
+        blob = bucket.blob(blob_path)
+
+        logger.info(f"[IMAGE PROXY] Checking if blob exists...")
+        if not blob.exists():
+            logger.error(f"[IMAGE PROXY] ERROR: Image not found at path: {blob_path}")
+            return jsonify({'error': 'Image not found.'}), 404
+
+        logger.info(f"[IMAGE PROXY] Downloading blob...")
+        # Download and return the image
+        image_bytes = blob.download_as_bytes()
+        content_type = blob.content_type or 'application/octet-stream'
+        
+        logger.info(f"[IMAGE PROXY] Successfully downloaded {len(image_bytes)} bytes, content-type: {content_type}")
+        
+        # Add cache headers for better performance
+        response = Response(image_bytes, content_type=content_type)
+        response.headers['Cache-Control'] = 'public, max-age=3600'  # Cache for 1 hour
+        return response
+
+    except IndexError as e:
+        logger.error(f"[IMAGE PROXY] URL parsing IndexError: {e}")
+        logger.error(f"[IMAGE PROXY] URL was: {image_url}")
+        return jsonify({'error': 'Invalid URL format.'}), 400
+    except Exception as e:
+        # This will catch parsing errors or other unexpected issues.
+        logger.error(f"[IMAGE PROXY] Unexpected error: {e}")
+        logger.error(f"[IMAGE PROXY] Error type: {type(e).__name__}")
+        logger.error(f"[IMAGE PROXY] URL was: {image_url}")
+        import traceback
+        logger.error(f"[IMAGE PROXY] Full traceback: {traceback.format_exc()}")
+        return jsonify({'error': 'Internal server error processing the image request.'}), 500
+
 @app.route('/api/sozo/projects/<string:project_id>/charts', methods=['POST'])
 def regenerate_sozo_chart(project_id):
     logger.info(f"Endpoint /charts POST for project {project_id}")