Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 from ultralytics import YOLO
 from fastapi import FastAPI
@@ -9,6 +10,10 @@ import cv2
 from pathlib import Path
 import tempfile
 
+# 从环境变量获取密码
+APP_USERNAME = "admin"  # 用户名保持固定
+APP_PASSWORD = os.getenv("APP_PASSWORD", "default_password")  # 从环境变量获取密码
+
 app = FastAPI()
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 model = YOLO('kunin-mice-pose.v0.1.0.pt')
@@ -22,7 +27,7 @@ auth_state = AuthState()
 
 def login(username, password):
     """登录验证"""
-    if username == "admin" and password == "your_password_here":
+    if username == APP_USERNAME and password == APP_PASSWORD:
         auth_state.is_logged_in = True
         return gr.update(visible=False), gr.update(visible=True), "登录成功"
     return gr.update(visible=True), gr.update(visible=False), "用户名或密码错误"