Function main

web_demo/server/server.py:1120–1158

()

        return ws

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--host", default="localhost", type=str)
    parser.add_argument("--port", default=11235, type=int)
    parser.add_argument("--sample-rate", default=24000, type=int, help="Audio sample rate (Opus)")
    parser.add_argument("--model-sample-rate", default=16000, type=int, help="Model sample rate")
    parser.add_argument("--output-dir", default="./output", type=str, help="Directory to save input audio files")
    parser.add_argument("--model-path", type=str, default="model/s2s", help="Path to S2S model")
    parser.add_argument("--tts-gpu", default=1, type=int, help="GPU device id for TTS model (default: 1)")

    args = parser.parse_args()

    log("info", f"Initializing server with Opus sample rate: {args.sample_rate}, Model sample rate: {args.model_sample_rate}")
    
    model_manager = GlobalModelManager()
    try:
        model_manager.initialize(
            model_path=args.model_path,
            target_sample_rate=args.model_sample_rate
        )
    except Exception as e:
        log("error", f"Model initialization failed: {e}")
        import traceback
        traceback.print_exc()
        return
    
    state = ServerState(
        model_manager=model_manager,
        sample_rate=args.sample_rate, 
        output_dir=args.output_dir,
        tts_gpu=args.tts_gpu
    )
    
    app = web.Application()
    app.router.add_get("/api/chat", state.handle_chat)
    
    protocol = "http"
    log("info", f"Access the Web UI directly at {protocol}://{args.host}:{args.port}")
    web.run_app(app, host=args.host, port=args.port)

if __name__ == "__main__":
    main()