Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -19,7 +19,8 @@ st.set_page_config(
|
|
| 19 |
)
|
| 20 |
|
| 21 |
def youtube_video_downloader(url):
|
| 22 |
-
yt_vid
|
|
|
|
| 23 |
title = yt_vid.title
|
| 24 |
vid_dld = (
|
| 25 |
yt_vid.streams.filter(progressive=True, file_extension="mp4")
|
|
@@ -54,12 +55,12 @@ def load_asr_model():
|
|
| 54 |
return asr_model
|
| 55 |
|
| 56 |
def transcribe_video(processed_audio):
|
| 57 |
-
|
| 58 |
transcriber_model = load_asr_model()
|
| 59 |
text_extract = transcriber_model(processed_audio)
|
| 60 |
-
et = time.time()
|
| 61 |
-
run_time = et -
|
| 62 |
-
return text_extract['text']
|
| 63 |
|
| 64 |
def generate_ai_summary(transcript):
|
| 65 |
model = google_genai.GenerativeModel('gemini-pro')
|
|
@@ -83,10 +84,10 @@ with youtube_url_tab:
|
|
| 83 |
with st.spinner("Transcribing..."):
|
| 84 |
audio = audio_extraction(os.fspath(yt_video), "mp3")
|
| 85 |
audio = audio_processing(audio)
|
| 86 |
-
ytvideo_transcript
|
| 87 |
st.success(f"Transcription successful")
|
| 88 |
st.write(ytvideo_transcript)
|
| 89 |
-
st.write(f'Completed in {run_time}')
|
| 90 |
|
| 91 |
if st.button("Generate Summary"):
|
| 92 |
summary = generate_ai_summary(ytvideo_transcript)
|
|
@@ -109,10 +110,9 @@ with file_select_tab:
|
|
| 109 |
with st.spinner("Transcribing..."):
|
| 110 |
audio = audio_extraction(video_file, "mp3")
|
| 111 |
audio = audio_processing(audio)
|
| 112 |
-
video_transcript
|
| 113 |
st.success(f"Transcription successful")
|
| 114 |
st.write(video_transcript)
|
| 115 |
-
st.write(f'Completed in {run_time}')
|
| 116 |
|
| 117 |
if st.button("Generate Summary", key="ti2"):
|
| 118 |
summary = generate_ai_summary(video_transcript)
|
|
@@ -130,10 +130,9 @@ with audio_file_tab:
|
|
| 130 |
if st.button("Transcribe", key="audiofile"):
|
| 131 |
with st.spinner("Transcribing..."):
|
| 132 |
processed_audio = audio_processing(audio_file)
|
| 133 |
-
audio_transcript
|
| 134 |
st.success(f"Transcription successful")
|
| 135 |
st.write(audio_transcript)
|
| 136 |
-
st.write(f'Completed in {run_time}')
|
| 137 |
|
| 138 |
|
| 139 |
if st.button("Generate Summary", key="ti1"):
|
|
|
|
| 19 |
)
|
| 20 |
|
| 21 |
def youtube_video_downloader(url):
|
| 22 |
+
yt_vid
|
| 23 |
+
= YouTube(url)
|
| 24 |
title = yt_vid.title
|
| 25 |
vid_dld = (
|
| 26 |
yt_vid.streams.filter(progressive=True, file_extension="mp4")
|
|
|
|
| 55 |
return asr_model
|
| 56 |
|
| 57 |
def transcribe_video(processed_audio):
|
| 58 |
+
# stime = time.time()
|
| 59 |
transcriber_model = load_asr_model()
|
| 60 |
text_extract = transcriber_model(processed_audio)
|
| 61 |
+
# et = time.time()
|
| 62 |
+
# run_time = et - stime
|
| 63 |
+
return text_extract['text']
|
| 64 |
|
| 65 |
def generate_ai_summary(transcript):
|
| 66 |
model = google_genai.GenerativeModel('gemini-pro')
|
|
|
|
| 84 |
with st.spinner("Transcribing..."):
|
| 85 |
audio = audio_extraction(os.fspath(yt_video), "mp3")
|
| 86 |
audio = audio_processing(audio)
|
| 87 |
+
ytvideo_transcript = transcribe_video(audio)
|
| 88 |
st.success(f"Transcription successful")
|
| 89 |
st.write(ytvideo_transcript)
|
| 90 |
+
# st.write(f'Completed in {run_time}')
|
| 91 |
|
| 92 |
if st.button("Generate Summary"):
|
| 93 |
summary = generate_ai_summary(ytvideo_transcript)
|
|
|
|
| 110 |
with st.spinner("Transcribing..."):
|
| 111 |
audio = audio_extraction(video_file, "mp3")
|
| 112 |
audio = audio_processing(audio)
|
| 113 |
+
video_transcript = transcribe_video(audio)
|
| 114 |
st.success(f"Transcription successful")
|
| 115 |
st.write(video_transcript)
|
|
|
|
| 116 |
|
| 117 |
if st.button("Generate Summary", key="ti2"):
|
| 118 |
summary = generate_ai_summary(video_transcript)
|
|
|
|
| 130 |
if st.button("Transcribe", key="audiofile"):
|
| 131 |
with st.spinner("Transcribing..."):
|
| 132 |
processed_audio = audio_processing(audio_file)
|
| 133 |
+
audio_transcript = transcribe_video(processed_audio)
|
| 134 |
st.success(f"Transcription successful")
|
| 135 |
st.write(audio_transcript)
|
|
|
|
| 136 |
|
| 137 |
|
| 138 |
if st.button("Generate Summary", key="ti1"):
|