Update app.py

app.py

@@ -24,13 +24,12 @@ def video_to_frames(video_path, output_folder):
         cv2.imwrite(frame_name, image)
         success, image = vidcap.read()
         count += 1
-        if count>550:
-          break
+        if count > 550:
+            break
     print(f"{count} frames extracted from {video_path}.")
-    return [count,fps]
+    return [count, fps]
 
-def frames_to_video(frame_folder, video_path, image_path, frame_count,fps):
-    print("ImagePath",image_path)
+def frames_to_video(frame_folder, video_path, image_path, frame_count, fps):
     global progress
     frames = [f for f in os.listdir(frame_folder) if f.endswith('.jpg')]
     frames.sort(key=lambda x: int(x.split('_')[1].split('.')[0]))  # Sort frames in ascending order
@@ -40,17 +39,15 @@ def frames_to_video(frame_folder, video_path, image_path, frame_count,fps):
 
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
     out = cv2.VideoWriter(video_path, fourcc, fps, (width, height))
+    
     providers = ["CUDAExecutionProvider"]
-    apps = insightface.app.FaceAnalysis(name='buffalo_l', providers=[provider])
-    swappers = insightface.model_zoo.get_model("inswapper_128.onnx", download=False, download_zip=False, providers=[provider])
-    face_enhancers = gfpgan.GFPGANer(model_path="GFPGANv1.4.pth", upscale=1, device='cuda')
+    app = insightface.app.FaceAnalysis(name='buffalo_l', providers=providers)
+    app.prepare(ctx_id=0, det_size=(640, 640))
+    swapper = insightface.model_zoo.get_model("inswapper_128.onnx", download=False, download_zip=False, providers=providers)
+    face_enhancer = gfpgan.GFPGANer(model_path="GFPGANv1.4.pth", upscale=1, device='cuda')
     
     for i in tqdm(range(frame_count), desc="Converting frames to video"):
-        print("Progress:",progress)
         img1 = cv2.imread(os.path.join(frame_folder, frames[i]))
-        #img2_pil = Image.open(image_path)
-        #img2_cv2 = cv2.cvtColor(np.array(img2_pil), cv2.COLOR_RGB2BGR)
-
         faces1 = app.get(img1)
         for _ in range(20):
             faces2 = app.get(image_path)
@@ -73,14 +70,19 @@ def frames_to_video(frame_folder, video_path, image_path, frame_count,fps):
     print(f"Video saved at {video_path}.")
 
 def face_swap(video_path, image_path):
+    global progress
+    progress = 0
     output_folder = "Out_Frames"
     frame_count = video_to_frames(video_path, output_folder)
     if frame_count[0] > 400:
         frame_count[0] = 400
-    output_video_path = "output_video.mp4" 
-    frames_to_video(output_folder, output_video_path, image_path, frame_count[0],frame_count[1])
-    return output_video_path 
-    
+    output_video_path = "output_video.mp4"
+    frames_to_video(output_folder, output_video_path, image_path, frame_count[0], frame_count[1])
+    return output_video_path
+
+def get_progress():
+    global progress
+    return progress
 
 iface = gr.Interface(
     fn=face_swap,
@@ -89,4 +91,14 @@ iface = gr.Interface(
     title="Profaker's Face Swap",
     description="Upload a video and an image. The faces in the video will be swapped with the face in the image.",
 )
-iface.launch(share=True)
+
+progress_iface = gr.Interface(
+    fn=get_progress,
+    inputs=[],
+    outputs="number",
+    live=True,
+    title="Progress Tracker"
+)
+
+iface.launch(share=True)
+progress_iface.launch(share=True)