Update app.py

app.py

@@ -36,8 +36,6 @@ from api_usage import (
 # ─────────────────────────────────────────
 def get_key_oai_info(key: str) -> Dict[str, Any]:
     session = requests.Session()
-    # raw_status_code는 HTTP 응답 코드 또는 유사한 상태 표시자
-    # org_data_or_error는 성공 시 데이터, 실패 시 에러 정보 등
     raw_status_code, org_data_or_error = check_key_availability(session, key)
 
     info_dict = {
@@ -85,8 +83,6 @@ def get_key_oai_info(key: str) -> Dict[str, Any]:
             info_dict["quota"] = current_quota_message
             if "insufficient_quota" not in current_quota_message.lower(): # 상세 메시지에 insufficient_quota가 없다면 상태 메시지에 추가
                  info_dict["status_message"] += f" Error: {current_quota_message}"
-        # 여기서 반환하면 아래 구독 정보 확인을 건너뛰지만, has_sufficient_quota가 이미 False임
-        # return info_dict # 필요에 따라 주석 해제하여 더 빠른 반환 가능
     else: # 기타 에러
         info_dict["status_message"] = f"Key check failed (status: {raw_status_code})."
         if isinstance(org_data_or_error, dict) and "error" in org_data_or_error:
@@ -97,10 +93,11 @@ def get_key_oai_info(key: str) -> Dict[str, Any]:
         return info_dict
 
     if not info_dict["key_availability"]:
-        return info_dict # 키가 유효하지 않으면 더 이상 진행 안 함
+        # 429가 아닌 다른 이유로 key_availability가 False면 반환
+        if raw_status_code != 429:
+             return info_dict
 
-    # org_data_for_subscription이 준비되지 않은 경우 (예: 초기 200이었지만 org_data가 부적절)
-    if not org_data_for_subscription:
+    if not org_data_for_subscription and info_dict["key_availability"]:
         status_me, orgs_me_data = get_orgs_me(session, key)
         if status_me == 200:
             org_data_for_subscription = orgs_me_data
@@ -110,6 +107,12 @@ def get_key_oai_info(key: str) -> Dict[str, Any]:
             info_dict["has_sufficient_quota"] = False
             return info_dict
 
+    if not org_data_for_subscription and info_dict["key_availability"]:
+        info_dict["status_message"] = (info_dict["status_message"] + " Organization data for subscription is missing.").strip()
+        info_dict["key_availability"] = False
+        info_dict["has_sufficient_quota"] = False
+        return info_dict
+
     subscription_info = get_subscription(key, session, org_data_for_subscription)
     if subscription_info:
         info_dict.update(
@@ -122,7 +125,7 @@ def get_key_oai_info(key: str) -> Dict[str, Any]:
                 "models": subscription_info.get("models", ""),
                 "requests_per_minute": subscription_info.get("rpm", ""),
                 "tokens_per_minute": subscription_info.get("tpm", ""),
-                "quota": subscription_info.get("quota", info_dict["quota"]),
+                "quota": subscription_info.get("quota") if subscription_info.get("quota") is not None else info_dict.get("quota", ""),
                 "all_models": subscription_info.get("all_models", ""),
             }
         )
@@ -140,20 +143,32 @@ def get_key_oai_info(key: str) -> Dict[str, Any]:
                ("billing" in err_msg and "issue" in err_msg):
                 info_dict["has_sufficient_quota"] = False
                 new_quota_message = f"Insufficient: {error_info.get('message', err_type)}"
-                info_dict["quota"] = new_quota_message # 쿼터 필드에 명시적 메시지
-                info_dict["status_message"] = (info_dict["status_message"] + f" Quota/Billing issue: {error_info.get('message', err_type)}").strip()
+                info_dict["quota"] = new_quota_message
+                current_status_msg = info_dict["status_message"]
+                additional_msg = f" Quota/Billing issue from error object: {error_info.get('message', err_type)}"
+                if additional_msg.strip() not in current_status_msg :
+                    info_dict["status_message"] = (current_status_msg + additional_msg).strip()
 
-        # 계정 비활성화 등 확인
         if "account_deactivated" in str(subscription_info).lower() or \
            "payment_failed" in str(subscription_info).lower():
             info_dict["has_sufficient_quota"] = False
-            if "Account issue" not in info_dict["status_message"]: # 중복 메시지 방지
-                 info_dict["status_message"] = (info_dict["status_message"] + " Account issue (e.g., deactivated, payment failed).").strip()
+            current_status_msg = info_dict["status_message"]
+            additional_msg = " Account issue (e.g., deactivated, payment failed) from subscription info."
+            if additional_msg.strip() not in current_status_msg:
+                 info_dict["status_message"] = (current_status_msg + additional_msg).strip()
     else:
-        info_dict["status_message"] = (info_dict["status_message"] + " Failed to retrieve full subscription details.").strip()
-        info_dict["has_sufficient_quota"] = False
-        # 구독 정보 없으면 key_availability도 False로 간주할 수 있음.
-        # info_dict["key_availability"] = False # 주석 처리하여 키 자체는 유효할 수 있음을 남김
+        if info_dict["key_availability"]:
+            info_dict["status_message"] = (info_dict["status_message"] + " Failed to retrieve full subscription details.").strip()
+            info_dict["has_sufficient_quota"] = False
+
+    if info_dict.get("has_sufficient_quota") is True:
+        current_quota_field_value = info_dict.get("quota")
+        if current_quota_field_value:
+            if "insufficient_quota" in str(current_quota_field_value).lower():
+                info_dict["has_sufficient_quota"] = False
+                insufficient_msg = "Insufficient quota indicated in 'quota' field."
+                if insufficient_msg not in info_dict["status_message"]:
+                    info_dict["status_message"] = (info_dict["status_message"] + f" {insufficient_msg}").strip()
 
     return info_dict
 
@@ -448,40 +463,28 @@ def not_supported(key: str) -> Dict[str, Any]:
 #  단일 키 비동기 처리 (업데이트됨)
 # ─────────────────────────────────────────
 async def process_single_key(key: str, rate_limit: bool, claude_model: str) -> Dict[str, Any]:
-    """주어진 key 하나를 분석해 정보 dict 반환. Gemini 키의 작동 여부 플래그 포함."""
     _key = key.strip()
 
     if not _key:
         return {"key": "", "key_type": "Empty", "key_availability": False}
 
-    # 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}
-
-    # 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}
-
-    # 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}
-
-    # Deepseek
     if re.match(re.compile(r"sk-[a-f0-9]{32}"), _key):
          result = get_key_deepseek_info(_key)
          return {"key": _key, **result}
-
-    # OpenAI (다른 sk- 패턴보다 뒤에 와야 함)
     if _key.startswith("sk-"):
-        result = get_key_oai_info(_key) # 수정된 함수 호출
+        result = get_key_oai_info(_key)
         return {"key": _key, **result}
-
-    # Google Gemini 처리
     if _key.startswith("AIzaSy"):
         gemini_info = get_key_gemini_info(_key)
         is_working = gemini_info.get("key_availability") and gemini_info.get("status") == "Working"
@@ -491,23 +494,15 @@ async def process_single_key(key: str, rate_limit: bool, claude_model: str) -> D
             "is_gemini_working": is_working
         }
         return result
-
-    # NovelAI
     if _key.startswith("pst-"):
         result = get_key_nai_info(_key)
         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}
-
-    # xAI
     if _key.startswith("xai-"):
         result = get_key_xai_info(_key)
         return {"key": _key, **result}
-
-    # Azure endpoint: "name:key"
     if len(_key.split(":")) == 2:
         name, potential_key = _key.split(":", 1)
         if re.fullmatch(r'[a-fA-F0-9]{32}', potential_key) and "openai.azure.com" not in name:
@@ -515,48 +510,32 @@ async def process_single_key(key: str, rate_limit: bool, claude_model: str) -> D
              api_key = potential_key
              result = get_key_azure_info(endpoint, api_key)
              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}
-
-    # 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}
-
-    # 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}
-
-    # 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}
-
-    # 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}
-
-    # 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}
-
-    # 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}
-
-    # Not supported
     result = not_supported(_key)
     return {"key": _key, **result}
 
@@ -565,10 +544,6 @@ async def process_single_key(key: str, rate_limit: bool, claude_model: str) -> D
 #  여러 key 비동기 처리 함수 (업데이트됨)
 # ─────────────────────────────────────────
 async def sort_keys(text: str, rate_limit: bool, claude_model: str) -> Tuple[List[Dict[str, Any]], str, str, str]:
-    """
-    텍스트 박스에 입력된 여러 키를 줄 단위로 분석하고,
-    전체 결과와 작동하는 OAI, Anthropic, Gemini 키 목록을 각각 반환합니다.
-    """
     keys = [k.strip() for k in text.splitlines() if k.strip()]
     if not keys:
         return [], "", "", ""
@@ -587,15 +562,12 @@ async def sort_keys(text: str, rate_limit: bool, claude_model: str) -> Tuple[Lis
 
         key_type = result.get("key_type")
 
-        # Gemini 키 필터링
         if result.get("is_gemini_working"):
             working_gemini_keys.append(key_value)
-        # OpenAI 키 필터링 (수정됨: has_sufficient_quota 확인)
         elif key_type == "OpenAI" and \
              result.get("key_availability") is True and \
-             result.get("has_sufficient_quota") is True: # 이 필드가 True여야 함
+             result.get("has_sufficient_quota") is True:
             working_oai_keys.append(key_value)
-        # Anthropic 키 필터링
         elif key_type == "Anthropic Claude" and result.get("key_availability") is True:
             working_anthropic_keys.append(key_value)