Upload app.py

app.py

@@ -51,18 +51,42 @@ VOICE_OPTIONS = [
     {"id": "shimmer", "name": "shimmer"},
 ]
 
-# 添加 API Keys 轮询功能
-def get_next_gemini_key(api_keys):
-    keys = [k.strip() for k in api_keys.split(',') if k.strip()]
-    if not hasattr(get_next_gemini_key, 'current_index'):
-        get_next_gemini_key.current_index = 0
-    key = keys[get_next_gemini_key.current_index]
-    get_next_gemini_key.current_index = (get_next_gemini_key.current_index + 1) % len(keys)
-    return key
+def get_next_gemini_key(api_keys_str):
+    """从多个key中轮询获取下一个key"""
+    if not hasattr(get_next_gemini_key, '_current_index'):
+        get_next_gemini_key._current_index = 0
+    
+    keys = [k.strip() for k in api_keys_str.split(',') if k.strip()]
+    if not keys:
+        return None
+        
+    key = keys[get_next_gemini_key._current_index]
+    get_next_gemini_key._current_index = (get_next_gemini_key._current_index + 1) % len(keys)
+    return key.strip()
 
 def get_api_key(key_name, ui_value):
-    if key_name == "GEMINI_API_KEY" and ui_value and ',' in ui_value:
-        return get_next_gemini_key(ui_value)
+    """获取API key的新逻辑"""
+    if key_name == "GEMINI_API_KEY":
+        # 1. 优先使用UI传入的key(s)
+        if ui_value:
+            if ',' in ui_value:
+                # 多个key的情况
+                selected_key = get_next_gemini_key(ui_value)
+            else:
+                # 单个key的情况
+                selected_key = ui_value.strip()
+        # 2. 其次使用GEMINI_API_KEYS环境变量
+        elif os.getenv("GEMINI_API_KEYS"):
+            selected_key = get_next_gemini_key(os.getenv("GEMINI_API_KEYS"))
+        # 3. 最后使用GEMINI_API_KEY环境变量
+        else:
+            selected_key = os.getenv("GEMINI_API_KEY")
+            
+        # 设置当前使用的key到GEMINI_API_KEY环境变量
+        if selected_key:
+            os.environ["GEMINI_API_KEY"] = selected_key
+        return selected_key
+        
     return ui_value if ui_value else os.getenv(key_name)
 
 def process_inputs(