Update src/streamlit_app.py

src/streamlit_app.py

@@ -25,15 +25,15 @@ MAX_LENGTH = RATE_HZ * MAX_SECONDS
 def download_video(url, output_dir="/app/tmp"):
     os.makedirs(output_dir, exist_ok=True)
     ydl_opts = {
-        'format': 'worstvideo[ext=mp4]+bestaudio[ext=m4a]/bestaudio',
-        "outtmpl": os.path.join(output_dir, "video.%(ext)s"),
-        "quiet": True,
-        'merge_output_format': 'mp4',
-        'quiet': True,
-        'noplaylist': True,
-        'nocheckcertificate': True,
-        'retries': 3,
-    }
+            'format': 'worstvideo[ext=mp4]+bestaudio[ext=m4a]/bestaudio',
+            "outtmpl": os.path.join(output_dir, "video.%(ext)s"),
+            "quiet": True,
+            'merge_output_format': 'mp4',
+            'quiet': True,
+            'noplaylist': True,
+            'nocheckcertificate': True,
+            'retries': 3,
+            }
 
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         ydl.download([url])
@@ -43,12 +43,12 @@ def extract_audio(input_path, output_dir="/app/tmp"):
     os.makedirs(output_dir, exist_ok=True)
     output_path = os.path.join(output_dir, "audio.mp3")
     (
-        ffmpeg
-        .input(input_path)
-        .output(output_path, format='mp3', acodec='libmp3lame', audio_bitrate='192k')
-        .overwrite_output()
-        .run(quiet=True)
-    )
+            ffmpeg
+            .input(input_path)
+            .output(output_path, format='mp3', acodec='libmp3lame', audio_bitrate='192k')
+            .overwrite_output()
+            .run(quiet=True)
+            )
     return output_path
 
 # Split files by chunks with == MAX_LENGTH size
@@ -104,20 +104,21 @@ if st.button("Analyze"):
     video_path = None
     output_dir="/app/tmp"
     os.makedirs(output_dir, exist_ok=True)
-    
+
     if uploaded_file:
         video_path = os.path.join(output_dir, "video.mp4")
             with open(video_path, "wb") as f:
-        f.write(uploaded_file.read())
+                f.write(uploaded_file.read())
         st.success("✅ Video uploaded successfully.")
     elif video_url.strip():
         with st.spinner("Downloading video from URL..."):
             try:
                 video_path = download_video(video_url)
-                st.success(f"✅ Video downloaded: {video_path}")
             except Exception as e:
                 st.error(f"❌ Failed to download video: {e}")
-            st.warning("Please enter a valid URL.")
+            else:
+                st.success(f"✅ Video downloaded: {video_path}")
+                
     else:
         st.warning("⚠️ Please upload a video file or enter a valid URL.")
 
@@ -129,7 +130,7 @@ if st.button("Analyze"):
             st.write("Exists:", os.path.exists(audio_path))
 
         # with st.spinner("Transcribing with Whisper..."):
-        #     whisper_model = whisper.load_model("base")
+            #     whisper_model = whisper.load_model("base")
         #     result = whisper_model.transcribe(audio_path)
         #     transcription = result['text']
             # transcription = "Hello There"
@@ -143,10 +144,10 @@ if st.button("Analyze"):
             waves = f"{np.concatenate(audio_df["audio"][:5].to_list())}"
             st.markdown("**Audio waves:**")
             st.text_area("Audio waves", waves, height=200)
-            
+
             # audio_df = split_audio(audio_path)
             # print(np.concatenate(audio_df["audio"][:50].to_list()))
-         
+
             accent_data = {"label": "American", "score": 0.9}
             accent = accent_data.get("label", "American")
             confidence = accent_data.get("score", 0.0)