Update main.py

main.py

@@ -176,11 +176,8 @@ def send_text_request(model_name, prompt, image):
 def image_proxy():
     """
     [NO AUTH - OPEN PROXY] Fetches an image from a given URL and returns it.
-    This acts as a proxy to get around client-side CORS issues for PDF generation.
-    The image URL is passed as a query parameter.
-
-    WARNING: This endpoint is an open proxy and is not secure. Anyone can use it
-    to route traffic through your server. Use with caution.
+    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:
@@ -188,29 +185,38 @@ def image_proxy():
 
     try:
         # Use a session object for better performance and connection management.
-        # Set a timeout to prevent requests from hanging indefinitely.
         with requests.Session() as s:
-            response = s.get(image_url, stream=True, timeout=15)
-            # Raise an exception for bad status codes (e.g., 404 Not Found, 500 Server Error).
+            # 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()
 
-            # Get the content type from the original image response to pass it along to the client.
+            # Get the content type from the original image response to pass it along.
             content_type = response.headers.get('Content-Type', 'application/octet-stream')
 
-            # Stream the response back to the client. This is memory-efficient as it doesn't
-            # load the entire image into memory at once.
+            # Stream the response back to the client.
             return Response(response.iter_content(chunk_size=1024), content_type=content_type)
 
-    except requests.exceptions.Timeout:
-        return jsonify({'error': 'The request to the external image URL timed out.'}), 504 # 504 Gateway Timeout
     except requests.exceptions.RequestException as e:
-        # This catches other network-related errors (e.g., DNS failure, connection error).
-        print(f"Error fetching image from URL '{image_url}': {e}")
-        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 # 502 Bad Gateway
+        # 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:
-        # Catch any other unexpected errors.
-        print(f"An unexpected error occurred while proxying the image '{image_url}': {e}")
-        return jsonify({'error': 'An internal server error occurred while trying to fetch the image.'}), 500
+        # 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
 
 # -----------------------------------------------------------------------------
 # 3. AUTHENTICATION & USER MANAGEMENT