Spaces:

coollsd
/

fileuploader

Running

App Files Files Community

coollsd commited on Sep 20, 2024

Commit

c0e61b6

verified ·

1 Parent(s): 885522c

Update app.py

Browse files

Files changed (1) hide show

app.py +33 -62

app.py CHANGED Viewed

@@ -1,14 +1,23 @@
-from fastapi import FastAPI, File, UploadFile, Request
-from fastapi.responses import HTMLResponse, JSONResponse, StreamingResponse
 import requests
 import time
 import asyncio
 from typing import Dict
 import base64
 import secrets
 app = FastAPI()
 HTML_CONTENT = """
 <!DOCTYPE html>
 <html lang="en">
@@ -866,54 +875,30 @@ async def handle_video_stream(path: str, request: Request):
     return StreamingResponse(generate(), status_code=response.status_code, headers=headers)
-@app.get("/embed")
-async def embed_video(url: str, thumbnail: str):
-    html = f'''
-    <html>
-    <head>
-        <meta property="og:type" content="video.other">
-        <meta property="og:video" content="{url}">
-        <meta property="og:video:url" content="{url}">
-        <meta property="og:video:secure_url" content="{url}">
-        <meta property="og:video:type" content="video/mp4">
-        <meta property="og:video:width" content="1280">
-        <meta property="og:video:height" content="720">
-        <meta property="og:image" content="{thumbnail}">
-        <meta property="og:image:secure_url" content="{thumbnail}">
-        <meta property="og:image:width" content="1280">
-        <meta property="og:image:height" content="720">
-        <meta property="og:image:type" content="image/png">
-        <style>
-            body, html {{ margin: 0; padding: 0; height: 100%; background: #000; }}
-            #thumbnail {{ width: 100%; height: 100%; object-fit: contain; cursor: pointer; }}
-            #video {{ display: none; width: 100%; height: 100%; object-fit: contain; }}
-        </style>
-    </head>
-    <body>
-        <img id="thumbnail" src="{thumbnail}" onclick="playVideo()">
-        <video id="video" controls autoplay>
-            <source src="{url}" type="video/mp4">
-            Your browser does not support the video tag.
-        </video>
-        <script>
-            function playVideo() {{
-                document.getElementById('thumbnail').style.display = 'none';
-                document.getElementById('video').style.display = 'block';
-                document.getElementById('video').play();
-            }}
-        </script>
-    </body>
-    </html>
-    '''
-    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)
-    if '/pbxt/' not in decrypted_url:
-        return JSONResponse(content={"error": "Invalid encrypted URL"}, status_code=400)
-    path = decrypted_url.split('/pbxt/')[1]
-    return await handle_video_stream(path, request)
 async def get_cookies() -> Dict[str, str]:
     try:
@@ -970,20 +955,6 @@ async def retry_upload(upload_url: str, file_content: bytes, content_type: str,
     return False
-# Generate a secure secret key
-SECRET_KEY = secrets.token_urlsafe(32)
-def encrypt_url(url: str) -> str:
-    encoded = base64.urlsafe_b64encode(url.encode()).decode()
-    return encoded[::-1]  # Simple reversal for obfuscation
-def decrypt_url(encrypted_url: str) -> str:
-    try:
-        decoded = base64.urlsafe_b64decode(encrypted_url[::-1]).decode()
-        return decoded
-    except:
-        return ""
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)

+from fastapi import FastAPI, Request, HTTPException
+from fastapi.responses import JSONResponse, StreamingResponse
 import requests
 import time
 import asyncio
 from typing import Dict
 import base64
 import secrets
+from cryptography.fernet import Fernet
+from starlette.middleware.sessions import SessionMiddleware
 app = FastAPI()
+# Add SessionMiddleware to the application
+app.add_middleware(SessionMiddleware, secret_key=secrets.token_urlsafe(32))
+# Generate a secret key for Fernet encryption
+FERNET_KEY = Fernet.generate_key()
+fernet = Fernet(FERNET_KEY)
 HTML_CONTENT = """
 <!DOCTYPE html>
 <html lang="en">
     return StreamingResponse(generate(), status_code=response.status_code, headers=headers)
+@app.get("/encrypt")
+async def encrypt_url(request: Request, url: str):
+    session = request.session
+    session_id = secrets.token_urlsafe(16)
+    session[session_id] = url
+    encrypted_url = fernet.encrypt(f"{session_id}:{url}".encode()).decode()
+    return JSONResponse(content={"encrypted_url": f"/encrypted/{encrypted_url}"})
 @app.get("/encrypted/{encrypted_url}")
 async def handle_encrypted_url(encrypted_url: str, request: Request):
+    try:
+        decrypted = fernet.decrypt(encrypted_url.encode()).decode()
+        session_id, original_url = decrypted.split(':', 1)
+        if session_id not in request.session or request.session[session_id] != original_url:
+            raise HTTPException(status_code=403, detail="Invalid or expired encrypted URL")
+        if '/pbxt/' not in original_url:
+            raise HTTPException(status_code=400, detail="Invalid URL format")
+        path = original_url.split('/pbxt/')[1]
+        return await handle_video_stream(path, request)
+    except Exception as e:
+        raise HTTPException(status_code=400, detail="Invalid encrypted URL")
 async def get_cookies() -> Dict[str, str]:
     try:
     return False
 if __name__ == "__main__":
     import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)