Update app.py

app.py

@@ -6,7 +6,7 @@ import warnings, tempfile
 from audio import gen_audio
 warnings.filterwarnings("ignore")
 warnings.simplefilter("ignore")
-import time
+import time, cv2, os
 
 videos = {
     # "教學視頻1": "teach/00.mp4",
@@ -46,8 +46,43 @@ def get_video_url(option):
     # print(videos[option])
     return videos[option]
 
+def flip_video_horizontally(video_path):
+    if not os.path.exists(video_path):
+        print(f"Error: {video_path} does not exist.")
+        return
+
+    cap = cv2.VideoCapture(video_path)
+    if not cap.isOpened():
+        print(f"Error: cannot open {video_path}.")
+        return
+
+    frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+    frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+    fps = int(cap.get(cv2.CAP_PROP_FPS))
+    fourcc = int(cap.get(cv2.CAP_PROP_FOURCC))
+
+    temp_output_path = video_path + "_temp.mp4"
+
+    out = cv2.VideoWriter(temp_output_path, fourcc, fps, (frame_width, frame_height))
+
+    while True:
+        ret, frame = cap.read()
+        if not ret:
+            break
+        flipped_frame = cv2.flip(frame, 1)
+        out.write(flipped_frame)
+
+    cap.release()
+    out.release()
+
+    os.replace(temp_output_path, video_path)
+    print(f"Finished flipping video：{video_path}")
+    
 @spaces.GPU()
 def evaluate_sign_language(user_video, standard_video_option):
+    if user_video.split('/')[-1].startswith("sample."):
+        flip_video_horizontally(user_video)
+        
     tmp = tempfile.TemporaryDirectory()
     tmpdir = tmp.name
     new_path = tmpdir + "/user.mp4"