Update main.py

main.py

@@ -411,27 +411,31 @@ class CoachingCodeRequest(BaseModel):
 
 # Function to get credentials
 def get_credentials():
+    """Get Google Sheets API credentials from environment variables."""
     try:
+        # Construct the service account info dictionary
         service_account_info = {
-            "type": userdata.get("SERVICE_ACCOUNT_TYPE"),
-            "project_id": userdata.get("PROJECT_ID"),
-            "private_key_id": userdata.get("PRIVATE_KEY_ID"),
-            "private_key": userdata.get("PRIVATE_KEY").replace('\\n', '\n'),
-            "client_email": userdata.get("CLIENT_EMAIL"),
-            "client_id": userdata.get("CLIENT_ID"),
-            "auth_uri": userdata.get("AUTH_URI"),
-            "token_uri": userdata.get("TOKEN_URI"),
-            "auth_provider_x509_cert_url": userdata.get("AUTH_PROVIDER_X509_CERT_URL"),
-            "client_x509_cert_url": userdata.get("CLIENT_X509_CERT_URL"),
-            "universe_domain": userdata.get("UNIVERSE_DOMAIN")
+            "type": os.getenv("SERVICE_ACCOUNT_TYPE"),
+            "project_id": os.getenv("PROJECT_ID"),
+            "private_key_id": os.getenv("PRIVATE_KEY_ID"),
+            "private_key": os.getenv("PRIVATE_KEY").replace('\\n', '\n'),
+            "client_email": os.getenv("CLIENT_EMAIL"),
+            "client_id": os.getenv("CLIENT_ID"),
+            "auth_uri": os.getenv("AUTH_URI"),
+            "token_uri": os.getenv("TOKEN_URI"),
+            "auth_provider_x509_cert_url": os.getenv("AUTH_PROVIDER_X509_CERT_URL"),
+            "client_x509_cert_url": os.getenv("CLIENT_X509_CERT_URL"),
+            "universe_domain": os.getenv("UNIVERSE_DOMAIN")
         }
         scope = ['https://spreadsheets.google.com/feeds', 'https://www.googleapis.com/auth/drive']
         creds = Credentials.from_service_account_info(service_account_info, scopes=scope)
         return creds
+
     except Exception as e:
         print(f"Error getting credentials: {e}")
         return None
 
+
 # Select files based on coaching code
 def select_files(coaching_code):
     creds = get_credentials()