Update app.py

app.py

@@ -132,11 +132,6 @@ def create_subtitles(video_path):  # Renamed from generate_subtitles
         audio = AudioSegment.from_file(video_path)
         audio = audio.set_channels(1).set_frame_rate(44100).set_sample_width(2)
         audio.export(temp_audio.name, format='mp3', bitrate='128k')
-        
-        # Check if the audio is silent
-        if audio.dBFS < -90:  # Adjust the threshold as needed
-            os.unlink(temp_audio.name)  # Clean up temp file
-            return "No audio detected in the video."
 
         # Generate subtitles using Groq
         with open(temp_audio.name, 'rb') as audio_file:
@@ -163,7 +158,7 @@ def create_subtitles(video_path):  # Renamed from generate_subtitles
     return vtt_content
 
 
-def upload_video_to_hf(video_file, original_video_name, title, description, uploader, should_generate_subs=False, custom_thumbnail=None):  # Renamed parameter
+def upload_video_to_hf(video_file, original_video_name, title, description, uploader, should_generate_subs=False, custom_thumbnail=None):
     temp_dir = "temp"
     if not os.path.exists(temp_dir):
         os.makedirs(temp_dir)
@@ -193,26 +188,33 @@ def upload_video_to_hf(video_file, original_video_name, title, description, uplo
         
         video_length = get_video_length(video_path)
         
+        # Analyze audio level
+        audio = AudioSegment.from_file(video_path)
+        audio_dBFS = audio.dBFS
+        
         # Generate and upload subtitles if requested and video is not too long
         subtitle_location = ""
         if should_generate_subs and video_length <= 3600:  # 1 hour in seconds
-            try:
-                vtt_content = create_subtitles(video_path)  # Using renamed function
-                subtitle_name = f"{base_name}.vtt"
-                subtitle_path = os.path.join(temp_dir, subtitle_name)
-                
-                with open(subtitle_path, 'w') as f:
-                    f.write(vtt_content)
-                
-                subtitle_location = f"subtitles/{subtitle_name}"
-                hf_api.upload_file(
-                    path_or_fileobj=subtitle_path,
-                    path_in_repo=subtitle_location,
-                    repo_id="vericudebuget/ok4231",
-                    repo_type="space",
-                )
-            except Exception as e:
-                st.warning(f"Failed to generate subtitles: {str(e)}")
+            if audio_dBFS < -90:
+                subtitle_location = ""  # Set to empty if audio is too quiet
+            else:
+                try:
+                    vtt_content = create_subtitles(video_path)  # Using renamed function
+                    subtitle_name = f"{base_name}.vtt"
+                    subtitle_path = os.path.join(temp_dir, subtitle_name)
+                    
+                    with open(subtitle_path, 'w') as f:
+                        f.write(vtt_content)
+                    
+                    subtitle_location = f"subtitles/{subtitle_name}"
+                    hf_api.upload_file(
+                        path_or_fileobj=subtitle_path,
+                        path_in_repo=subtitle_location,
+                        repo_id="vericudebuget/ok4231",
+                        repo_type="space",
+                    )
+                except Exception as e:
+                    st.warning(f"Failed to generate subtitles: {str(e)}")
 
         # Upload video and thumbnail
         video_location = f"videos/{video_name}"