Update app.py

app.py

@@ -881,6 +881,22 @@ async def handle_video_stream(path: str, request: Request):
 
     return StreamingResponse(generate(), status_code=response.status_code, headers=headers)
 
+@app.get("/encrypted/{encrypted_data}")
+async def handle_encrypted_stream(encrypted_data: str, request: Request):
+    try:
+        browser_info = request.headers.get('User-Agent', '') + request.headers.get('Accept-Language', '') + str(request.client.host)
+        key = hashlib.sha256(browser_info.encode()).hexdigest()[:32].encode()
+        fernet = Fernet(base64.urlsafe_b64encode(key))
+        decrypted_url = fernet.decrypt(base64.urlsafe_b64decode(encrypted_data)).decode()
+        
+        if not decrypted_url.startswith('/rbxg/'):
+            raise ValueError("Invalid decrypted URL")
+        
+        path = decrypted_url.split('/rbxg/')[1]
+        return await handle_video_stream(path, request)
+    except Exception as e:
+        raise HTTPException(status_code=400, detail="Invalid or expired link")
+
 @app.get("/embed")
 async def embed_video(url: str, thumbnail: str):
     html = f'''
@@ -922,17 +938,6 @@ async def embed_video(url: str, thumbnail: str):
     '''
     return HTMLResponse(content=html)
 
-@app.get("/decrypt")
-async def decrypt_url(data: str, request: Request):
-    try:
-        browser_info = request.headers.get('User-Agent', '') + request.headers.get('Accept-Language', '') + str(request.client.host)
-        key = hashlib.sha256(browser_info.encode()).hexdigest()
-        fernet = Fernet(base64.urlsafe_b64encode(key[:32].encode()))
-        decrypted_url = fernet.decrypt(data.encode()).decode()
-        return RedirectResponse(url=decrypted_url)
-    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={
@@ -984,7 +989,7 @@ 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) 
+        delay = min(delay * 2, 60)
     
     return False