Update main.py

main.py

@@ -105,11 +105,35 @@ def get_cached_1psidts_path(psid: str) -> str:
 	return os.path.join(GEMINI_COOKIE_PATH, f".cached_1psidts_{psid}.txt")
 
 
+def get_cookie_cache_json_path(psid: str) -> str:
+	"""Return the gemini-webapi cache path for persisted cookies."""
+	if not psid:
+		return ""
+	return os.path.join(GEMINI_COOKIE_PATH, f".cached_cookies_{psid}.json")
+
+
+def load_cached_cookie_value(psid: str, name: str) -> str:
+	"""Load a cookie value from gemini-webapi's persisted cookie cache."""
+	cached_file_path = get_cookie_cache_json_path(psid)
+	if not cached_file_path or not os.path.exists(cached_file_path):
+		return ""
+
+	try:
+		cookies = json.loads(Path(cached_file_path).read_text())
+		for cookie in cookies:
+			if cookie.get("name") == name:
+				return cookie.get("value", "")
+	except Exception as e:
+		logger.warning(f"Error reading cookie cache file {cached_file_path}: {e}")
+
+	return ""
+
+
 def load_cached_1psidts(psid: str) -> str:
 	"""Load a cached rotated 1PSIDTS value for the given 1PSID."""
 	cached_file_path = get_cached_1psidts_path(psid)
 	if not cached_file_path:
-		return ""
+		return load_cached_cookie_value(psid, "__Secure-1PSIDTS")
 
 	if os.path.exists(cached_file_path):
 		try:
@@ -119,7 +143,7 @@ def load_cached_1psidts(psid: str) -> str:
 		except Exception as e:
 			logger.warning(f"Error reading cache file {cached_file_path}: {e}")
 
-	return ""
+	return load_cached_cookie_value(psid, "__Secure-1PSIDTS")
 
 
 def get_cookie_value(cookies, name: str) -> str:
@@ -159,7 +183,8 @@ TEMPORARY_CHAT = os.environ.get("TEMPORARY_CHAT", "false").lower() == "true"
 AUTO_DELETE_CHAT = os.environ.get("AUTO_DELETE_CHAT", "true").lower() == "true" and not TEMPORARY_CHAT
 PUBLIC_BASE_URL = os.environ.get("PUBLIC_BASE_URL", "").rstrip("/")
 SECRET_FILE_PATH = os.path.join(os.path.dirname(__file__), "secrets", "proxy_secret")
-GEMINI_COOKIE_PATH = os.path.join(os.path.dirname(__file__), "secrets")
+DEFAULT_GEMINI_COOKIE_PATH = "/data/gemini_webapi" if os.path.isdir("/data") else os.path.join(os.path.dirname(__file__), "secrets")
+GEMINI_COOKIE_PATH = os.environ.get("GEMINI_COOKIE_PATH", DEFAULT_GEMINI_COOKIE_PATH)
 SESSION_VALIDATION_PROMPT = "Reply with exactly OK."
 AUTH_FAILURE_TEXT_PATTERNS = (
 	"are you signed in",
@@ -170,7 +195,7 @@ AUTH_FAILURE_TEXT_PATTERNS = (
 )
 DEFAULT_USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/144.0.0.0 Safari/537.36 Edg/144.0.0.0"
 
-os.environ.setdefault("GEMINI_COOKIE_PATH", GEMINI_COOKIE_PATH)
+os.environ["GEMINI_COOKIE_PATH"] = GEMINI_COOKIE_PATH
 
 
 async def background_delete_chat(client: GeminiClient, cid: str):
@@ -727,7 +752,12 @@ async def get_gemini_client():
 					logger.info("Initializing Gemini client using %s credentials", source)
 
 					tmp_client = GeminiClient(psid, psidts)
-					await tmp_client.init(timeout=300)
+					await tmp_client.init(
+						timeout=300,
+						auto_refresh=True,
+						refresh_interval=600,
+						auto_close=False,
+					)
 					await validate_gemini_client_session(tmp_client, source)
 
 					gemini_client = tmp_client