Spaces:

uploadts
/

uploads

Sleeping

App Files Files Community

uploadts commited on Feb 27, 2025

Commit

7a319ff

verified ·

1 Parent(s): bdc505c

Update app.py

Browse files

Files changed (1) hide show

app.py +133 -15

app.py CHANGED Viewed

@@ -6,16 +6,21 @@ import asyncio
 from typing import Dict
 import os
 import mimetypes
 app = FastAPI()
 HTML_CONTENT = """
 <!DOCTYPE html>
 <html lang="en">
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Radd PRO Uploader</title>
     <link href="https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;600&display=swap" rel="stylesheet">
     <style>
         body {
@@ -445,7 +450,7 @@ HTML_CONTENT = """
             text-align: center;
         }
-                .quick-open-content img,
         .quick-open-content video,
         .quick-open-content audio {
             max-width: 100%;
@@ -534,7 +539,7 @@ HTML_CONTENT = """
 </head>
 <body>
     <div class="container">
-        <h1>Radd PRO Uploader</h1>
         <form id="uploadForm">
             <div id="dropZone" class="drop-zone">
                 <input type="file" name="file" id="file" class="file-input" accept="*" required>
@@ -583,6 +588,18 @@ HTML_CONTENT = """
         </div>
     </div>
     <script>
         const fileInput = document.getElementById('file');
         const fileName = document.getElementById('fileName');
@@ -592,10 +609,12 @@ HTML_CONTENT = """
         const resultContainer = document.getElementById('resultContainer');
         const dropZone = document.getElementById('dropZone');
         const modal = document.getElementById('embedModal');
         const historyModal = document.getElementById('historyModal');
         const quickOpenModal = document.getElementById('quickOpenModal');
         const span = document.getElementsByClassName("close");
         const embedLinkInput = document.getElementById('embedLink');
         const uploadBtn = document.getElementById('uploadBtn');
         const historyBtn = document.getElementById('historyBtn');
         const clearHistoryBtn = document.getElementById('clearHistoryBtn');
@@ -637,16 +656,14 @@ HTML_CONTENT = """
             }
         });
-        span[0].onclick = function() {
-            modal.style.display = "none";
-        }
-        span[1].onclick = function() {
-            historyModal.style.display = "none";
-        }
-        span[2].onclick = function() {
-            quickOpenModal.style.display = "none";
         }
         window.onclick = function(event) {
@@ -659,6 +676,9 @@ HTML_CONTENT = """
             if (event.target == quickOpenModal) {
                 quickOpenModal.style.display = "none";
             }
         }
         historyBtn.onclick = function() {
@@ -796,6 +816,14 @@ HTML_CONTENT = """
             };
             buttonsContainer.appendChild(copyBtn);
             const openBtn = document.createElement('button');
             openBtn.textContent = 'Open';
             openBtn.className = 'small-btn';
@@ -832,6 +860,33 @@ HTML_CONTENT = """
             alert('Embed link copied to clipboard!');
         }
         function saveToHistory(fileName, url, originalExtension) {
             let history = JSON.parse(localStorage.getItem('uploadHistory')) || [];
             history.unshift({ fileName, url, originalExtension, timestamp: new Date().toISOString() });
@@ -864,6 +919,15 @@ HTML_CONTENT = """
                 };
                 actionsContainer.appendChild(copyBtn);
                 const openBtn = document.createElement('button');
                 openBtn.textContent = 'Open';
                 openBtn.className = 'small-btn';
@@ -889,7 +953,7 @@ HTML_CONTENT = """
             historyModal.style.display = "block";
         }
-                function quickOpen(url, fileName, originalExtension) {
             quickOpenContent.innerHTML = '';
             const fullUrl = window.location.origin + url;
@@ -976,6 +1040,60 @@ async def handle_upload(file: UploadFile = File(...)):
     return JSONResponse(content={"url": mirrored_url, "originalExtension": original_extension})
 @app.get("/rbxg/{path:path}")
 async def handle_file_stream(path: str, request: Request):
     original_url = f'https://replicate.delivery/pbxt/{path}'
@@ -1099,4 +1217,4 @@ async def retry_upload(upload_url: str, file_content: bytes, content_type: str,
         await asyncio.sleep(delay)
         delay = min(delay * 2, 60)
-    return False

 from typing import Dict
 import os
 import mimetypes
+import secrets
+import json
 app = FastAPI()
+# Store shortened URLs
+SHORTENED_URLS = {}
 HTML_CONTENT = """
 <!DOCTYPE html>
 <html lang="en">
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>File Uploader</title>
     <link href="https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;600&display=swap" rel="stylesheet">
     <style>
         body {
             text-align: center;
         }
+        .quick-open-content img,
         .quick-open-content video,
         .quick-open-content audio {
             max-width: 100%;
 </head>
 <body>
     <div class="container">
+        <h1>File Uploader</h1>
         <form id="uploadForm">
             <div id="dropZone" class="drop-zone">
                 <input type="file" name="file" id="file" class="file-input" accept="*" required>
         </div>
     </div>
+    <div id="shortenModal" class="modal">
+        <div class="modal-content">
+            <span class="close">&times;</span>
+            <h2>Short Link</h2>
+            <p>Share this shortened link:</p>
+            <div class="embed-container">
+                <input type="text" id="shortenedLink" readonly>
+                <button onclick="copyShortenedLink()" class="small-btn">Copy</button>
+            </div>
+        </div>
+    </div>
     <script>
         const fileInput = document.getElementById('file');
         const fileName = document.getElementById('fileName');
         const resultContainer = document.getElementById('resultContainer');
         const dropZone = document.getElementById('dropZone');
         const modal = document.getElementById('embedModal');
+        const shortenModal = document.getElementById('shortenModal');
         const historyModal = document.getElementById('historyModal');
         const quickOpenModal = document.getElementById('quickOpenModal');
         const span = document.getElementsByClassName("close");
         const embedLinkInput = document.getElementById('embedLink');
+        const shortenedLinkInput = document.getElementById('shortenedLink');
         const uploadBtn = document.getElementById('uploadBtn');
         const historyBtn = document.getElementById('historyBtn');
         const clearHistoryBtn = document.getElementById('clearHistoryBtn');
             }
         });
+        // Close modals when clicking X
+        for (let i = 0; i < span.length; i++) {
+            span[i].onclick = function() {
+                modal.style.display = "none";
+                historyModal.style.display = "none";
+                quickOpenModal.style.display = "none";
+                shortenModal.style.display = "none";
+            }
         }
         window.onclick = function(event) {
             if (event.target == quickOpenModal) {
                 quickOpenModal.style.display = "none";
             }
+            if (event.target == shortenModal) {
+                shortenModal.style.display = "none";
+            }
         }
         historyBtn.onclick = function() {
             };
             buttonsContainer.appendChild(copyBtn);
+            const shortenBtn = document.createElement('button');
+            shortenBtn.textContent = 'Create Short Link';
+            shortenBtn.className = 'small-btn';
+            shortenBtn.onclick = () => {
+                createShortLink(url);
+            };
+            buttonsContainer.appendChild(shortenBtn);
             const openBtn = document.createElement('button');
             openBtn.textContent = 'Open';
             openBtn.className = 'small-btn';
             alert('Embed link copied to clipboard!');
         }
+        function copyShortenedLink() {
+            shortenedLinkInput.select();
+            document.execCommand('copy');
+            alert('Short link copied to clipboard!');
+        }
+        function createShortLink(url) {
+            fetch('/shorten', {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+                body: JSON.stringify({ url: url })
+            })
+            .then(response => response.json())
+            .then(data => {
+                if (data.shortUrl) {
+                    shortenedLinkInput.value = window.location.origin + data.shortUrl;
+                    shortenModal.style.display = "block";
+                }
+            })
+            .catch(error => {
+                console.error('Error creating short link:', error);
+                alert('Failed to create short link');
+            });
+        }
         function saveToHistory(fileName, url, originalExtension) {
             let history = JSON.parse(localStorage.getItem('uploadHistory')) || [];
             history.unshift({ fileName, url, originalExtension, timestamp: new Date().toISOString() });
                 };
                 actionsContainer.appendChild(copyBtn);
+                const shortenBtn = document.createElement('button');
+                shortenBtn.textContent = 'Short Link';
+                shortenBtn.className = 'small-btn';
+                shortenBtn.onclick = () => {
+                    createShortLink(item.url);
+                    historyModal.style.display = "none";
+                };
+                actionsContainer.appendChild(shortenBtn);
                 const openBtn = document.createElement('button');
                 openBtn.textContent = 'Open';
                 openBtn.className = 'small-btn';
             historyModal.style.display = "block";
         }
+        function quickOpen(url, fileName, originalExtension) {
             quickOpenContent.innerHTML = '';
             const fullUrl = window.location.origin + url;
     return JSONResponse(content={"url": mirrored_url, "originalExtension": original_extension})
+@app.post("/shorten")
+async def shorten_url(request: Request):
+    data = await request.json()
+    original_url = data.get("url")
+    if not original_url:
+        return JSONResponse(content={"error": "No URL provided"}, status_code=400)
+    # Generate a short code
+    short_code = ''.join(secrets.choice('abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789') for _ in range(6))
+    # Store the mapping
+    SHORTENED_URLS[short_code] = original_url
+    # Save to file for persistence
+    try:
+        with open("shortened_urls.json", "w") as f:
+            json.dump(SHORTENED_URLS, f)
+    except:
+        # Not critical if it fails to save
+        pass
+    return JSONResponse(content={"shortUrl": f"/s/{short_code}"})
+@app.get("/s/{short_code}")
+async def redirect_short_url(short_code: str):
+    # Try to load from file first (for persistence)
+    try:
+        if not SHORTENED_URLS and os.path.exists("shortened_urls.json"):
+            with open("shortened_urls.json", "r") as f:
+                shortened_urls_data = json.load(f)
+                SHORTENED_URLS.update(shortened_urls_data)
+    except:
+        pass
+    # Get the original URL
+    original_url = SHORTENED_URLS.get(short_code)
+    if not original_url:
+        return JSONResponse(content={"error": "Short URL not found"}, status_code=404)
+    # HTML for redirecting
+    html_content = f"""
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <meta http-equiv="refresh" content="0;url={original_url}">
+    </head>
+    <body>
+        <p>Redirecting...</p>
+    </body>
+    </html>
+    """
+    return HTMLResponse(content=html_content)
 @app.get("/rbxg/{path:path}")
 async def handle_file_stream(path: str, request: Request):
     original_url = f'https://replicate.delivery/pbxt/{path}'
         await asyncio.sleep(delay)
         delay = min(delay * 2, 60)
+    return False