Update main.py

main.py

@@ -21,9 +21,25 @@ def proxy_image():
         return 'No URL provided', 400
     
     try:
+        # Get headers from the incoming request
+        headers = {key: value for key, value in request.headers if key.lower() != 'host'}
+        headers['Referer'] = TARGET_BASE_URL
+        
         # Forward the request to get the image
-        resp = requests.get(image_url, timeout=30)
-        return Response(resp.content, mimetype=resp.headers['Content-Type'])
+        resp = requests.get(
+            image_url,
+            headers=headers,
+            cookies=request.cookies,
+            timeout=30,
+            allow_redirects=True
+        )
+        
+        # Check response status
+        if resp.status_code != 200:
+            logger.error(f"Error response from target: {resp.status_code}")
+            return f"Error: {resp.status_code}", resp.status_code
+            
+        return Response(resp.content, mimetype=resp.headers.get('Content-Type', 'image/jpeg'))
     except Exception as e:
         logger.error(f"Error proxying image: {str(e)}")
         return str(e), 500