Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import time
 from transformers import pipeline
 from pytube import YouTube
 from pydub import AudioSegment
@@ -37,7 +38,7 @@ def audio_extraction(video_file, output_format):
         input_path=os.fspath(video_file), output_path=f"{str(video_file)[:-4]}.mp3", output_format=f"{output_format}"
     )
     return audio
-
+  
 
 def audio_processing(mp3_audio):
     audio = AudioSegment.from_file(mp3_audio, format="mp3")
@@ -52,9 +53,12 @@ def load_asr_model():
     return asr_model
     
 def transcribe_video(processed_audio):
+    st = time.now()
     transcriber_model = load_asr_model()
     text_extract = transcriber_model(processed_audio)
-    return text_extract['text']
+    et = time.now()
+    run_time = et - st 
+    return text_extract['text'], run_time
 
 def generate_ai_summary(transcript):
     model = google_genai.GenerativeModel('gemini-pro')
@@ -76,11 +80,13 @@ with youtube_url_tab:
         if url:
            if st.button("Transcribe", key="yturl"):
                with st.spinner("Transcribing..."):
-                   audio = audio_extraction(yt_video, "mp3")
+                   audio = audio_extraction(os.fspath(yt_video), "mp3")
                    audio = audio_processing(audio)
-                   ytvideo_transcript = transcribe_video(audio)
+                   ytvideo_transcript, run_time = transcribe_video(audio)
                st.success(f"Transcription successful")
                st.write(ytvideo_transcript)
+               st.write(f'Completed in {run_time}')
+               
                if st.button("Generate Summary"):
                    summary = generate_ai_summary(ytvideo_transcript)
                    st.write(summary)
@@ -102,12 +108,15 @@ with file_select_tab:
                with st.spinner("Transcribing..."):
                    audio = audio_extraction(video_file, "mp3")
                    audio = audio_processing(audio)
-                   video_transcript = transcribe_video(audio) 
+                   video_transcript, run_time = transcribe_video(audio)
                    st.success(f"Transcription successful")
                    st.write(video_transcript)
+                   st.write(f'Completed in {run_time}')
+                   
                    if st.button("Generate Summary", key="ti2"):
                        summary = generate_ai_summary(video_transcript)
                        st.write(summary)
+                       
 
     except Exception as e: 
         st.error(e)
@@ -120,9 +129,10 @@ with audio_file_tab:
         if st.button("Transcribe", key="audiofile"):
             with st.spinner("Transcribing..."):
                 processed_audio = audio_processing(audio_file)
-                audio_transcript = transcribe_video(processed_audio)
+                audio_transcript, run_time = transcribe_video(processed_audio)
                 st.success(f"Transcription successful")
                 st.write(audio_transcript)
+                st.write(f'Completed in {run_time}')
 
 
                 if st.button("Generate Summary", key="ti1"):