Update app.py

app.py

@@ -125,12 +125,33 @@ def run_xportrait(source_image, driving_video, seed, uc_scale, best_frame, out_f
         video_files = glob.glob(os.path.join(output_dir, "*.mp4"))
         print(video_files)
         if video_files:
-            return f"Output video saved at: {video_files[0]}", video_files[0]
+            final_vid = convert_video_to_h264_aac(video_files[0])
+            return f"Output video saved at: {final_vid}", final_vid
         else:
             return "No video file was found in the output directory.", None
     except subprocess.CalledProcessError as e:
         return f"An error occurred: {e}", None
 
+def convert_video_to_h264_aac(video_path):
+    # Get the directory and original filename
+    original_dir = os.path.dirname(video_path)
+    original_name, _ = os.path.splitext(os.path.basename(video_path))
+    
+    # Define the output path in the same directory
+    output_path = os.path.join(original_dir, f"{original_name}_converted.mp4")
+    
+    # Load the video
+    with VideoFileClip(video_path) as video:
+        # Write the video with H.264 and AAC codecs
+        video.write_videofile(
+            output_path,
+            codec="libx264",      # H.264 video codec
+            audio_codec="aac",     # AAC audio codec
+            temp_audiofile="temp-audio.m4a",  # Temporary audio file (moviepy requirement)
+            remove_temp=True       # Remove temporary files after writing
+        )
+        
+    return output_path
 
 # Set up Gradio interface
 css="""