From 2a81b1e115c61856d93c00a37df333254c094e7e Mon Sep 17 00:00:00 2001 From: yoshi Date: Thu, 1 May 2025 12:17:49 -0700 Subject: [PATCH] modified snakaAPI --- deployment-service/fly.toml | 2 +- deployment-service/snakeapi_service/snakeapi_server.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/deployment-service/fly.toml b/deployment-service/fly.toml index 09af4c1..fc1d758 100644 --- a/deployment-service/fly.toml +++ b/deployment-service/fly.toml @@ -11,7 +11,7 @@ dockerfile = "Dockerfile" AWS_SECRET_ACCESS_KEY="tsec_6Bz1aMbfYQftuq5WfIVEDZkHwskU4MMjVywdtxSP6uxetEBvkSC2VHI9HfTeDgHr4D6kiz" AWS_ENDPOINT_URL_S3="https://fly.storage.tigris.dev" AWS_REGION="auto" - IMAGE_REF = "registry.fly.io/snakeapi-template:latest" + IMAGE_REF="registry.fly.io/snake-api-template:latest" FLY_API_BASE_URL = "https://api.machines.dev/v1" [http_service] diff --git a/deployment-service/snakeapi_service/snakeapi_server.py b/deployment-service/snakeapi_service/snakeapi_server.py index 79d0578..2fa0cdc 100644 --- a/deployment-service/snakeapi_service/snakeapi_server.py +++ b/deployment-service/snakeapi_service/snakeapi_server.py @@ -5,13 +5,14 @@ import importlib.util from flask import Flask, request, jsonify from flask_cors import CORS -# load handlers from converted notebook notebook_path = os.environ.get("NOTEBOOK_PATH") if not notebook_path or not os.path.isfile(notebook_path): raise RuntimeError(f"Notebook module not found: {notebook_path}") + spec = importlib.util.spec_from_file_location("notebook_module", notebook_path) notebook_module = importlib.util.module_from_spec(spec) spec.loader.exec_module(notebook_module) + handlers = { "info": notebook_module.info, "start": notebook_module.start, @@ -53,9 +54,10 @@ def run_server(handlers: typing.Dict): def get_notebook_path(): return jsonify({"path": notebook_path}) - host = "::" + host = "0.0.0.0" port = int(os.environ.get("PORT", "3006")) logging.getLogger("werkzeug").setLevel(logging.ERROR) + app.run(host=host, port=port) if __name__ == "__main__":