Spaces:
Sleeping
Sleeping
| import os | |
| from flask import Flask, send_from_directory | |
| from flask_cors import CORS | |
| from database.db import init_db | |
| from routes.emails import emails_bp | |
| from routes.tonnage import tonnage_bp | |
| from routes.cargo_vc import cargo_vc_bp | |
| from routes.cargo_tc import cargo_tc_bp | |
| from routes.stats import stats_bp | |
| from routes.matching import matching_bp | |
| from routes.map_data import map_data_bp | |
| app = Flask(__name__, static_folder='static', static_url_path='') | |
| app.config['MAX_CONTENT_LENGTH'] = 5 * 1024 * 1024 # 5 MB file size limit | |
| # Configure CORS dynamically based on allowed origins environment variable | |
| allowed_origins = os.environ.get('ALLOWED_ORIGINS', '*') | |
| if allowed_origins == '*': | |
| CORS(app, resources={r"/api/*": {"origins": "*"}}) | |
| else: | |
| CORS(app, resources={r"/api/*": {"origins": allowed_origins.split(',')}}) | |
| # ── API blueprints ───────────────────────────────────────────────────────────── | |
| app.register_blueprint(emails_bp, url_prefix='/api') | |
| app.register_blueprint(tonnage_bp, url_prefix='/api') | |
| app.register_blueprint(cargo_vc_bp, url_prefix='/api') | |
| app.register_blueprint(cargo_tc_bp, url_prefix='/api') | |
| app.register_blueprint(stats_bp, url_prefix='/api') | |
| app.register_blueprint(matching_bp, url_prefix='/api') | |
| app.register_blueprint(map_data_bp, url_prefix='/api') | |
| # ── Frontend SPA ────────────────────────────────────────────────────────────── | |
| def index(): | |
| return send_from_directory(app.static_folder, 'index.html') | |
| def catch_all(path): | |
| # Serve static files; fall back to index.html for SPA routing | |
| full_path = os.path.join(app.static_folder, path) | |
| if os.path.isfile(full_path): | |
| return send_from_directory(app.static_folder, path) | |
| return send_from_directory(app.static_folder, 'index.html') | |
| # ── Startup ─────────────────────────────────────────────────────────────────── | |
| def startup(): | |
| init_db() | |
| # Import classifier to trigger training at startup (fast, ~1 s) | |
| from classification.classifier import classify # noqa: F401 | |
| print("[ShipSeg] Database initialised. Classifier ready.") | |
| startup() | |
| if __name__ == '__main__': | |
| port = int(os.environ.get('PORT', 5000)) | |
| app.run(host='0.0.0.0', port=port, debug=False) | |