Update app.py

app.py

@@ -4,6 +4,8 @@ import requests
 import time
 import asyncio
 from typing import Dict
+import base64
+import hashlib
 
 app = FastAPI()
 
@@ -436,7 +438,7 @@ HTML_CONTENT = """
                 align-items: flex-start;
             }
 
-            .history-item-actions {
+                        .history-item-actions {
                 margin-top: 10px;
                 flex-wrap: wrap;
             }
@@ -445,7 +447,7 @@ HTML_CONTENT = """
                 font-size: 0.9rem;
             }
 
-                        .drop-zone {
+            .drop-zone {
                 padding: 15px;
             }
 
@@ -789,11 +791,28 @@ HTML_CONTENT = """
                     actionsContainer.appendChild(embedBtn);
                 }
 
+                const encryptBtn = document.createElement('button');
+                encryptBtn.textContent = 'Encrypt';
+                encryptBtn.className = 'small-btn';
+                encryptBtn.onclick = () => {
+                    const encryptedUrl = encryptUrl(item.url);
+                    navigator.clipboard.writeText(window.location.origin + '/encrypted/' + encryptedUrl).then(() => {
+                        alert('Encrypted link copied to clipboard!');
+                    });
+                };
+                actionsContainer.appendChild(encryptBtn);
+
                 historyItem.appendChild(actionsContainer);
                 historyList.appendChild(historyItem);
             });
             historyModal.style.display = "block";
         }
+
+        function encryptUrl(url) {
+            const key = 'your-secret-key'; // Replace with a secure key
+            const encodedUrl = btoa(url);
+            return encodedUrl.split('').reverse().join('');
+        }
     </script>
 </body>
 </html>
@@ -888,6 +907,11 @@ async def embed_video(url: str, thumbnail: str):
     '''
     return HTMLResponse(content=html)
 
+@app.get("/encrypted/{encrypted_url}")
+async def handle_encrypted_url(encrypted_url: str, request: Request):
+    decrypted_url = decrypt_url(encrypted_url)
+    return await handle_video_stream(decrypted_url.split('/pbxt/')[1], request)
+
 async def get_cookies() -> Dict[str, str]:
     try:
         response = requests.get('https://replicate.com/levelsio/neon-tokyo', headers={
@@ -939,6 +963,11 @@ async def retry_upload(upload_url: str, file_content: bytes, content_type: str,
             print(f"Error during upload: {e}")
         
         await asyncio.sleep(delay)
-        delay = min(delay * 2, 60)  # Exponential backoff, capped at 60 seconds
+        delay = min(delay * 2, 60)
     
-    return False
+    return False
+
+def decrypt_url(encrypted_url: str) -> str:
+    key = '875843758RR'
+    decoded_url = base64.b64decode(encrypted_url[::-1]).decode('utf-8')
+    return decoded_url