Update app.py

app.py

@@ -48,11 +48,19 @@ CORS(app, resources={r"/*": {"origins": "*"}})
 executor = ThreadPoolExecutor(max_workers=10)
 
 proxy_url = os.getenv('PROXY_URL')
-NOTDIAMOND_IP = os.getenv('NOTDIAMOND_IP')
-NOTDIAMOND_DOMAIN = os.getenv('NOTDIAMOND_DOMAIN')
+NOTDIAMOND_DOMAIN = os.getenv('NOTDIAMOND_DOMAIN', 'not-diamond-workers.t7-cc4.workers.dev')
+
+# 尝试自动获取域名的 IP 地址
+try:
+    NOTDIAMOND_IP = socket.gethostbyname(NOTDIAMOND_DOMAIN)
+    logger.info(f"Automatically resolved {NOTDIAMOND_DOMAIN} to IP: {NOTDIAMOND_IP}")
+except socket.gaierror:
+    # 如果解析失败，回退到默认值
+    NOTDIAMOND_IP = os.getenv('NOTDIAMOND_IP', '172.67.136.20')
+    logger.warning(f"Failed to resolve {NOTDIAMOND_DOMAIN}, using default IP: {NOTDIAMOND_IP}")
 
 if not NOTDIAMOND_IP:
-    logger.error("NOTDIAMOND_IP environment variable is not set!")
+    logger.error("NOTDIAMOND_IP could not be determined!")
     raise ValueError("NOTDIAMOND_IP must be set")
 
 # API密钥验证装饰器