Update main.py

main.py

@@ -169,7 +169,7 @@ def send_text_request(model_name, prompt, image):
         return None
 
 
-# =============================================================================
+#=============================================================================
 # OPEN IMAGE PROXY ENDPOINT (NO AUTHENTICATION)
 # =============================================================================
 @app.route('/api/image-proxy', methods=['GET'])
@@ -179,21 +179,47 @@ def image_proxy():
         return jsonify({'error': 'URL parameter is missing.'}), 400
 
     try:
-        bucket_name = bucket.name
-        # Extract the file path from the full URL
-        # Example: https://.../bucket-name/users/file.png -> users/file.png
-        blob_path = image_url.split(f"{bucket_name}/")[1].split("?")[0]
+        # Parse Firebase Storage URL
+        # Expected format: https://storage.googleapis.com/bucket-name/path/to/file.ext
+        if 'storage.googleapis.com' not in image_url:
+            return jsonify({'error': 'Invalid Firebase Storage URL.'}), 400
+        
+        # Extract bucket name and blob path from the URL
+        url_parts = image_url.split('storage.googleapis.com/')[1]
+        # Remove query parameters if present
+        url_parts = url_parts.split('?')[0]
+        
+        # Split into bucket name and blob path
+        path_components = url_parts.split('/', 1)
+        if len(path_components) < 2:
+            return jsonify({'error': 'Invalid URL format.'}), 400
+            
+        url_bucket_name = path_components[0]
+        blob_path = path_components[1]
+        
+        # Verify bucket name matches (optional security check)
+        expected_bucket_name = bucket.name
+        if url_bucket_name != expected_bucket_name:
+            return jsonify({'error': 'Bucket name mismatch.'}), 403
         
+        # Get the blob
         blob = bucket.blob(blob_path)
 
         if not blob.exists():
             return jsonify({'error': 'Image not found.'}), 404
 
+        # Download and return the image
         image_bytes = blob.download_as_bytes()
         content_type = blob.content_type or 'application/octet-stream'
         
-        return Response(image_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}")
+        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}")