Update main.py

main.py

@@ -174,49 +174,30 @@ def send_text_request(model_name, prompt, image):
 # =============================================================================
 @app.route('/api/image-proxy', methods=['GET'])
 def image_proxy():
-    """
-    [NO AUTH - OPEN PROXY] Fetches an image from a given URL and returns it.
-    This version includes better error handling and a workaround for potential
-    SSL verification issues in sandboxed server environments.
-    """
     image_url = request.args.get('url')
     if not image_url:
-        return jsonify({'error': 'The "url" query parameter is required.'}), 400
+        return jsonify({'error': 'URL parameter is missing.'}), 400
 
     try:
-        # Use a session object for better performance and connection management.
-        with requests.Session() as s:
-            # The key change is adding `verify=False`. This is often necessary
-            # in environments like Hugging Face Spaces or behind corporate proxies
-            # that can interfere with SSL certificate verification.
-            response = s.get(image_url, stream=True, timeout=15, verify=False)
-            
-            # Raise an exception for bad status codes (e.g., 404 Not Found).
-            response.raise_for_status()
+        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]
+        
+        blob = bucket.blob(blob_path)
 
-            # Get the content type from the original image response to pass it along.
-            content_type = response.headers.get('Content-Type', 'application/octet-stream')
+        if not blob.exists():
+            return jsonify({'error': 'Image not found.'}), 404
 
-            # Stream the response back to the client.
-            return Response(response.iter_content(chunk_size=1024), content_type=content_type)
+        image_bytes = blob.download_as_bytes()
+        content_type = blob.content_type or 'application/octet-stream'
+        
+        return Response(image_bytes, content_type=content_type)
 
-    except requests.exceptions.RequestException as e:
-        # This will catch network-related errors.
-        print("="*20, "REQUESTS ERROR", "="*20)
-        print(f"Failed to fetch image from URL: {image_url}")
-        # The following line prints the full error, which is crucial for debugging.
-        print(traceback.format_exc())
-        print("="*50)
-        return jsonify({'error': f'Failed to fetch the image from the provided URL. The server may be down or the URL may be incorrect.'}), 502
-    
     except Exception as e:
-        # This will catch any other unexpected errors in the function.
-        print("="*20, "UNEXPECTED PROXY ERROR", "="*20)
-        print(f"An unexpected error occurred while proxying the image: {image_url}")
-        # Print the full traceback to the logs for detailed debugging.
-        print(traceback.format_exc())
-        print("="*60)
-        return jsonify({'error': 'An internal server error occurred while trying to fetch the image.'}), 50
+        # This will catch parsing errors or other unexpected issues.
+        print(f"Error in image proxy: {e}")
+        return jsonify({'error': 'Internal server error processing the image request.'}), 500
 
 # -----------------------------------------------------------------------------
 # 3. AUTHENTICATION & USER MANAGEMENT