Update app.py

app.py

@@ -1,29 +1,56 @@
 from openai import OpenAI
 import gradio as gr
 
-# 验证API Key的函数
 def validate_tran_api_key(api_key):
     if api_key == '':
         raise gr.Error('请输入您的中转API Key')
     try:
         client = OpenAI(api_key=api_key, base_url='https://lmzh.top/v1')
-        # 调用OpenAI API来验证Key
         response = client.models.list()
         models_list = [model.id for model in response.data]
-        return f"API key is valid.\nModels: {', '.join(models_list)}"
+
+        # 进行简单的聊天测试
+        try:
+            test_response = client.chat.completions.create(
+                model="gpt-4o-mini",
+                messages=[
+                    {"role": "user", "content": "This is a test"}
+                ],
+                max_tokens=5
+            )
+            quota_status = "Quota is available."
+        except Exception as e:
+            quota_status = "Quota might be exhausted or an error occurred."
+
+        return f"API key is valid.\n{quota_status}\nModels: {', '.join(models_list)}"
     except Exception as e:
         return f"Error: {str(e)}"
 
 def validate_offi_api_key(api_key):
     if api_key == '':
-        raise gr.Error('请输入您的中转API Key')
+        raise gr.Error('请输入您的官方API Key')
     try:
         client = OpenAI(api_key=api_key, base_url='https://api.openai.com/v1')
         response = client.models.list()
         models_list = [model.id for model in response.data]
-        return f"API key is valid.\nModels: {', '.join(models_list)}"
+
+        # 进行简单的聊天测试
+        try:
+            test_response = client.chat.create(
+                model="davinci",  # 使用一个合适的模型名称进行测试，如'gpt-4o-mini'等
+                messages=[
+                    {"role": "user", "content": "This is a test"}
+                ],
+                max_tokens=5
+            )
+            quota_status = "Quota is available."
+        except Exception as e:
+            quota_status = "Quota might be exhausted or an error occurred."
+
+        return f"API key is valid.\n{quota_status}\nModels: {', '.join(models_list)}"
     except Exception as e:
         return f"Error: {str(e)}"
+
 css = """
 /* 更新CSS 样式 */
 .fixed-size-button button {