Update main.py

main.py

@@ -169,60 +169,96 @@ def send_text_request(model_name, prompt, image):
         return None
 
 
-#=============================================================================
+import logging
+
+# Configure logging at the top of your file if not already done
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# =============================================================================
 # OPEN IMAGE PROXY ENDPOINT (NO AUTHENTICATION)
 # =============================================================================
 @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:
-        print(f"URL parsing error in image proxy: {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.
-        print(f"Error in image proxy: {e}")
+        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
 
 # -----------------------------------------------------------------------------