Update index.js

index.js

@@ -86,38 +86,42 @@ app.get('/', (req, res) => {
 });
 
 app.post('/extract-transcript', async (req, res) => {
-    const { videoUrl, videoTitle } = req.body;
-    if (!videoUrl || !videoTitle) {
-        return res.status(400).send('videoUrl and videoTitle are required');
+    const { videoUrl } = req.body;
+
+    if (!videoUrl) {
+        return res.status(400).send('videoUrl is required');
     }
-    
+
     const browser = await chromium.launch();
     const context = await browser.newContext();
     const page = await context.newPage();
 
     try {
         await page.goto(videoUrl, { waitUntil: 'networkidle' });
-        
+
         // Set viewport size
         await page.setViewportSize({ width: 1920, height: 1080 });
-        
-        // Click the "Expand" button to expand the video description
-        await page.click('tp-yt-paper-button#expand');
-        
-        // Wait for the "Show transcript" button and click it
+
+        // Try to expand description if button is present (optional)
+        const expandButton = await page.$('tp-yt-paper-button#expand');
+        if (expandButton) {
+            await expandButton.click();
+        }
+
+        // Click the "Show transcript" button
         await page.click('button[aria-label="Show transcript"]');
-        
+
         // Wait for the transcript container to appear
         await page.waitForSelector('ytd-transcript-segment-list-renderer');
-        
+
         // Extract the transcript text
         const transcript = await page.evaluate(() => {
             const elements = Array.from(document.querySelectorAll('ytd-transcript-segment-renderer .segment-text'));
             return elements.map(element => element.innerText).join('\n');
         });
-        
+
         res.json({ transcript });
-        
+
     } catch (error) {
         console.error('Error extracting transcript:', error);
         res.status(500).send('Error extracting transcript');
@@ -125,7 +129,6 @@ app.post('/extract-transcript', async (req, res) => {
         await browser.close();
     }
 });
-
 const PORT = 7860;
 app.listen(PORT, () => {
     console.log(`Server is running on port ${PORT}`);