Update app.py

app.py

@@ -15,13 +15,13 @@ def run(*args):
     output = f"./output/{filename}"
     frame_processor = rest_args[0]
     selected_frame_processors = ' '.join(frame_processor)
-    
+
     face_analyser_direction = rest_args[1]
     face_analyser_age = rest_args[2]
     face_analyser_gender = rest_args[3]
-    
+
     cmd = (
-        f"python run.py --execution-providers cpu -s {source} -t {target} -o {output} "
+        f"python run.py --execution-providers cuda -s {source} -t {target} -o {output} "
         f"--frame-processors {selected_frame_processors} "
         f"--face-analyser-direction {face_analyser_direction} "
     )
@@ -29,7 +29,7 @@ def run(*args):
         cmd += f" --face-analyser-age {face_analyser_age}"
     if face_analyser_gender != 'none':
         cmd += f" --face-analyser-gender {face_analyser_gender}"
-    
+
     if len(rest_args) > 4:
         skip_audio = rest_args[4]
         keep_fps = rest_args[5]
@@ -40,21 +40,24 @@ def run(*args):
             cmd += " --keep-fps"
         if keep_temp:
             cmd += " --keep-temp"
-    
+
     try:
         print("Started...", cmd)
-        sp.run(cmd, shell=True, capture_output=True, text=True).stdout
-        
+        output_text = sp.run(cmd, shell=True, capture_output=True, text=True).stdout
+        print(output_text)
         return output
     except Exception as e:
         return f"An error occurred: {str(e)}"
-def clear_output(output):
-    if os.path.exists(output):
-        os.remove(output)
-        return "Output file deleted"
+def clear_output():
+    output_path = "./output"
+    if os.path.exists(output_path):
+        for filename in os.listdir(output_path):
+            file_path = os.path.join(output_path, filename)
+            if os.path.isfile(file_path):
+                os.remove(file_path)
+        return "Output files deleted"
     else:
-        return "Output file does not exist"
-
+        return "Output directory does not exist"
 
 def get_theme() -> gr.Theme:
     return gr.themes.Soft(
@@ -78,7 +81,7 @@ with gr.Blocks(theme=get_theme(), title="DeepFakeAI 1.0.0") as ui:
             label = 'FRAME PROCESSORS',
             value = ['face_swapper']  # Default value
         )
-        
+
     with gr.Box():
         with gr.Column(scale=3):
             face_analyser_direction_dropdown = gr.Dropdown(
@@ -100,16 +103,18 @@ with gr.Blocks(theme=get_theme(), title="DeepFakeAI 1.0.0") as ui:
         source_image = gr.Image(type="filepath", label="SOURCE IMAGE")
         target_image = gr.Image(type="filepath", label="TARGET IMAGE")
         image_button = gr.Button("START")
-        clear_button = gr.Button("CLEAR")
+        clear_button = gr.ClearButton(value="CLEAR")
         image_output = gr.Image(label="OUTPUT")
-        
+        clear_button.add(image_output)
+
         image_button.click(
-            run, 
+            run,
             inputs=[source_image, target_image, frame_processor_checkbox, face_analyser_direction_dropdown, face_analyser_age_dropdown, face_analyser_gender_dropdown],
             outputs=image_output
         )
-        clear_button.click(clear_output, inputs=[image_output])
-
+        print("Clearing output")
+        clear_button.click(fn=clear_output)
+        print("Clearing output Done")
     with gr.Tab("Video:               "):
         source_image_video = gr.Image(type="filepath", label="SOURCE IMAGE")
         target_video = gr.Video(label="TARGET VIDEO")
@@ -118,13 +123,13 @@ with gr.Blocks(theme=get_theme(), title="DeepFakeAI 1.0.0") as ui:
             keep_fps = gr.Checkbox(label="KEEP FPS")
             keep_temp = gr.Checkbox(label="KEEP TEMP")
         video_button = gr.Button("START")
-        clear_video_button = gr.Button("CLEAR")
-        video_output = gr.Video(label="OUTPUT")
+        clear_video_button = gr.ClearButton(value="CLEAR")
+        output_video = gr.Video(label="OUTPUT")
+        clear_video_button.add(output_video)
         video_button.click(
-            run, 
+            run,
             inputs=[source_image_video, target_video, frame_processor_checkbox, face_analyser_direction_dropdown, face_analyser_age_dropdown, face_analyser_gender_dropdown, skip_audio, keep_fps, keep_temp],
-            outputs=video_output
+            outputs=output_video
         )
-        clear_video_button.click(clear_output, inputs=[video_output])
-
+        clear_button.click(fn=clear_output)
 ui.launch(debug=True)