Update app.py

app.py

@@ -367,10 +367,10 @@ def not_supported(key: str) -> Dict[str, Any]:
 
 
 # ─────────────────────────────────────────
-#  단일 키 비동기 처리 (업데이트됨)
+#  단일 키 비동기 처리 (업데이트됨 - OpenAI, Anthropic 작동 플래그 추가)
 # ─────────────────────────────────────────
 async def process_single_key(key: str, rate_limit: bool, claude_model: str) -> Dict[str, Any]:
-    """주어진 key 하나를 분석해 정보 dict 반환. Gemini 키의 작동 여부 플래그 포함."""
+    """주어진 key 하나를 분석해 정보 dict 반환. 작동하는 키들의 플래그 포함."""
     _key = key.strip()
 
     if not _key:
@@ -379,58 +379,63 @@ async def process_single_key(key: str, rate_limit: bool, claude_model: str) -> D
     # OpenRouter
     if re.match(re.compile(r"sk-or-v1-[a-z0-9]{64}"), _key):
         result = get_key_openrouter_info(_key)
-        return {"key": _key, **result} # key_type은 get_key_openrouter_info에서 반환
+        return {"key": _key, **result}
 
-    # Anthropic Claude
+    # Anthropic Claude (작동 플래그 추가)
     if re.match(re.compile(r"sk-ant-api03-[a-zA-Z0-9\-_]{93}AA"), _key) or \
        (_key.startswith("sk-ant-") and len(_key) == 93) or \
        (len(_key) == 89 and re.match(re.compile(r"sk-[a-zA-Z0-9]{86}"), _key)):
         result = await get_key_ant_info(_key, rate_limit, claude_model)
-        return {"key": _key, **result} # key_type 포함
+        # Anthropic 키 작동 여부 판단: key_availability가 True이고 status에 오류가 없으면 작동
+        is_working = result.get("key_availability") and \
+                    not any(error_word in str(result.get("status", "")).lower() 
+                           for error_word in ["error", "invalid", "expired", "forbidden", "unauthorized"])
+        result["is_anthropic_working"] = is_working
+        return {"key": _key, **result}
 
     # Stability
     if re.match(re.compile(r"sk-[a-zA-Z0-9]{48}"), _key) and len(_key) == 51 and "T3BlbkFJ" not in _key:
         result = get_key_stability_info(_key)
-        return {"key": _key, **result} # key_type 포함
+        return {"key": _key, **result}
 
     # Deepseek
     if re.match(re.compile(r"sk-[a-f0-9]{32}"), _key):
          result = get_key_deepseek_info(_key)
-         return {"key": _key, **result} # key_type 포함
+         return {"key": _key, **result}
 
-    # OpenAI (다른 sk- 패턴보다 뒤에 와야 함)
+    # OpenAI (다른 sk- 패턴보다 뒤에 와야 함) (작동 플래그 추가)
     if _key.startswith("sk-"):
         result = get_key_oai_info(_key)
-        return {"key": _key, **result} # key_type 포함
+        # OpenAI 키 작동 여부 판단: key_availability가 True이면 작동
+        is_working = result.get("key_availability") == True
+        result["is_openai_working"] = is_working
+        return {"key": _key, **result}
 
-    # Google Gemini 처리 (수정됨)
+    # Google Gemini 처리 (기존 유지)
     if _key.startswith("AIzaSy"):
-        gemini_info = get_key_gemini_info(_key) # 원본 함수 호출
-        # 작동 여부 플래그 추가 (status 확인)
+        gemini_info = get_key_gemini_info(_key)
         is_working = gemini_info.get("key_availability") and gemini_info.get("status") == "Working"
-        # 결과 딕셔너리 구성: 원본 정보 + key_type + 작동 여부 플래그
-        # gemini_info에 이미 'key'가 포함되어 있음
         result = {
-            "key_type": "Google Gemini", # 타입 정보 명시적 추가
+            "key_type": "Google Gemini",
             **gemini_info,
-            "is_gemini_working": is_working # 작동 여부 플래그
+            "is_gemini_working": is_working
         }
         return result
 
     # NovelAI
     if _key.startswith("pst-"):
         result = get_key_nai_info(_key)
-        return {"key": _key, **result} # key_type 포함
+        return {"key": _key, **result}
 
     # Replicate
     if (_key.startswith("r8_") and len(_key) == 40) or (_key.islower() and len(_key) == 40):
         result = get_key_replicate_info(_key)
-        return {"key": _key, **result} # key_type 포함
+        return {"key": _key, **result}
 
     # xAI
     if _key.startswith("xai-"):
         result = get_key_xai_info(_key)
-        return {"key": _key, **result} # key_type 포함
+        return {"key": _key, **result}
 
     # Azure endpoint: "name:key"
     if len(_key.split(":")) == 2:
@@ -439,59 +444,58 @@ async def process_single_key(key: str, rate_limit: bool, claude_model: str) -> D
              endpoint = f"https://{name}.openai.azure.com/"
              api_key = potential_key
              result = get_key_azure_info(endpoint, api_key)
-             return {"key": _key, **result} # key_type 포함
+             return {"key": _key, **result}
 
     # Azure endpoint: "https://xxx.openai.azure.com;key"
     if ";" in _key and "openai.azure.com" in _key.split(";")[0]:
         endpoint, api_key = _key.split(";", 1)
         result = get_key_azure_info(endpoint, api_key)
-        return {"key": _key, **result} # key_type 포함
+        return {"key": _key, **result}
 
     # AWS
     if _key.startswith("AKIA") and len(_key.split(":")[0]) == 20 and _key.split(":")[0].isalnum() and _key.split(":")[0].isupper() and len(_key.split(':')) == 2:
         result = await get_key_aws_info(_key)
-        return {"key": _key, **result} # key_type 포함
+        return {"key": _key, **result}
 
     # ElevenLabs
     if re.fullmatch(r"[a-f0-9]{32}", _key) or re.fullmatch(r"sk_[a-f0-9]{48}", _key):
         result = get_key_elevenlabs_info(_key)
-        return {"key": _key, **result} # key_type 포함
+        return {"key": _key, **result}
 
     # Mistral
     if re.fullmatch(r"[a-zA-Z0-9]{32}", _key) and not _key.startswith('sk-'):
         result = get_key_mistral_info(_key)
-        return {"key": _key, **result} # key_type 포함
+        return {"key": _key, **result}
 
     # Groq
     if re.match(re.compile(r"gsk_[a-zA-Z0-9]{20}WGdyb3FY[a-zA-Z0-9]{24}"), _key):
         result = get_key_groq_info(_key)
-        return {"key": _key, **result} # key_type 포함
+        return {"key": _key, **result}
 
     # GCP - refresh token
     if re.match(re.compile(r"[\w\-]+:[\w\-@\.]+:.+:.+"), _key):
         parts = _key.split(':')
         if len(parts) >= 4:
             result = await get_key_gcp_info(_key, 0)
-            return {"key": _key, **result} # key_type 포함
+            return {"key": _key, **result}
 
     # GCP - service account
     if re.match(re.compile(r"[\w\-]+:[\w\-@\.]+:.+\\n"), _key):
          parts = _key.split(':')
          if len(parts) >= 3:
             result = await get_key_gcp_info(_key, 1)
-            return {"key": _key, **result} # key_type 포함
+            return {"key": _key, **result}
 
     # Not supported
-    result = not_supported(_key) # key_type, key_availability, status 포함
-    # 키 값 추가하여 반환
+    result = not_supported(_key)
     return {"key": _key, **result}
 
 
 # ─────────────────────────────────────────
-#  여러 key 비동기 처리 함수 (업데이트됨)
+#  여러 key 비동기 처리 함수 (업데이트됨 - 모든 작동 키 반환)
 # ─────────────────────────────────────────
 async def sort_keys(text: str, rate_limit: bool, claude_model: str) -> Tuple[List[Dict[str, Any]], str]:
-    """텍스트 박스에 입력된 여러 키를 줄 단위로 분석하고, 전체 결과와 작동하는 Gemini 키 목록을 반환."""
+    """텍스트 박스에 입력된 여러 키를 줄 단위로 분석하고, 전체 결과와 작동하는 키 목록을 반환."""
     keys = [k.strip() for k in text.splitlines() if k.strip()]
     if not keys:
         return [], ""
@@ -499,23 +503,34 @@ async def sort_keys(text: str, rate_limit: bool, claude_model: str) -> Tuple[Lis
     tasks = [process_single_key(k, rate_limit, claude_model) for k in keys]
     results = await asyncio.gather(*tasks)
 
-    # 작동하는 Gemini 키만 필터링 (is_gemini_working 플래그 사용)
-    working_gemini_keys = []
+    # 작동하는 키들을 수집
+    working_keys = []
+    
     for result in results:
-        # is_gemini_working 키가 존재하고 True인지 확인
-        if result.get("is_gemini_working"): # get() 사용하여 키 부재 시 오류 방지
-            # 원본 get_key_gemini_info는 'key' 필드를 포함하므로 바로 사용
-            working_gemini_keys.append(result["key"])
-
-    # 결과를 JSON용 리스트와 Gemini 키 목록 문자열로 반환
-    return results, "\n".join(working_gemini_keys)
+        key_value = result.get("key", "")
+        key_type = result.get("key_type", "")
+        
+        # OpenAI 키 확인
+        if result.get("is_openai_working"):
+            working_keys.append(f"[OpenAI] {key_value}")
+        
+        # Anthropic 키 확인
+        elif result.get("is_anthropic_working"):
+            working_keys.append(f"[Anthropic] {key_value}")
+        
+        # Gemini 키 확인 (기존)
+        elif result.get("is_gemini_working"):
+            working_keys.append(f"[Gemini] {key_value}")
+
+    # 결과를 JSON용 리스트와 작동하는 키 목록 문자열로 반환
+    return results, "\n".join(working_keys)
 
 
 # ─────────────────────────────────────────
 #  UI util
 # ─────────────────────────────────────────
-def clear_inputs(): # 입력 인자 제거 (key_box 값은 필요 없음)
-    return "", "", "" # 모든 입력/출력 초기화
+def clear_inputs():
+    return "", "", ""
 
 
 # ─────────────────────────────────────────
@@ -524,7 +539,7 @@ def clear_inputs(): # 입력 인자 제거 (key_box 값은 필요 없음)
 with gr.Blocks() as demo:
     gr.Markdown(
         """
-# Multi-API Key Status Checker (Gemini Enhanced)
+# Multi-API Key Status Checker (Enhanced Working Keys Display)
 *(Based on shaocongma, CncAnon1, su, Drago, kingbased key checkers)*
 Check the status and details of various API keys including OpenAI, Anthropic, Gemini, Azure, Mistral, Replicate, AWS Claude, OpenRouter, Vertex AI (GCP Anthropic), Groq, NovelAI, ElevenLabs, xAI, Stability AI, and DeepSeek.
 
@@ -567,27 +582,27 @@ Check the status and details of various API keys including OpenAI, Anthropic, Ge
 
         with gr.Column(scale=2):
             info = gr.JSON(label=" API Key Information (All Results)", open=True)
-            # Gemini 전용 출력 박스 추가
-            gemini_keys_output = gr.Textbox(
-                label="Working Gemini Keys",
-                info="Lists Gemini keys confirmed as 'Working'.",
-                lines=3,
-                max_lines=10,
-                interactive=False, # 사용자가 편집 불가
+            # 작동하는 키들 출력 박스 (확장됨)
+            working_keys_output = gr.Textbox(
+                label="Working Keys (OpenAI, Anthropic, Gemini)",
+                info="Lists all keys confirmed as working for OpenAI, Anthropic Claude, and Gemini services.",
+                lines=5,
+                max_lines=15,
+                interactive=False,
             )
 
     # Clear 버튼 클릭 시 입력 박스와 출력 박스 모두 초기화
     clear_button.click(
-        fn=clear_inputs, # 수정된 함수 사용
+        fn=clear_inputs,
         inputs=None,
-        outputs=[key_box, info, gemini_keys_output]
+        outputs=[key_box, info, working_keys_output]
     )
 
     # Submit 버튼 클릭 시 sort_keys 함수 호출 및 결과 분배
     submit_button.click(
         fn=sort_keys,
         inputs=[key_box, rate_limit, claude_model],
-        outputs=[info, gemini_keys_output], # 두 개의 출력 컴포넌트에 결과 매핑
+        outputs=[info, working_keys_output],
         api_name="sort_keys",
     )