Update server.js

server.js

@@ -6,9 +6,9 @@ const axios = require("axios");
 
 const app = express();
 const PORT = 7860;
+const DOMAIN = "https://reikerxx-ytm.hf.space";
 const DOWNLOADS_DIR = path.join(__dirname, "downloads");
 
-// Ensure the downloads directory exists
 if (!fs.existsSync(DOWNLOADS_DIR)) {
     fs.mkdirSync(DOWNLOADS_DIR, { recursive: true });
 }
@@ -16,20 +16,16 @@ if (!fs.existsSync(DOWNLOADS_DIR)) {
 async function getDownloadLink(youtubeUrl) {
     const browser = await chromium.launch({
         headless: true,
-        executablePath: "/usr/bin/chromium", // Use system Chromium path in Docker
+        executablePath: "/usr/bin/chromium",
         args: ["--no-sandbox", "--disable-setuid-sandbox"]
     });
     const page = await browser.newPage();
     let downloadLink = null;
 
     try {
-        // Open the conversion website
         await page.goto("https://ogmp3.com", { waitUntil: "domcontentloaded" });
-
-        // Enter the YouTube link
         await page.fill("#url", youtubeUrl);
 
-        // Intercept network responses to find the download URL
         page.on("response", async (response) => {
             const url = response.url();
             if (url.includes("safestytmp3.cc") && url.includes("/download/")) {
@@ -37,12 +33,8 @@ async function getDownloadLink(youtubeUrl) {
             }
         });
 
-        // Click the convert button
         await page.click("#convert-button");
-
-        // Wait up to 15 seconds for the link to appear
         await page.waitForTimeout(15000);
-
     } catch (error) {
         console.error("Error extracting download link:", error);
     } finally {
@@ -60,10 +52,7 @@ async function downloadMP3(url) {
             responseType: "stream",
         });
 
-        // Default filename
         let fileName = `download_${Date.now()}.mp3`;
-
-        // Extract filename from headers
         const contentDisposition = response.headers["content-disposition"];
         if (contentDisposition) {
             const match = contentDisposition.match(/filename="?(.+?)"?$/);
@@ -71,29 +60,33 @@ async function downloadMP3(url) {
                 fileName = match[1];
             }
         }
-
-        // Ensure it ends with .mp3
         if (!fileName.endsWith(".mp3")) {
             fileName += ".mp3";
         }
 
         const filePath = path.join(DOWNLOADS_DIR, fileName);
         const writer = fs.createWriteStream(filePath);
-
         response.data.pipe(writer);
 
         return new Promise((resolve, reject) => {
-            writer.on("finish", () => resolve(fileName));
+            writer.on("finish", () => {
+                setTimeout(() => {
+                    fs.unlink(filePath, (err) => {
+                        if (!err) {
+                            console.log(`Deleted: ${fileName}`);
+                        }
+                    });
+                }, 300000);
+                resolve(fileName);
+            });
             writer.on("error", reject);
         });
-
     } catch (error) {
         console.error("Error downloading MP3:", error.message);
         return null;
     }
 }
 
-// API Endpoint
 app.get("/api/q", async (req, res) => {
     const youtubeUrl = req.query.url;
 
@@ -108,9 +101,8 @@ app.get("/api/q", async (req, res) => {
 
         if (downloadLink) {
             const fileName = await downloadMP3(downloadLink);
-
             if (fileName) {
-                res.json({ success: true, file: `/downloads/${fileName}` });
+                res.json({ success: true, file: `${DOMAIN}/downloads/${fileName}` });
             } else {
                 res.status(500).json({ success: false, error: "Failed to download MP3" });
             }
@@ -123,10 +115,8 @@ app.get("/api/q", async (req, res) => {
     }
 });
 
-// Serve downloaded MP3s
 app.use("/downloads", express.static(DOWNLOADS_DIR));
 
-// Start Server
 app.listen(PORT, () => {
     console.log(`Server running on http://localhost:${PORT}`);
 });