run_api.py

# Run script for the Stock Prediction API
import argparse
from src.api import start_api

def main():
    """
    Main function to run the Stock Prediction API
    """
    parser = argparse.ArgumentParser(description='Stock Price Prediction API')
    parser.add_argument('--host', type=str, default='0.0.0.0', help='Host to run the API on')
    parser.add_argument('--port', type=int, default=5000, help='Port to run the API on')
    parser.add_argument('--debug', action='store_true', help='Run in debug mode')
    
    args = parser.parse_args()
    
    print(f"Starting Stock Prediction API on {args.host}:{args.port}")
    print("Press Ctrl+C to stop the server")
    
    start_api(host=args.host, port=args.port, debug=args.debug)

if __name__ == "__main__":
    main()