New file : config.py (for models)

gemini_for_labels.py

@@ -12,9 +12,10 @@ import argparse
 import re
 from collections import defaultdict
 from concurrent.futures import ThreadPoolExecutor
+import config
 
-MODEL_ID = "gemini-3-flash-preview"
-api_key = os.environ["GEMINI_API_KEY"]
+MODEL_ID = config.MODEL_FLASH_ID
+api_key = config.API_KEY
 
 my_prompt = """I'm giving you an image of the left columns of a written exam.
 Students answer several exercises, which can have several questions.