Update app.py

app.py

@@ -1,10 +1,12 @@
 from fastapi import FastAPI, File, UploadFile, Request
 from fastapi.responses import HTMLResponse, JSONResponse, StreamingResponse
-import requests
-import time
+import aiohttp
 import asyncio
 from typing import Dict
-import aiohttp
+import base64
+import hmac
+import hashlib
+import time
 
 app = FastAPI()
 
@@ -1001,25 +1003,26 @@ async def get_cookies() -> Dict[str, str]:
         async with session.get('https://replicate.com/levelsio/neon-tokyo', headers={
             'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36'
         }) as response:
-            return dict(response.cookies)
+            return {k: v.value for k, v in response.cookies.items()}
 
 async def initiate_upload(cookies: Dict[str, str], filename: str, content_type: str) -> Dict:
     url = f'https://replicate.com/api/upload/{filename}?content_type={content_type}'
+    headers = {
+        'X-CSRFToken': cookies.get('csrftoken', ''),
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36',
+        'Referer': 'https://replicate.com/levelsio/neon-tokyo',
+        'Origin': 'https://replicate.com',
+        'Accept': '*/*',
+        'Accept-Language': 'en-US,en;q=0.5',
+        'Accept-Encoding': 'identity',
+        'Sec-Fetch-Dest': 'empty',
+        'Sec-Fetch-Mode': 'cors',
+        'Sec-Fetch-Site': 'same-origin',
+        'Sec-GPC': '1',
+        'Priority': 'u=1, i'
+    }
     async with aiohttp.ClientSession() as session:
-        async with session.post(url, cookies=cookies, headers={
-            'X-CSRFToken': cookies.get('csrftoken'),
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36',
-            'Referer': 'https://replicate.com/levelsio/neon-tokyo',
-            'Origin': 'https://replicate.com',
-            'Accept': '*/*',
-            'Accept-Language': 'en-US,en;q=0.5',
-            'Accept-Encoding': 'identity',
-            'Sec-Fetch-Dest': 'empty',
-            'Sec-Fetch-Mode': 'cors',
-            'Sec-Fetch-Site': 'same-origin',
-            'Sec-GPC': '1',
-            'Priority': 'u=1, i'
-        }) as response:
+        async with session.post(url, cookies=cookies, headers=headers) as response:
             return await response.json()
 
 async def chunked_upload(upload_url: str, file: UploadFile, content_type: str, chunk_size: int = 1024 * 1024) -> bool: