Upd dynamic local/cloud mode rotator from ENV

requirements-dev.txt

@@ -22,6 +22,7 @@ scipy>=1.10.0
 scikit-learn>=1.3.0
 sentencepiece
 sacremoses
+protobuf
 
 # Vietnamese translation (Opus model fallback)
 # Note: google-genai and google-auth packages are excluded for local mode

utils/cloud_llm.py

@@ -3,7 +3,27 @@ import os
 import logging
 import requests
 from typing import Optional
-from google import genai
+
+# Dynamic import for Google GenAI (only when not in local mode)
+def _import_google_genai():
+    """Dynamically import Google GenAI only when needed"""
+    try:
+        from google import genai
+        return genai
+    except ImportError as e:
+        raise ImportError(f"Google GenAI not available: {e}. Make sure IS_LOCAL=false and google-genai is installed.")
+
+# Check if we're in local mode
+IS_LOCAL = os.getenv("IS_LOCAL", "false").lower() == "true"
+
+# Only import Google GenAI if not in local mode
+if not IS_LOCAL:
+    try:
+        genai = _import_google_genai()
+    except ImportError:
+        genai = None
+else:
+    genai = None
 
 logger = logging.getLogger("llm")
 if not logger.handlers:
@@ -49,8 +69,13 @@ class GeminiClient:
     def __init__(self, rotator: KeyRotator, default_model: str):
         self.rotator = rotator
         self.default_model = default_model
+        self.available = genai is not None and not IS_LOCAL
 
     def generate(self, prompt: str, model: Optional[str] = None, temperature: float = 0.2, max_output_tokens: int = 512) -> Optional[str]:
+        if not self.available:
+            logger.warning("[LLM][Gemini] Google GenAI not available (local mode or import failed)")
+            return None
+            
         key = self.rotator.next_key()
         if not key:
             return None

utils/drive_saver.py

@@ -2,29 +2,51 @@
 import os, json, logging
 from typing import Optional
 
-# Conditional imports for Google Drive (only when needed)
-try:
-    from google.oauth2 import service_account
-    from googleapiclient.discovery import build
-    from googleapiclient.http import MediaFileUpload
-    GOOGLE_DRIVE_AVAILABLE = True
-except ImportError:
+# Check if we're in local mode
+IS_LOCAL = os.getenv("IS_LOCAL", "false").lower() == "true"
+
+# Conditional imports for Google Drive (only when not in local mode)
+if not IS_LOCAL:
+    try:
+        from google.oauth2 import service_account
+        from googleapiclient.discovery import build
+        from googleapiclient.http import MediaFileUpload
+        GOOGLE_DRIVE_AVAILABLE = True
+    except ImportError:
+        GOOGLE_DRIVE_AVAILABLE = False
+        # Create dummy classes for when Google Drive is not available
+        class service_account:
+            class Credentials:
+                @staticmethod
+                def from_service_account_info(*args, **kwargs):
+                    raise ImportError("Google Drive dependencies not available")
+        
+        class build:
+            @staticmethod
+            def build(*args, **kwargs):
+                raise ImportError("Google Drive dependencies not available")
+        
+        class MediaFileUpload:
+            def __init__(self, *args, **kwargs):
+                raise ImportError("Google Drive dependencies not available")
+else:
+    # Local mode: Google Drive is not available
     GOOGLE_DRIVE_AVAILABLE = False
-    # Create dummy classes for when Google Drive is not available
+    # Create dummy classes for local mode
     class service_account:
         class Credentials:
             @staticmethod
             def from_service_account_info(*args, **kwargs):
-                raise ImportError("Google Drive dependencies not available")
+                raise ImportError("Google Drive not available in local mode")
     
     class build:
         @staticmethod
         def build(*args, **kwargs):
-            raise ImportError("Google Drive dependencies not available")
+            raise ImportError("Google Drive not available in local mode")
     
     class MediaFileUpload:
         def __init__(self, *args, **kwargs):
-            raise ImportError("Google Drive dependencies not available")
+            raise ImportError("Google Drive not available in local mode")
 
 from utils.token import get_credentials
 
@@ -45,6 +67,11 @@ class DriveSaver:
         self.supports_all_drives = os.getenv("GDRIVE_FOLDER_IS_SHARED", "false").lower() in ("1","true","yes")
         self.allow_sa_fallback = os.getenv("GDRIVE_ALLOW_SA_FALLBACK", "false").lower() in ("1","true","yes")
         
+        # Check if we're in local mode
+        if IS_LOCAL:
+            logger.info("🏠 Local mode: Google Drive integration disabled")
+            return
+            
         # Check if Google Drive is available
         if not GOOGLE_DRIVE_AVAILABLE:
             logger.warning("⚠️ Google Drive dependencies not available - DriveSaver will be disabled")
@@ -55,6 +82,10 @@ class DriveSaver:
         self._initialize_service()
 
     def _initialize_service(self):
+        if IS_LOCAL:
+            logger.info("🏠 Local mode: Skipping Google Drive service initialization")
+            return
+            
         if not GOOGLE_DRIVE_AVAILABLE:
             logger.warning("⚠️ Google Drive dependencies not available - skipping service initialization")
             return
@@ -90,6 +121,10 @@ class DriveSaver:
         logger.info("✅ Google Drive service initialized")
 
     def upload_file_to_drive(self, file_path: str, folder_id: Optional[str] = None, mimetype: Optional[str] = None) -> bool:
+        if IS_LOCAL:
+            logger.info("🏠 Local mode: File upload to Google Drive skipped")
+            return False
+            
         if not GOOGLE_DRIVE_AVAILABLE:
             logger.warning("⚠️ Google Drive dependencies not available - upload skipped")
             return False
@@ -116,7 +151,7 @@ class DriveSaver:
             return False
 
     def is_service_available(self) -> bool:
-        return GOOGLE_DRIVE_AVAILABLE and self.service is not None
+        return not IS_LOCAL and GOOGLE_DRIVE_AVAILABLE and self.service is not None
 
     def set_folder_id(self, folder_id: str):
         self.folder_id = folder_id

utils/token.py

@@ -1,9 +1,49 @@
 # GCS credential token refresher
 import os, json, logging
 from typing import Optional
-from google.oauth2.credentials import Credentials
-from google_auth_oauthlib.flow import Flow
-from google.auth.transport.requests import Request
+
+# Dynamic imports for Google OAuth (only when not in local mode)
+def _import_google_oauth():
+    """Dynamically import Google OAuth libraries only when needed"""
+    try:
+        from google.oauth2.credentials import Credentials
+        from google_auth_oauthlib.flow import Flow
+        from google.auth.transport.requests import Request
+        return Credentials, Flow, Request
+    except ImportError as e:
+        raise ImportError(f"Google OAuth libraries not available: {e}. Make sure IS_LOCAL=false and google-auth packages are installed.")
+
+# Check if we're in local mode
+IS_LOCAL = os.getenv("IS_LOCAL", "false").lower() == "true"
+
+# Only import Google OAuth libraries if not in local mode
+if not IS_LOCAL:
+    try:
+        Credentials, Flow, Request = _import_google_oauth()
+    except ImportError:
+        # Create dummy classes for when Google OAuth is not available
+        class Credentials:
+            @staticmethod
+            def from_authorized_user_info(*args, **kwargs):
+                raise ImportError("Google OAuth not available")
+        class Flow:
+            @staticmethod
+            def from_client_config(*args, **kwargs):
+                raise ImportError("Google OAuth not available")
+        class Request:
+            pass
+else:
+    # Create dummy classes for local mode
+    class Credentials:
+        @staticmethod
+        def from_authorized_user_info(*args, **kwargs):
+            raise ImportError("Google OAuth not available in local mode")
+    class Flow:
+        @staticmethod
+        def from_client_config(*args, **kwargs):
+            raise ImportError("Google OAuth not available in local mode")
+    class Request:
+        pass
 
 logger = logging.getLogger("token")
 if not logger.handlers:
@@ -31,6 +71,10 @@ def _ensure_dirs():
         os.makedirs(base, exist_ok=True)
 
 def get_credentials() -> Optional[Credentials]:
+    if IS_LOCAL:
+        logger.info("🏠 Local mode: Google OAuth credentials not needed")
+        return None
+        
     # 1) Token file
     if os.path.exists(TOKEN_FILE):
         try:
@@ -68,6 +112,9 @@ def get_credentials() -> Optional[Credentials]:
     return None
 
 def build_auth_url(redirect_uri: str) -> str:
+    if IS_LOCAL:
+        raise RuntimeError("Google OAuth not available in local mode")
+        
     web = _load_oauth_client_web()
     if not web:
         raise RuntimeError("GDRIVE_CREDENTIALS_JSON missing or invalid ('web' section required)")
@@ -80,6 +127,9 @@ def build_auth_url(redirect_uri: str) -> str:
     return auth_url
 
 def exchange_code(code: str, redirect_uri: str) -> Credentials:
+    if IS_LOCAL:
+        raise RuntimeError("Google OAuth not available in local mode")
+        
     web = _load_oauth_client_web()
     if not web:
         raise RuntimeError("GDRIVE_CREDENTIALS_JSON missing or invalid ('web' section required)")