from argparse import ArgumentParser
from huggingface_hub import HfApi


def upload_videos(api, upload_dir):
    api.upload_folder(
        folder_path=upload_dir,
        repo_id="jinggogogo/survey-videos",
        repo_type="dataset",
    )

def delete_optional_feedback(api):
    api.delete_folder("optional_feedback", "jinggogogo/survey-videos", repo_type="dataset")

def delete_response(api):
    api.delete_folder("response", "jinggogogo/survey-videos", repo_type="dataset")

def delete_videos(api, video_name):
    api.delete_file(path_in_repo=f'images/method_A/{video_name}', repo_id="jinggogogo/survey-videos", repo_type="dataset")
    api.delete_file(path_in_repo=f'images/method_B/{video_name}', repo_id="jinggogogo/survey-videos", repo_type="dataset")


def main():
    parser = ArgumentParser()
    parser.add_argument(
        "--upload_dir",
        type=str,
        default="survey-videos",
    )
    parser.add_argument("--upload_videos", type=eval, default=False, choices=[True, False])
    parser.add_argument("--delete_response", type=eval, default=False, choices=[True, False])
    parser.add_argument(
        "--delete_optional_feedback", type=eval, default=False, choices=[True, False]
    )
    args = parser.parse_args()

    api = HfApi()
    if args.upload_videos:
        upload_videos(api, args.upload_dir)

    if args.delete_response:
        answer = input("Are you sure you want to delete the response folder? (y/n)")
        if answer == "y":
            try:
                delete_response(api)
            except Exception as e:
                print(e)

    if args.delete_optional_feedback:
        answer = input("Are you sure you want to delete the optional_feedback folder? (y/n)")
        if answer == "y":
            try:
                delete_optional_feedback(api)
            except Exception as e:
                print(e)

    api.super_squash_history("jinggogogo/survey-videos", repo_type="dataset")


if __name__ == "__main__":
    main()