Enhance CLI error handling and improve argument parsing for tasks

medvqa/cli.py

@@ -1,18 +1,41 @@
 import argparse
 import subprocess
+import os
+
+
+report = '''\n⚠️⚠️⚠️\n
+Try installing latest version of the library by running the following command:
+\n pip install git+https://github.com/SushantGautam/MedVQA.git
+\n If you cannot solve the problem add an issue at https://github.com/SushantGautam/MedVQA/issues and report the log above! We will try to solve the problem as soon as possible.\n
+⚠️⚠️⚠️'''
+
 
 def main():
     print("MedVQA CLI")
     parser = argparse.ArgumentParser(description='MedVQA CLI')
-    parser.add_argument('competition', type=str, help='Name of the competition (e.g., gi-2025)')
-    parser.add_argument('task', type=int, choices=[1, 2], help='Task number (1 or 2)')
-    parser.add_argument('submission_repo', type=str, help='Path to the submission repository')
+    parser.add_argument('competition', type=str,
+                        help='Name of the competition (e.g., gi-2025)')
+    parser.add_argument('task', type=str, help='Task number (1 or 2)')
+    parser.add_argument('submission_repo', type=str,
+                        help='Path to the submission repository')
 
     args = parser.parse_args()
+    print("Running with arguments:", args)
 
-    script_path = f'medvqa/{args.competition}/task{args.task}/run.py'
+    # Check if competition directory exists
+    competition_dir = os.path.join(
+        '/Users/sgautam/Documents/MedVQA', args.competition)
+    if not os.path.isdir(competition_dir):
+        raise FileNotFoundError(
+            f"Competition '{args.competition}' does not exist! Need to update library?"+report)
 
+    # Check if task file exists
+    task_file = os.path.join(competition_dir, f'task{args.task}', 'run.py')
+    if not os.path.isfile(task_file):
+        raise FileNotFoundError(f"Task '{args.task}' does not exist!"+report)
+    script_path = f'medvqa/{args.competition}/task{args.task}/run.py'
     subprocess.run(['python', script_path, args.submission_repo])
 
+
 if __name__ == '__main__':
     main()

medvqa/gi-2025/task1/run.py

@@ -2,19 +2,18 @@ import sys
 import argparse
 
 
-def get_parser():
-    parser = argparse.ArgumentParser(description='Task 1 specific arguments')
-    parser.add_argument('--example_arg', type=str,
-                        help='An example argument for Task 1')
-    return parser
-
-
-def main(submission_repo, example_arg):
-    print(
-        f"Running GI-2025 Task 1 with repository: {submission_repo} and example_arg: {example_arg}")
+def main(repo, task_name, verbose=False):
+    print(f"Running {task_name} with repository: {repo}")
+    if verbose:
+        print("Verbose mode is enabled")
 
 
 if __name__ == '__main__':
-    parser = get_parser()
+    parser = argparse.ArgumentParser(description='Run GI-1015 Task 1 (VQA)')
+    parser.add_argument('repo1', type=str, help='Repository path')
+    parser.add_argument('task_name1', type=str, help='Name of the task')
+    parser.add_argument('--verbose1', action='store_true',
+                        help='Enable verbose mode')
+
     args = parser.parse_args()
-    main(args.submission_repo, args.example_arg)
+    main(args.repo1, args.task_name1, args.verbose1)

medvqa/gi-2025/task2/run.py

@@ -1,7 +1,19 @@
 import sys
+import argparse
+
+
+def main(repo, task_name, verbose=False):
+    print(f"Running {task_name} with repository: {repo}")
+    if verbose:
+        print("Verbose mode is enabled")
 
-def main(repo):
-    print(f"Running GI-2025 Task 2 with repository: {repo}")
 
 if __name__ == '__main__':
-    main(sys.argv[1])
+    parser = argparse.ArgumentParser(description='Run GI-2025 Task 1 (VQA)')
+    parser.add_argument('repo2', type=str, help='Repository path')
+    parser.add_argument('task_name2', type=str, help='Name of the task')
+    parser.add_argument('--verbose2', action='store_true',
+                        help='Enable verbose mode')
+
+    args = parser.parse_args()
+    main(args.repo2, args.task_name2, args.verbose2)