diff --git a/backend/config.yaml b/backend/config.yaml index 3892524..9ce8471 100644 --- a/backend/config.yaml +++ b/backend/config.yaml @@ -1,3 +1,7 @@ +backend: + ip: 192.168.193.141 + port: 5000 + model: save_dir: models diff --git a/backend/flaskApp.py b/backend/flaskApp.py index f407d7b..a7ce06c 100644 --- a/backend/flaskApp.py +++ b/backend/flaskApp.py @@ -29,6 +29,8 @@ OUTPUT_DIR = cfg.data.run.output_dir MODELS_DIR = str((CONFIG_PATH.parent / cfg.model.save_dir).resolve()) TRAIN_DIR = cfg.data.train.train_dir TEST_DIR = cfg.data.train.test_dir +BACKEND_IP = cfg.backend.ip +BACKEND_PORT = cfg.backend.port os.makedirs(UPLOAD_DIR, exist_ok=True) executor = ThreadPoolExecutor(max_workers=4) @@ -37,7 +39,7 @@ r = redis.Redis(host="127.0.0.1", port=6379, db=0) # 启动测试服务器 def run_dev(): - app.run(host="192.168.193.141", port=5000) + app.run(host=BACKEND_IP, port=int(BACKEND_PORT)) def set_status(task_id, status, **extra): payload = {"status": status, "updated_at": datetime.datetime.utcnow().isoformat(), **extra} diff --git a/frontend/api.js b/frontend/api.js index 53013a3..b84607a 100644 --- a/frontend/api.js +++ b/frontend/api.js @@ -1 +1,9 @@ -const API_BASE = "http://192.168.193.141:5000/"; \ No newline at end of file +const config = { + server: { + protocol: 'http', + host: '192.168.193.141', + port: 5000 + } +}; + +const API_BASE = `${config.server.protocol}://${config.server.host}:${config.server.port}/`; \ No newline at end of file