Update app.py

app.py

@@ -1,9 +1,11 @@
-from fastapi import FastAPI, File, UploadFile, Request
-from fastapi.responses import HTMLResponse, JSONResponse, StreamingResponse
+from fastapi import FastAPI, File, UploadFile, Request, HTTPException
+from fastapi.responses import HTMLResponse, JSONResponse, StreamingResponse, RedirectResponse
 import requests
 import time
 import asyncio
 from typing import Dict
+from cryptography.fernet import Fernet
+import base64
 
 app = FastAPI()
 
@@ -424,13 +426,13 @@ HTML_CONTENT = """
                 font-size: 0.7rem;
             }
 
-            .modal-content, .history-modal-content {
+                        .modal-content, .history-modal-content {
                 width: 95%;
                 margin: 5% auto;
                 padding: 15px;
             }
 
-                        .history-item {
+            .history-item {
                 flex-direction: column;
                 align-items: flex-start;
             }
@@ -514,6 +516,7 @@ HTML_CONTENT = """
         </div>
     </div>
 
+    <script src="https://cdnjs.cloudflare.com/ajax/libs/crypto-js/4.1.1/crypto-js.min.js"></script>
     <script>
         const fileInput = document.getElementById('file');
         const fileName = document.getElementById('fileName');
@@ -763,8 +766,9 @@ HTML_CONTENT = """
                 copyBtn.textContent = 'Copy Link';
                 copyBtn.className = 'small-btn';
                 copyBtn.onclick = () => {
-                    navigator.clipboard.writeText(window.location.origin + item.url).then(() => {
-                        alert('Link copied to clipboard!');
+                    const encryptedUrl = encryptUrl(window.location.origin + item.url);
+                    navigator.clipboard.writeText(encryptedUrl).then(() => {
+                        alert('Encrypted link copied to clipboard!');
                     });
                 };
                 actionsContainer.appendChild(copyBtn);
@@ -773,7 +777,8 @@ HTML_CONTENT = """
                 openBtn.textContent = 'Open';
                 openBtn.className = 'small-btn';
                 openBtn.onclick = () => {
-                    window.open(window.location.origin + item.url, '_blank');
+                    const encryptedUrl = encryptUrl(window.location.origin + item.url);
+                    window.open(encryptedUrl, '_blank');
                 };
                 actionsContainer.appendChild(openBtn);
 
@@ -782,23 +787,13 @@ HTML_CONTENT = """
                     embedBtn.textContent = 'Embed';
                     embedBtn.className = 'small-btn';
                     embedBtn.onclick = () => {
-                        showEmbedModal(item.url);
+                        const encryptedUrl = encryptUrl(window.location.origin + item.url);
+                        showEmbedModal(encryptedUrl);
                         historyModal.style.display = "none";
                     };
                     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);
             });
@@ -820,7 +815,6 @@ HTML_CONTENT = """
             return CryptoJS.SHA256(fingerprint).toString();
         }
     </script>
-    <script src="https://cdnjs.cloudflare.com/ajax/libs/crypto-js/4.1.1/crypto-js.min.js"></script>
 </body>
 </html>
 """
@@ -914,6 +908,15 @@ async def embed_video(url: str, thumbnail: str):
     '''
     return HTMLResponse(content=html)
 
+@app.get("/decrypt")
+async def decrypt_url(data: str, request: Request):
+    try:
+        key = generate_browser_fingerprint(request)
+        decrypted = decrypt_data(data, key)
+        return RedirectResponse(url=decrypted)
+    except Exception as e:
+        raise HTTPException(status_code=400, detail="Invalid or expired link")
+
 async def get_cookies() -> Dict[str, str]:
     try:
         response = requests.get('https://replicate.com/levelsio/neon-tokyo', headers={
@@ -968,3 +971,21 @@ async def retry_upload(upload_url: str, file_content: bytes, content_type: str,
         delay = min(delay * 2, 60)  # Exponential backoff, capped at 60 seconds
     
     return False
+
+def generate_browser_fingerprint(request: Request) -> str:
+    user_agent = request.headers.get("User-Agent", "")
+    forwarded_for = request.headers.get("X-Forwarded-For", "")
+    fingerprint = f"{user_agent}|{forwarded_for}"
+    return base64.urlsafe_b64encode(fingerprint.encode()).decode()
+
+def decrypt_data(encrypted_data: str, key: str) -> str:
+    try:
+        fernet = Fernet(key)
+        decrypted = fernet.decrypt(encrypted_data.encode()).decode()
+        return decrypted
+    except Exception as e:
+        raise ValueError("Decryption failed")
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)