import argparse from copaint.gradio_ui import build_gradio_ui import os def main(): parser = argparse.ArgumentParser() parser.add_argument( "-s", "--share", action=argparse.BooleanOptionalAction, # enables --share / --no-share default=True, help="Share the app publicly (default: true)" ) parser.add_argument( "-l", "--language", type=str, default="fr", choices=["en", "fr"], help="Language to use (default: en)" ) args = parser.parse_args() demo = build_gradio_ui(language=args.language) # check if /dev/shm exists if os.path.exists("/dev/shm"): demo.launch(share=args.share, allowed_paths=["/dev/shm"]) else: demo.launch(share=args.share) if __name__ == "__main__": main()