Update 01_📼_Upload_Video_File.py

01_📼_Upload_Video_File.py

@@ -48,7 +48,12 @@ def main():
                 temp_file = tempfile.NamedTemporaryFile(delete=False)
                 temp_file.write(uploaded_file.read())
 
-                audio = mp.AudioFileClip(temp_file.name)
+                # Check if the video has audio
+                try:
+                    audio = mp.AudioFileClip(temp_file.name)
+                except Exception as e:
+                    audio = None
+                    st.warning("No audio found in the video. Proceeding without audio.")
 
                 # Open the video using cv2.VideoCapture
                 video = cv2.VideoCapture(temp_file.name)
@@ -111,9 +116,10 @@ def main():
                         # Convert the output video to a format compatible with Streamlit
                         converted_filename = "converted_output.mp4"
                         clip = mp.VideoFileClip(output_filename)
-                        clip = clip.set_audio(audio)
 
-                        #clip.write_videofile(converted_filename, codec="libx264")
+                        if audio is not None:
+                            clip = clip.set_audio(audio)
+
                         clip.write_videofile(converted_filename)
 
                         # Display the converted video using st.video()
@@ -139,4 +145,4 @@ if __name__ == "__main__":
         "image link](https://i.imgur.com/thJhzOO.png)](https://www.buymeacoffee.com/clementdelteil)")
     st.markdown(
         "###### [Blog post of the project](https://medium.com/geekculture/creating-a-web-app-to-colorize-images-and-youtube-videos-80f5be2d0f68)"
-    )
+    )