Update index.js

index.js

@@ -21,46 +21,35 @@ function getFileSize(links) {
     return fileSize;
 }
 
-app.get('/', (req, res) => {
-    const shared = req.query.shared;
-    const filename = req.query.filename;
-
-    if (shared) {
-        const originalUrl = `https://huggingface.co/tranquan24/video/resolve/main/${shared}`;
-        const token = process.env.TOKEN;
-        const headers = {
-            'Authorization': `Bearer ${token}`,
-            'Range': req.headers['range'] || ''
-        };
-        
-        request.get({
-            url: originalUrl,
-            headers: headers
-        }).on('response', (originalResponse) => {
-            const links = [originalUrl];
-            const fileSize = getFileSize(links);
-
-            let headers_dict = originalResponse.headers;
-            headers_dict['Access-Control-Allow-Origin'] = '*';
-
-            if (filename) {
-                headers_dict['Content-Disposition'] = `attachment; filename="${filename}"`;
-            } else {
-                const filenameFromUrl = shared.split('/').pop();
-                headers_dict['Content-Disposition'] = `attachment; filename="${filenameFromUrl}"`;
-            }
-
-            
-            const { statusCode, headers } = originalResponse;
-            res.writeHead(statusCode, headers);
-            originalResponse.pipe(res);
-        }).on('error', (err) => {
-            console.error(err);
-            res.status(500).send('Internal Server Error');
-        });
-    } else {
-        res.status(400).send('Missing shared parameter');
-    }
+app.get('/:filename', (req, res) => {
+    const filename = req.params.filename;
+    
+    const originalUrl = `https://huggingface.co/tranquan24/video/resolve/main/${filename}`;
+    const token = process.env.TOKEN;
+    const headers = {
+        'Authorization': `Bearer ${token}`,
+        'Range': req.headers['range'] || ''
+    };
+    
+    request.get({
+        url: originalUrl,
+        headers: headers
+    }).on('response', (originalResponse) => {
+        const links = [originalUrl];
+        const fileSize = getFileSize(links);
+
+        let headers_dict = originalResponse.headers;
+        headers_dict['Access-Control-Allow-Origin'] = '*';
+
+        headers_dict['Content-Disposition'] = `inline; filename="${filename}"`;
+
+        const { statusCode, headers } = originalResponse;
+        res.writeHead(statusCode, headers);
+        originalResponse.pipe(res);
+    }).on('error', (err) => {
+        console.error(err);
+        res.status(500).send('Internal Server Error');
+    });
 });
 
 const port = 7860;