Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import os
 import tempfile
+import shutil
 
 # Function to execute the anime upscaler command
 def execute_upscaler(input_video, output_path, save_intermediate, async_mode):
@@ -11,39 +12,43 @@ def execute_upscaler(input_video, output_path, save_intermediate, async_mode):
     # Create a temporary directory to store the uploaded video
     temp_dir = tempfile.mkdtemp()
 
-    # Save the uploaded video to the temporary directory
-    input_video_path = os.path.join(temp_dir, "input_video.mp4")
-    input_video.save(input_video_path)
-
-    # Build the command
-    command = [
-        "python3",
-        "anime_upscaler.py",
-        "-m", model_path,
-        "-i", input_video_path,
-        "-o", output_path
-    ]
-
-    # Add optional flags
-    if save_intermediate:
-        command.append("-s")
-    if async_mode:
-        command.append("-a")
-
-    # Execute the command
-    import subprocess
-    result = subprocess.run(command, capture_output=True, text=True)
-
-    # Return the output of the command
-    return result.stdout
+    try:
+        # Save the uploaded video to the temporary directory
+        input_video_path = os.path.join(temp_dir, "input_video.mp4")
+        input_video.save(input_video_path)
+
+        # Build the command
+        command = [
+            "python3",
+            "anime_upscaler.py",
+            "-m", model_path,
+            "-i", input_video_path,
+            "-o", output_path
+        ]
+
+        # Add optional flags
+        if save_intermediate:
+            command.append("-s")
+        if async_mode:
+            command.append("-a")
+
+        # Execute the command
+        import subprocess
+        result = subprocess.run(command, capture_output=True, text=True)
+
+        # Return the output of the command
+        return result.stdout
+
+    finally:
+        # Cleanup: Remove the temporary directory
+        shutil.rmtree(temp_dir)
 
 # Define the Gradio interface
 iface = gr.Interface(
     fn=execute_upscaler,
     inputs=gr.Video(),
     outputs=gr.Video(),
-    live=True,
-    capture_session=True
+    live=True
 )
 
 # Launch the Gradio interface