Update functions.py

functions.py

@@ -3,7 +3,7 @@ import os
 import random
 import openai
 import yt_dlp
-from pytube import YouTube
+from pytube import YouTube, extract
 import pandas as pd
 import plotly_express as px
 import nltk
@@ -225,10 +225,11 @@ def inference(link, upload, _asr_model):
                 chunks = song[::twenty_minutes]
                 
                 transcriptions = []
-                
+
+                video_id = extract.video_id(link)
                 for i, chunk in enumerate(chunks):
-                    chunk.export(f'output/chunk_{i}_{link}.mp4', format='mp4')
-                    transcriptions.append(load_whisper_api(f'output/chunk_{i}_{link}.mp4')['text'])
+                    chunk.export(f'output/chunk_{i}_{video_id}.mp4', format='mp4')
+                    transcriptions.append(load_whisper_api(f'output/chunk_{i}_{video_id}.mp4')['text'])
     
                 results = ','.join(transcriptions)
 
@@ -263,10 +264,10 @@ def inference(link, upload, _asr_model):
                 transcriptions = []
 
                 st.info("`Transcribing uploaded audio...`")
-                
+
                 for i, chunk in enumerate(chunks):
                     chunk.export(f'output/chunk_{i}.mp4', format='mp4')
-                    transcriptions.append(load_whisper_api('output/chunk_{i}.mp4')['text'])
+                    transcriptions.append(load_whisper_api(f'output/chunk_{i}.mp4')['text'])
     
                 results = ','.join(transcriptions)