Update app.py

app.py

@@ -201,10 +201,11 @@ HTML_CONTENT = """
             margin: 15% auto;
             padding: 20px;
             border: 1px solid #333;
-            width: 80%;
+            width: 90%;
             max-width: 600px;
             border-radius: 10px;
             color: #e0e0e0;
+            text-align: center;
         }
         .close {
             color: #aaa;
@@ -226,7 +227,23 @@ HTML_CONTENT = """
             border: 1px solid #555;
             color: #e0e0e0;
             border-radius: 5px;
-            text-align: center;
+            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;
         }
     </style>
 </head>
@@ -254,9 +271,9 @@ HTML_CONTENT = """
         <div class="modal-content">
             <span class="close">&times;</span>
             <h2>Embed Video Link</h2>
-            <p>Copy the link to embed the video for discord:</p>
+            <p>Copy the following link to embed the video:</p>
             <input type="text" id="embedLink" readonly>
-            <button onclick="copyEmbedLink()" class="btn" style="margin-top: 10px;">Copy Embed Link</button>
+            <button onclick="copyEmbedLink()" class="copy-embed-btn">Copy Embed Link</button>
         </div>
     </div>
 
@@ -421,7 +438,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!');
                 });
@@ -430,7 +447,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';
@@ -493,11 +510,11 @@ async def handle_video_stream(path: str, request: Request):
     range_header = request.headers.get('Range')
 
     headers = {'Range': range_header} if range_header else {}
-    response = requests.get(original_url, headers=headers, stream=True)
+    async with httpx.AsyncClient() as client:
+        response = await client.get(original_url, headers=headers)
 
     def generate():
-        for chunk in response.iter_content(chunk_size=8192):
-            yield chunk
+        yield from response.iter_bytes()
 
     headers = dict(response.headers)
     headers['Access-Control-Allow-Origin'] = '*'
@@ -550,19 +567,16 @@ async def embed_video(url: str, thumbnail: str):
     return HTMLResponse(content=html)
 
 async def get_cookies() -> Dict[str, str]:
-    try:
-        response = requests.get('https://replicate.com/levelsio/neon-tokyo', headers={
+    async with httpx.AsyncClient() as client:
+        response = await client.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}'
-    try:
-        response = requests.post(url, cookies=cookies, headers={
+    async with httpx.AsyncClient() as client:
+        response = await client.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',
@@ -580,19 +594,13 @@ 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:
-    try:
-        response = requests.put(upload_url, data=file_content, headers={'Content-Type': content_type})
+    async with httpx.AsyncClient() as client:
+        response = await client.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):
@@ -603,7 +611,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...')
-            time.sleep(delay)
+            await asyncio.sleep(delay)
             delay *= 2  # Exponential backoff
     print('Upload failed after all retry attempts')
     return False