Update app.py

app.py

@@ -16,7 +16,8 @@ def images_to_video(folder_path, output_video):
     frame = cv2.imread(os.path.join(folder_path, images[0]))
     height, width, layers = frame.shape
 
-    video = cv2.VideoWriter(output_video, cv2.VideoWriter_fourcc(*'mpg2'), 1, (width, height))
+    # Change to MP4 format
+    video = cv2.VideoWriter(output_video, cv2.VideoWriter_fourcc(*'mp4v'), 1, (width, height))
 
     for image in images:
         video.write(cv2.imread(os.path.join(folder_path, image)))
@@ -48,10 +49,10 @@ if uploaded_file is not None:
     extract_zip(file_name, output_folder)
 
     # Convert to Video
-    output_video = f"{output_folder}.mpg"
+    output_video = f"{output_folder}.mp4"
     images_to_video(output_folder, output_video)
 
     # Generate download link
     video_file_base64 = get_video_base64(output_video)
-    href = f'<a href="data:file/mpg;base64,{video_file_base64}" download="{output_video}">Download Video File</a>'
-    st.markdown(href, unsafe_allow_html=True)
+    href = f'<a href="data:file/mp4;base64,{video_file_base64}" download="{output_video}">Download Video File</a>'
+    st.markdown(href, unsafe_allow_html=True)