Add Literal type to imports for enhanced type hinting

main.py

@@ -7,7 +7,7 @@ from jose import JWTError, jwt
 from datetime import datetime, timedelta, timezone
 from openai import OpenAI
 from pathlib import Path
-from typing import List, Optional, Dict
+from typing import List, Optional, Dict, Literal
 from datasets import Dataset, load_dataset
 from sentence_transformers import SentenceTransformer
 from huggingface_hub import login