push

gradio_app.py

@@ -77,6 +77,16 @@ def run_epic_inference(video_path, caption, motion_type):
         return logs, str(video_output_path)
     else:
         return f"Inference succeeded but no output video found in {output_dir}", None
+def print_output_directory(out_dir): 
+    result = ""
+    for root, dirs, files in os.walk(out_dir):
+        level = root.replace(out_dir, '').count(os.sep)
+        indent = ' ' * 4 * level
+        result += f"{indent}{os.path.basename(root)}/"
+        sub_indent = ' ' * 4 * (level + 1)
+        for f in files:
+            result += f"{sub_indent}{f}\n"
+    return result
 
 def inference(video_path, caption, motion_type):
     logs, video_masked = run_epic_inference(video_path, caption, motion_type)
@@ -121,20 +131,12 @@ def inference(video_path, caption, motion_type):
         print("Inference completed successfully.")
     else:
         print(f"Error occurred during inference: {result.stderr}")
-    
+
     # Print output directory contents
-    print(f"\nContents of output directory: {out_dir}")
-    for root, dirs, files in os.walk(out_dir):
-        level = root.replace(out_dir, '').count(os.sep)
-        indent = ' ' * 4 * level
-        print(f"{indent}{os.path.basename(root)}/")
-        sub_indent = ' ' * 4 * (level + 1)
-        for f in files:
-            print(f"{sub_indent}{f}")
-    
-        logs = result.stdout
+    logs = result.stdout
+    result = print_output_directory(out_dir)
 
-    return logs, str(f"{out_dir}/output.mp4")
+    return logs+result, str(f"{out_dir}/output.mp4")
 
 # -----------------------------
 # Step 3: Create Gradio UI