from app import create_app | |
import os | |
import sys | |
import argparse | |
app = create_app() | |
if __name__ == '__main__': | |
# Parse command line arguments | |
parser = argparse.ArgumentParser(description='Run DP-SGD Explorer') | |
parser.add_argument('--port', type=int, default=5000, help='Port to run the server on (default: 5000)') | |
# parser.add_argument('--host', type=str, default='127.0.0.1', help='Host to run the server on (default: 127.0.0.1)') | |
parser.add_argument("--host", type=str, default=os.getenv("HOST", "0.0.0.0"), | |
help="Host to run the server on (default: 0.0.0.0)") | |
args = parser.parse_args() | |
# Enable debug mode for development | |
app.config['DEBUG'] = True | |
# Disable CORS in development | |
app.config['CORS_HEADERS'] = 'Content-Type' | |
print(f"Starting server on http://{args.host}:{args.port}") | |
# Run the application | |
app.run(host=args.host, port=args.port, debug=True) |