Update server.js

server.js

@@ -1,30 +1,61 @@
-'use strict';
+// File: app.js
 
-import express from 'express';
-import Lens from 'chrome-lens-ocr';
+import express from "express";
+import bodyParser  from "body-parser";
+import ytdl from "@distube/ytdl-core";
+import yts from "youtube-yts";
 
-// Constants
-const PORT = 7860;
-const HOST = '0.0.0.0';
-
-// App
 const app = express();
-app.get('/', (req, res) => {
-  res.send('Hello World from ExpressJS! This example is from the NodeJS Docs: https://nodejs.org/en/docs/guides/nodejs-docker-webapp/');
-});
-const lens = new Lens();
+const PORT = 3000;
+
+// Middleware
+app.use(bodyParser.json());
+
+// POST endpoint for downloading YouTube videos
+app.get("/download", async (req, res) => {
+  const { url, query } = req.body;
+
+  if (!url && !query) {
+    return res.status(400).json({ error: "Either 'url' or 'query' must be provided." });
+  }
 
-app.get('/scanByUrl', async (req, res) => {
-  const { url } = req.query;
   try {
-    const data = await lens.scanByURL(url);
-    const combinedText = data.segments.map(segment => segment.text).join('\n\n');
-    res.json({ combinedText, detailedData: data });
+    let videoInfo;
+    if (url) {
+      // Fetch video info using the provided URL
+      videoInfo = await ytdl.getInfo(url, {
+        playerClients: ["IOS", "WEB_CREATOR", "ANDROID", "WEB"],
+      });
+    } else {
+      // Search YouTube and get the first result
+      const searchResults = await yts(query);
+      if (!searchResults.videos || searchResults.videos.length === 0) {
+        return res.status(404).json({ error: "No videos found for the query." });
+      }
+      const firstResult = searchResults.videos[0];
+      videoInfo = await ytdl.getInfo(firstResult.url, {
+        playerClients: ["IOS", "WEB_CREATOR", "ANDROID", "WEB"],
+      });
+    }
+
+    // Get the best format with video and audio
+    const format = ytdl.filterFormats(videoInfo.formats, "videoandaudio")[0];
+    if (!format) {
+      return res.status(500).json({ error: "No suitable format found." });
+    }
+
+    // Respond with video URL and details
+    res.json({
+      title: videoInfo.videoDetails.title,
+      videoUrl: format.url,
+    });
   } catch (error) {
-    res.status(500).json({ error: error.message });
+    console.error(error);
+    res.status(500).json({ error: "Failed to process the request." });
   }
 });
 
-app.listen(PORT, HOST, () => {
-  console.log(`Running on http://${HOST}:${PORT}`);
+// Start the server
+app.listen(PORT, () => {
+  console.log(`Server is running on http://localhost:${PORT}`);
 });