Update from brunch

Dockerfile

@@ -34,6 +34,9 @@ RUN mkdir -p /workspace/hub/.cache && \
     mkdir -p /app/.cache && \
     chmod -R 777 /app/.cache
 
+# Copy token.json to /workspace
+COPY token.json /workspace/
+
 # Create a non-root user
 RUN useradd -m -u 1000 user && \
     chown -R user /workspace

app/utils/gmail_utils.py

@@ -12,10 +12,18 @@ SCOPES = [
 ]
 
 
+def get_token_path():
+    # Use /workspace/token.json if running in Docker/HF Space, else local
+    if os.environ.get("IS_HF_SPACE") == "1" or os.path.exists("/workspace"):
+        return "/workspace/token.json"
+    return "token.json"
+
+
 def authenticate_gmail(gmail_creds_path):
     creds = None
-    if os.path.exists('token.json'):
-        creds = Credentials.from_authorized_user_file('token.json', SCOPES)
+    token_path = get_token_path()
+    if os.path.exists(token_path):
+        creds = Credentials.from_authorized_user_file(token_path, SCOPES)
     if not creds or not creds.valid:
         if creds and creds.expired and creds.refresh_token:
             creds.refresh(Request())
@@ -27,7 +35,7 @@ def authenticate_gmail(gmail_creds_path):
             )
             creds = flow.run_local_server(port=8796, access_type='offline', prompt='consent')
         # Save the credentials for the next run
-        with open('token.json', 'w') as token:
+        with open(token_path, 'w') as token:
             token.write(creds.to_json())
     return creds