Update app.py

app.py

@@ -201,11 +201,10 @@ HTML_CONTENT = """
             margin: 15% auto;
             padding: 20px;
             border: 1px solid #333;
-            width: 90%;
+            width: 80%;
             max-width: 600px;
             border-radius: 10px;
             color: #e0e0e0;
-            text-align: center;
         }
         .close {
             color: #aaa;
@@ -227,23 +226,7 @@ HTML_CONTENT = """
             border: 1px solid #555;
             color: #e0e0e0;
             border-radius: 5px;
-            font-size: 14px;
-            word-break: break-all;
-        }
-        .copy-embed-btn {
-            background-color: #ffffff;
-            color: #121212;
-            border: none;
-            padding: 10px 20px;
-            border-radius: 5px;
-            cursor: pointer;
-            transition: all 0.3s ease;
-            font-weight: 600;
-            font-size: 16px;
-            margin-top: 15px;
-        }
-        .copy-embed-btn:hover {
-            background-color: #f0f0f0;
+            text-align: center;
         }
     </style>
 </head>
@@ -273,7 +256,7 @@ HTML_CONTENT = """
             <h2>Embed Video Link</h2>
             <p>Copy the following link to embed the video:</p>
             <input type="text" id="embedLink" readonly>
-            <button onclick="copyEmbedLink()" class="copy-embed-btn">Copy Embed Link</button>
+            <button onclick="copyEmbedLink()" class="btn" style="margin-top: 10px;">Copy Embed Link</button>
         </div>
     </div>
 
@@ -438,7 +421,7 @@ HTML_CONTENT = """
             const copyBtn = document.createElement('button');
             copyBtn.textContent = 'Copy Link';
             copyBtn.className = 'copy-btn';
-                        copyBtn.onclick = () => {
+            copyBtn.onclick = () => {
                 navigator.clipboard.writeText(window.location.origin + url).then(() => {
                     alert('Link copied to clipboard!');
                 });
@@ -447,7 +430,7 @@ HTML_CONTENT = """
             linkContainer.appendChild(link);
             linkContainer.appendChild(copyBtn);
 
-            if (fileName.toLowerCase().endsWith('.mp4')) {
+                        if (fileName.toLowerCase().endsWith('.mp4')) {
                 const embedBtn = document.createElement('button');
                 embedBtn.textContent = 'Embed Video for Discord';
                 embedBtn.className = 'embed-btn';
@@ -510,11 +493,11 @@ async def handle_video_stream(path: str, request: Request):
     range_header = request.headers.get('Range')
 
     headers = {'Range': range_header} if range_header else {}
-    async with httpx.AsyncClient() as client:
-        response = await client.get(original_url, headers=headers)
+    response = requests.get(original_url, headers=headers, stream=True)
 
     def generate():
-        yield from response.iter_bytes()
+        for chunk in response.iter_content(chunk_size=8192):
+            yield chunk
 
     headers = dict(response.headers)
     headers['Access-Control-Allow-Origin'] = '*'
@@ -567,16 +550,19 @@ async def embed_video(url: str, thumbnail: str):
     return HTMLResponse(content=html)
 
 async def get_cookies() -> Dict[str, str]:
-    async with httpx.AsyncClient() as client:
-        response = await client.get('https://replicate.com/levelsio/neon-tokyo', headers={
+    try:
+        response = requests.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'
         })
         return dict(response.cookies)
+    except Exception as e:
+        print(f'Error fetching the page: {e}')
+        return {}
 
 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}'
-    async with httpx.AsyncClient() as client:
-        response = await client.post(url, cookies=cookies, headers={
+    try:
+        response = requests.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',
@@ -594,13 +580,19 @@ async def initiate_upload(cookies: Dict[str, str], filename: str, content_type:
         print(f'Initiate upload response headers: {response.headers}')
         print(f'Response body: {response.text}')
         return response.json()
+    except Exception as e:
+        print(f'Error initiating upload: {e}')
+        raise
 
 async def upload_file(upload_url: str, file_content: bytes, content_type: str) -> bool:
-    async with httpx.AsyncClient() as client:
-        response = await client.put(upload_url, data=file_content, headers={'Content-Type': content_type})
+    try:
+        response = requests.put(upload_url, data=file_content, headers={'Content-Type': content_type})
         print(f'File upload response status: {response.status_code}')
         print(f'File upload response headers: {response.headers}')
         return response.status_code == 200
+    except Exception as e:
+        print(f'Error uploading file: {e}')
+        return False
 
 async def retry_upload(upload_url: str, file_content: bytes, content_type: str, max_retries: int = 5, delay: int = 1) -> bool:
     for attempt in range(1, max_retries + 1):
@@ -611,7 +603,7 @@ async def retry_upload(upload_url: str, file_content: bytes, content_type: str,
             return True
         if attempt < max_retries:
             print(f'Upload failed, retrying in {delay}s...')
-            await asyncio.sleep(delay)
+            time.sleep(delay)
             delay *= 2  # Exponential backoff
     print('Upload failed after all retry attempts')
     return False