Update app.py

app.py

@@ -5,7 +5,7 @@ from api_usage import get_subscription, check_key_availability, check_key_ant_av
 async def sort_key(key, rate_limit, claude_model):
     _key = key.strip()
     
-    if _key.startswith("sk-or-v1-") and re.match(re.compile("sk-or-v1-[a-z0-9]{64}"), _key):    
+    if re.match(re.compile("sk-or-v1-[a-z0-9]{64}"), _key):    
         return get_key_openrouter_info(_key)
         
     if re.match(re.compile("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("sk-[a-zA-Z0-9]{86}"), _key)):
@@ -226,7 +226,7 @@ def get_key_openrouter_info(key):
         info_dict['limit'] = key_avai[1]['limit']
         info_dict['limit_remaining'] = key_avai[1]['limit_remaining']
         info_dict['usage'] = f"${format(key_avai[1]['usage'], '.4f')}"
-        info_dict['balance'] = f"${format(models_info[0], '.4f')}" if models_info[0] else "No info"
+        info_dict['balance'] = f"${format(models_info[0], '.4f')}" if models_info[0] else f"${key_avai[2]/60} (estimated)" 
         info_dict['rate_limit_per_minite'] = key_avai[2]
         info_dict['4_turbo_per_request_tokens_limit'] = models_info[1]['openai/gpt-4o']
         info_dict['sonnet_per_request_tokens_limit'] = models_info[1]['anthropic/claude-3.5-sonnet:beta']
@@ -301,6 +301,7 @@ with gr.Blocks() as demo:
     'claude-3-opus-20240229',
     'claude-3-5-sonnet-20240620',
     'claude-3-5-sonnet-20241022',
+    'claude-3-5-haiku-20241022'
     ]
     with gr.Row():
         with gr.Column():