Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
-import os 
+import os
+import shutil
 import yaml
 import tempfile
 import huggingface_hub
@@ -43,13 +44,43 @@ def print_directory_contents(path):
         for f in files:
             print(f"{subindent}{f}")
 
-# Path to the directory you want to print
-directory_path = './models'
+def check_outputs_folder(folder_path):
+    # Check if the folder exists
+    if os.path.exists(folder_path) and os.path.isdir(folder_path):
+        # Delete all contents inside the folder
+        for filename in os.listdir(folder_path):
+            file_path = os.path.join(folder_path, filename)
+            try:
+                if os.path.isfile(file_path) or os.path.islink(file_path):
+                    os.unlink(file_path)  # Remove file or link
+                elif os.path.isdir(file_path):
+                    shutil.rmtree(file_path)  # Remove directory
+            except Exception as e:
+                print(f'Failed to delete {file_path}. Reason: {e}')
+    else:
+        print(f'The folder {folder_path} does not exist.')
 
-# Print the directory contents
-print_directory_contents(directory_path)
+def check_for_mp4_in_outputs():
+    # Define the path to the outputs folder
+    outputs_folder = './outputs'
+    
+    # Check if the outputs folder exists
+    if not os.path.exists(outputs_folder):
+        return None
+    
+    # Check if there is a .mp4 file in the outputs folder
+    mp4_files = [f for f in os.listdir(outputs_folder) if f.endswith('.mp4')]
+    
+    # Return the path to the mp4 file if it exists
+    if mp4_files:
+        return os.path.join(outputs_folder, mp4_files[0])
+    else:
+        return None
 
-def infer(ref_video_in, ref_image_in):
+def infer(ref_video_in, ref_image_in, progress=gr.Progress()):
+    # check if 'outputs' dir exists and empty it if necessary
+    check_outputs_folder('./outputs')
+    
     # Create a temporary directory
     with tempfile.TemporaryDirectory() as temp_dir:
         print("Temporary directory created:", temp_dir)
@@ -115,15 +146,20 @@ def infer(ref_video_in, ref_image_in):
         stderr_thread.join()
     
         print("Inference script finished with return code:", process.returncode)
-        # Print the directory contents
+        
+        # Print the outputs directory contents
         print_directory_contents('./outputs')
+
+    # Call the function and print the result
+    mp4_file_path = check_for_mp4_in_outputs()
+    print(mp4_file_path)
     
-    return "done"
+    return mp4_file_path
 
 demo = gr.Interface(
     fn = infer,
     inputs = [gr.Video(), gr.Image(type="filepath")],
-    outputs = [gr.Textbox()]
+    outputs = [gr.Video()]
 )
 
 demo.launch()