Update app.py

app.py

@@ -4,6 +4,9 @@ import requests
 import time
 import asyncio
 from typing import Dict
+import base64
+import hashlib
+from cryptography.fernet import Fernet
 
 app = FastAPI()
 
@@ -425,7 +428,7 @@ HTML_CONTENT = """
                 font-size: 0.7rem;
             }
 
-            .modal-content, .history-modal-content {
+                        .modal-content, .history-modal-content {
                 width: 95%;
                 margin: 5% auto;
                 padding: 15px;
@@ -445,7 +448,7 @@ HTML_CONTENT = """
                 font-size: 0.9rem;
             }
 
-                        .drop-zone {
+            .drop-zone {
                 padding: 15px;
             }
 
@@ -720,6 +723,17 @@ HTML_CONTENT = """
                 buttonsContainer.appendChild(embedBtn);
             }
 
+            const encryptBtn = document.createElement('button');
+            encryptBtn.textContent = 'Encrypt Link';
+            encryptBtn.className = 'small-btn encrypt-btn';
+            encryptBtn.onclick = () => {
+                const encryptedUrl = encryptUrl(url);
+                navigator.clipboard.writeText(encryptedUrl).then(() => {
+                    alert('Encrypted link copied to clipboard!');
+                });
+            };
+            buttonsContainer.appendChild(encryptBtn);
+
             linkContainer.appendChild(buttonsContainer);
 
             resultContainer.appendChild(linkContainer);
@@ -789,12 +803,36 @@ 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(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 = generateEncryptionKey();
+            const encryptedData = CryptoJS.AES.encrypt(url, key).toString();
+            const encryptedUrl = `${window.location.origin}/decrypt?data=${encodeURIComponent(encryptedData)}&key=${encodeURIComponent(key)}`;
+            return encryptedUrl;
+        }
+
+        function generateEncryptionKey() {
+            const browserInfo = navigator.userAgent + navigator.language + screen.width + screen.height + new Date().getTimezoneOffset();
+            return CryptoJS.SHA256(browserInfo).toString();
+        }
     </script>
+    <script src="https://cdnjs.cloudflare.com/ajax/libs/crypto-js/4.1.1/crypto-js.min.js"></script>
 </body>
 </html>
 """
@@ -888,6 +926,14 @@ async def embed_video(url: str, thumbnail: str):
     '''
     return HTMLResponse(content=html)
 
+@app.get("/decrypt")
+async def decrypt_url(data: str, key: str):
+    try:
+        decrypted_url = Fernet(key.encode()).decrypt(data.encode()).decode()
+        return RedirectResponse(url=decrypted_url)
+    except:
+        return HTMLResponse(content="Invalid or expired link", status_code=400)
+
 async def get_cookies() -> Dict[str, str]:
     try:
         response = requests.get('https://replicate.com/levelsio/neon-tokyo', headers={
@@ -939,6 +985,10 @@ 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
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)