PirddlZddlZddlZddlZddlZddlZdZdZdZdZ e dkr e dSdS)Nc\ tj|dgdddS#t$rYdSwxYw)Nz --versionTF)capture_outputcheck) subprocessrunFileNotFoundError)cmds `/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vieneu/serve.py check_commandr sJ[)$eLLLLt uus  ++cF tjdjS#YdSxYw)Nzhttps://api.ipify.orgzyour-server-ip)requestsgettextr get_public_iprs- |34499 s ctdtd|jddd|jdddt|jd t|jd t|jd |jg}|jr)|d t|jgtd d |tj |}td|jdd}d}|j r>tdrtdddt|jddg}tj |tjtjd}t!j}t!j|z dkrw|j}d|vr@|dd}td|nt!j|z dkwnstdtdt+d|jn=td |jtd!t+d|jtd"|rd#|nd#t+d|j}td$td%|d&|jd'td( |dS#t.$r@td)||r|YdSYdSwxYw)*z) Starts the LMDeploy API server. u)🚀 Starting VieNeu-TTS Remote Server...u 📦 Model: lmdeployserve api_serverz --server-namez0.0.0.0z --server-port--tpz--cache-max-entry-count --model-name--quant-policyu🛠️ Command:  u-⏳ Waiting for server to initialize on port z...Nboreu"🌐 Starting tunnel via 'bore'...localz--tozbore.pubT)stdoutstderrr z listening atu✅ Public URL: http://u\⚠️ 'bore' not found. Please install it to use --tunnel (https://github.com/ekzhang/bore)u!📍 Using local address: http://:u.✅ Server running locally at: http://0.0.0.0:u(📍 Public access (if enabled): http://u% 💡 To use this server in your SDK:zhttp://z from vieneu import Vieneuz) tts = Vieneu(mode='remote', api_base='z/v1', model_name='z')u 🛑 Stopping server...)printmodelstrporttp memory_util model_name quant_policyextendjoinrPopentunnelr PIPESTDOUTtimerreadlinesplitstriprwaitKeyboardInterrupt terminate) argsr server_processtunnel_process public_url tunnel_cmd start_timelinesdk_urls r run_serverr@s 6777 % % %&&& G\ TYDG !3t'7#8#8 C ? $c$*;&<&<=>>> -chhsmm - -... %c**N H$) H H HIIINJ {X   U 6 7 7 7 '3ty>>6:NJ'-jYcYjquvvvNJ)++ *R//%,5577!T))!%N!;!;B!?!E!E!G!GJ@J@@AAA )++ *R// p q q q SmooSS SS T T T T JtyJJKKK VVV49VVWWW 2333(2_$ $$$8_-//8_8_TXT]8_8_G )*** dg d dQUQ` d d deee "III' ''' )***  """  '  $ $ & & & & & & ' ' ''s(K>>AMMctjd}|dtdd|dtdd|d td d |d td d|dt dd|dtdd|ddd|}|jdkr|jdkr |j|_tds2tdtdtj d t|dS)NzVieNeu-TTS Remote Server CLI) descriptionz--modelzpnnbao-ump/VieNeu-TTSz"HuggingFace model ID or local path)typedefaulthelprzModel name for API mappingz--porti%[z Server portrzTensor parallel sizez --memory-utilg333333?z GPU memory utilization (0.0-1.0)rrzKV cache quantization (0, 4, 8)z--tunnel store_truez,Automatically expose the server via bore.pub)actionrEru❌ 'lmdeploy' not found!z3 Please install it using: pip install vieneu[gpu])argparseArgumentParser add_argumentr%intfloat parse_argsr$r)r r#sysexitr@)parserr8s r mainrRXsp  $1O P P PF  5LSwxxx S:QXtuuu sE NNN S!:PQQQ eSGijjj (sADefff  <>lmmm     D z,,,D[1[1[*  $ $ )*** CDDD  tr__main__) rIosrrOr1r r rr@rR__name__rrr rVs     A'A'A'F0 zDFFFFFr