import os

if __name__ == "__main__":
    video_dir = "../test_video"
    video_size = {}
    for video_file in os.listdir(video_dir):
        if video_file.endswith(".mp4"):
            size = os.path.getsize(os.path.join(video_dir, video_file))
            video_size[video_file] = size
    sorted_videos = sorted(video_size.items(), key=lambda x: x[1], reverse=True)

    duplicate_videos = []
    seen_sizes = set()
    for video, size in sorted_videos:
        if size in seen_sizes:
            duplicate_videos.append(video)
        else:
            seen_sizes.add(size)
    print("Duplicate videos based on size:")
    for idx, video in enumerate(duplicate_videos):
        # if idx == 0: continue
        print(video)
        os.remove(os.path.join(video_dir, video))

# if __name__ == "__main__":
#     video_dir1 = "../test_video1"
#     video_dir2 = "../test_video2"
#     video_name1 = []
#     for video_file in os.listdir(video_dir1):
#         if video_file.endswith(".mp4"):
#             video_name1.append(video_file)

#     video_name2 = []
#     for video_file in os.listdir(video_dir2):
#         if video_file.endswith(".mp4"):
#             video_name2.append(video_file)

#     duplicate_videos = []
#     for video in video_name1:
#         if video in video_name2:
#             duplicate_videos.append(video)
#     print("Duplicate videos based on name:")
#     for idx, video in enumerate(duplicate_videos):
#         # if idx == 0: continue
#         print(video)
#         os.remove(os.path.join(video_dir1, video))