diff --git a/main.py b/main.py
index 88c345145759c4549fcd7112476729c9700fa065..ba98df6a2b71a981c9917d66eab543b8a9401213 100644
--- a/main.py
+++ b/main.py
@@ -9,9 +9,7 @@ app = Flask(__name__)
 # 设置 DeepSeek API 的相关信息
 DEEPSEEK_API_URL = "https://api.deepseek.com/v1/chat/completions"
 DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")  # 从环境变量获取 API 密钥
-
-if not DEEPSEEK_API_KEY:
-    raise ValueError("请设置 DEEPSEEK_API_KEY 环境变量")
+DEEPSEEK_API_KEY = input("请设置DEEPSEEK_API_KEY:")
 
 # 配置日志记录
 logging.basicConfig(level=logging.DEBUG)
@@ -49,6 +47,10 @@ def judge():
     }
 
     logging.debug(f"payload: {json.dumps(payload, indent=2)}")
+    #if not DEEPSEEK_API_KEY:
+    #exit()
+    if not DEEPSEEK_API_KEY:
+        raise ValueError("请设置 DEEPSEEK_API_KEY 环境变量")
 
     try:
         # 发送请求到 DeepSeek API