competitions/cli/submit.py

import os
from argparse import ArgumentParser

import requests

from . import BaseCompetitionsCommand


def submit_commands_factory(args):
    return SubmitCompetitionAppCommand(args)


class SubmitCompetitionAppCommand(BaseCompetitionsCommand):
    def __init__(self, args):
        self.args = args

    @staticmethod
    def register_subcommand(parser: ArgumentParser):
        submit_competition_parser = parser.add_parser("submit", description="Submit to a competition")
        submit_competition_parser.add_argument(
            "--competition_id", type=str, help="ID of the competition, e.g. huggingface/cool-competition"
        )
        submit_competition_parser.add_argument(
            "--submission", type=str, help="Path to submission file or HuggingFace hub repo"
        )
        submit_competition_parser.add_argument("--comment", type=str, help="Submission comment", default="")
        submit_competition_parser.add_argument("--token", type=str, help="User token, read-only", default="")
        submit_competition_parser.set_defaults(func=submit_commands_factory)

    def run(self):
        if os.path.isfile(self.args.submission):
            files = {"submission_file": open(self.args.submission, "rb")}
            data = {
                "hub_model": "None",
                "submission_comment": self.args.comment,
            }
        else:
            files = {"submission_file": None}
            data = {
                "hub_model": self.args.submission,
                "submission_comment": self.args.comment,
            }

        headers = {"Authorization": f"Bearer {self.args.token}"}

        api_url = "https://" + self.args.competition_id.replace("/", "-") + ".hf.space/new_submission"

        response = requests.post(api_url, data=data, files=files, headers=headers)
        print(response.json())