Update main.py

main.py

@@ -6,6 +6,7 @@ from google.oauth2.service_account import Credentials
 import pandas as pd
 from collections import defaultdict
 from google.colab import userdata
+import os
 
 # Initialize the FastAPI app
 app = FastAPI()