INNER CODE UNIT · Python
main
changyeyu/LLM-RL-Visualized · src/rename_images.py:119
def main():
parser = argparse.ArgumentParser(
description="Execute one of two functions based on command line parameter: add column to Excel or rename image files."
)
parser.add_argument(
"--func",
choices=["add_column", "rename_images"],
required=True,
help="Choose the function to run: 'add_column' to add a column to Excel, 'rename_images' to rename image files."
)
args = parser.parse_args()
if args.func == "add_column":
add_column_to_excel()
elif args.func == "rename_images":
rename_images()
if __name__ == "__main__":