tensorkelechi commited on
Commit
023ef49
·
verified ·
1 Parent(s): c0f6850

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -11
app.py CHANGED
@@ -19,7 +19,8 @@ st.set_page_config(
19
  )
20
 
21
  def youtube_video_downloader(url):
22
- yt_vid = YouTube(url)
 
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
- st = time.time()
58
  transcriber_model = load_asr_model()
59
  text_extract = transcriber_model(processed_audio)
60
- et = time.time()
61
- run_time = et - st
62
- return text_extract['text'], run_time
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, run_time = transcribe_video(audio)
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, run_time = transcribe_video(audio)
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, run_time = transcribe_video(processed_audio)
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"):