Create app.py

app.py

@@ -0,0 +1,86 @@
+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)}"
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+def validate_offi_api_key(api_key):
+    if api_key == '':
+        raise gr.Error('请输入您的中转API Key')
+    try:
+        client = OpenAI(api_key=api_key, base_url='https://api.openai.com')
+        response = client.models.list()
+        models_list = [model.id for model in response.data]
+        return f"API key is valid.\nModels: {', '.join(models_list)}"
+    except Exception as e:
+        return f"Error: {str(e)}"
+css = """
+/* 更新CSS 样式 */
+.fixed-size-button button {
+    width: 150px !important
+    height: 50px !important
+    font-size: 16px !important
+    color: white !important
+    border: none !important
+    border-radius: 5px !important
+    cursor: pointer !important
+}
+#warning {background-color: #4CAF50}
+.fixed-size-button button:hover {
+    background-color: #45a049 ; /* 浅绿色背景 */
+}
+
+/* 可滚动的输出框 */
+.output-box {
+    width: 100% !important
+    height: 200px !important
+    overflow-y: auto !important
+    border: 1px solid #ccc !important
+    padding: 10px !important
+    box-sizing: border-box !important
+}
+"""
+# Gradio 前端设计
+app = gr.Blocks()
+
+with app:
+    gr.Markdown("# <center>api-key验证</center>")
+    gr.Markdown(
+        "### <center>中转key购买地址[here](https://buy.sipola.cn),ai文案生成可使用中转key,请访问 [here](https://ai.sipola.cn)</center>")
+    with gr.Row(variant='panel'):
+        with gr.Column():
+            transmit_api_key = gr.Textbox(type='password', label='中转API-Key', placeholder='请在此填写您的中转API Key')
+            transmit_btn_text = gr.Button("点击验证中转key", elem_id="warning", elem_classes="fixed-size-button")
+
+        with gr.Column():
+            transmit_output_textbox = gr.Textbox(label="显示key状态及可访问的模型列表", elem_classes="output-box", lines=5,max_lines=5)
+        transmit_btn_text.click(validate_tran_api_key,transmit_api_key,transmit_output_textbox)
+
+    with gr.Row(variant='panel'):
+        with gr.Column():
+            official_api_key = gr.Textbox(type='password', label='官方API Key', placeholder='请在此输入你的官方API Key')
+            official_btn_text = gr.Button("点击验证官方key", elem_id="warning", elem_classes="fixed-size-button")
+
+        with gr.Column():
+            official_output_textbox = gr.Textbox(label="显示key状态及可访问的模型列表", elem_classes="output-box", lines=5, max_lines=5 )
+        official_btn_text.click(validate_offi_api_key, official_api_key, official_btn_text)
+
+
+#添加页面底部
+    gr.HTML('''
+        <div class="footer">
+         <center><p>Power by sipola </p></center>
+        </div>
+    ''')
+
+app.launch(show_error=True)