Upload app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
+from faster_whisper import WhisperModel
 import yt_dlp as youtube_dl
 import os
-import random
 
 def download_youtube_video(url: str, file_name: str) -> str:
     ydl_opts = {
@@ -17,56 +17,35 @@ def download_youtube_video(url: str, file_name: str) -> str:
         ydl.download([url])
         return f"{file_name}.mp3"
 
-# Initialize session state to track CAPTCHA solving status
-if "captcha_solved" not in st.session_state:
-    st.session_state.captcha_solved = False
+def transcribe_audio(file_name: str):
+    model = WhisperModel("medium", device="cpu", compute_type="float32")
+    segments, _ = model.transcribe(file_name)
+    return ''.join([segment.text for segment in segments])
 
-# Only generate CAPTCHA if it hasn't been solved yet
-if not st.session_state.captcha_solved:
-    num1 = random.randint(1, 10)
-    num2 = random.randint(1, 10)
-    captcha_result = num1 + num2
-    st.session_state.captcha_answer = captcha_result
-
-st.title("YouTube Audio Downloader")
+st.title("YouTube Video Transcription with Whisper")
 
 url = st.text_input("Enter YouTube URL:")
 
-# Display CAPTCHA only if it hasn't been solved
-if not st.session_state.captcha_solved:
-    user_captcha_input = st.text_input(f"What is {num1} + {num2}?")
-    if st.button("Verify CAPTCHA"):
+if st.button("Download Audio"):
+    if url:
+        st.write("Downloading audio...")
         try:
-            if int(user_captcha_input) == st.session_state.captcha_answer:
-                st.success("CAPTCHA solved successfully!")
-                st.session_state.captcha_solved = True
+            audio_file = download_youtube_video(url, "podcast_audio")
+            st.write("Audio downloaded successfully.")
+
+            if os.path.exists(audio_file):
+                with open(audio_file, "rb") as file:
+                    st.download_button(
+                        label="Download Audio",
+                        data=file,
+                        file_name="audio.mp3",
+                        mime="audio/mpeg"
+                    )
             else:
-                st.error("Incorrect CAPTCHA answer. Please try again.")
-        except ValueError:
-            st.error("Please enter a valid number for the CAPTCHA.")
+                st.error("Audio file was not saved correctly.")
+        except Exception as e:
+            st.error(f"An error occurred: {e}")
+    else:
+        st.error("Please enter a valid YouTube URL.")
 
-# Allow downloading if CAPTCHA has been solved
-if st.session_state.captcha_solved:
-    if st.button("Download Audio"):
-        if url:
-            st.write("Downloading audio...")
-            try:
-                audio_file = download_youtube_video(url, "audio")
-                st.write("Audio downloaded successfully.")
 
-                if os.path.exists(audio_file):
-                    with open(audio_file, "rb") as file:
-                        st.download_button(
-                            label="Download Audio",
-                            data=file,
-                            file_name="audio.mp3",
-                            mime="audio/mpeg"
-                        )
-                else:
-                    st.error("Audio file was not saved correctly.")
-            except youtube_dl.utils.DownloadError as e:
-                st.error("An error occurred while downloading the audio: The video might require additional authentication (e.g., YouTube login).")
-            except Exception as e:
-                st.error(f"An unexpected error occurred: {e}")
-        else:
-            st.error("Please enter a valid YouTube URL.")