Update app.py

app.py

@@ -1,25 +1,27 @@
 import gradio as gr
 import subprocess as sp
 import os
+import uuid
 
 os.makedirs("./output", exist_ok=True)
 
 def run(*args):
-    source, target, *rest_args = args
+    source, target, unique_id, *rest_args = args
     if not os.path.exists(source):
         return "Source file does not exist"
     if not os.path.exists(target):
         return "Target file does not exist"
 
     filename = os.path.basename(target)
-    output = f"./output/{filename}"
+    os.makedirs(f"./output/{unique_id}",exist_ok=True)
+    output = f"./output/{unique_id}/{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"--frame-processors {selected_frame_processors} "
@@ -29,7 +31,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 +42,31 @@ 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"
-    else:
-        return "Output file does not exist"
 
+def clear_output(unique_id):
+    try:
+        output_path = f"./output/{unique_id}"
+        if os.path.exists(output_path):
+            print("Trying to delete ")
+            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)
+                    print(f"Output files in {output_path} are deleted")
+                    return "Output files for unique_id deleted"
+                else:
+                    print(f"Output files in {output_path} does not exist")
+                    return "Output directory for (output_path} does not exist"
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
 
 def get_theme() -> gr.Theme:
     return gr.themes.Soft(
@@ -78,7 +90,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(
@@ -96,19 +108,21 @@ with gr.Blocks(theme=get_theme(), title="DeepFakeAI 1.0.0") as ui:
                 choices = ['none'] + ['male', 'female'],
                 value = 'none'
             )
+            unique_id = gr.Textbox(value=str(uuid.uuid4()), visible=False)
     with gr.Tab("Image:               "):
         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, 
-            inputs=[source_image, target_image, frame_processor_checkbox, face_analyser_direction_dropdown, face_analyser_age_dropdown, face_analyser_gender_dropdown],
+            run,
+            inputs=[source_image, target_image, unique_id, 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])
+        clear_button.click(fn=clear_output, inputs=unique_id)
 
     with gr.Tab("Video:               "):
         source_image_video = gr.Image(type="filepath", label="SOURCE IMAGE")
@@ -118,13 +132,14 @@ 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")
+        clear_video_button = gr.ClearButton(value="CLEAR")
         video_output = gr.Video(label="OUTPUT")
+        clear_video_button.add(video_output)
         video_button.click(
-            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],
+            run,
+            inputs=[source_image_video, target_video, unique_id, frame_processor_checkbox, face_analyser_direction_dropdown, face_analyser_age_dropdown, face_analyser_gender_dropdown, skip_audio, keep_fps, keep_temp],
             outputs=video_output
         )
-        clear_video_button.click(clear_output, inputs=[video_output])
+        clear_video_button.click(fn=clear_output, inputs=unique_id)
 
 ui.launch(debug=True)