Update main.py

main.py

@@ -2,6 +2,7 @@ from flask import Flask, request, Response
 import requests
 import logging
 from urllib.parse import urljoin
+from urllib.parse import quote
 from bs4 import BeautifulSoup
 import re
 
@@ -16,7 +17,8 @@ TARGET_BASE_URL = "https://superetka.com"
 
 @app.route('/proxy_image')
 def proxy_image():
-    image_url = request.args.get('url')
+    from urllib.parse import unquote
+    image_url = unquote(request.args.get('url'))
     if not image_url:
         return 'No URL provided', 400
     
@@ -122,7 +124,8 @@ def proxy(path):
                 img_src = img['src']
                 if not img_src.startswith('http'):
                     img_src = urljoin(TARGET_BASE_URL, img_src)
-                proxy_url = f"/proxy_image?url={img_src}"
+                from urllib.parse import quote
+                proxy_url = f"/proxy_image?url={quote(img_src)}"
                 button['onclick'] = f'window.open("{proxy_url}", "_blank")'
                 button['style'] = 'margin: 5px;'
                 button.string = 'Показать изображение'
@@ -146,6 +149,18 @@ def proxy(path):
             if key.lower() not in ('transfer-encoding', 'content-encoding', 'content-length'):
                 response.headers[key] = value
                 
+        # Copy cookies from target response
+        for cookie in resp.cookies:
+            response.set_cookie(
+                key=cookie.name,
+                value=cookie.value,
+                # domain=cookie.domain,
+                path=cookie.path,
+                expires=cookie.expires,
+                secure=cookie.secure,
+                httponly=cookie.httponly
+            )
+        
         return response
         
     except requests.RequestException as e: