jB UdZddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z ddlmZddlmZidd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1Zd2ed3<d4ZGd5d6eZdaej0Zd7ZdZd8Zd[d9Zed:; d\d<Z d] d^d=Zd>Zd_d?Z d@ddAdBdCdD d`dEZ!ddAdF dadGZ"ddHdbdIZ#ddHdcdJZ$dddKZ%ddHdedLZ&ddCdM dfdNZ'dgdOZ(dgdPZ)dddQ dhdRZ*didjdSZ+dTdddCdUdV dkdWZ,dldXZ-dmdYZ.y)nu GOLS - OpenAI Language & Voice Intelligence =========================================== Single integration point for all OpenAI usage across the platform. Capabilities ------------ • Script generation (role-play, interview, coaching, assessment, learning, custom) • Script rewrite / improve / summarize / translate / personalize • Language detection • Avatar-optimized rewriting (clean spoken text for TTS + facial delivery) • Gendered, multilingual Text-to-Speech (OpenAI TTS) Design notes ------------ • Lazy, cached singleton client — created once, reused. • Graceful degradation: if no OPENAI_API_KEY is configured, text helpers fall back to deterministic local transforms and TTS callers fall back to gTTS in the pipeline. Nothing crashes when the key is missing. • Response caching: identical (model, prompt) text requests are cached in-memory (LRU) to cut token spend on repeated calls (e.g. polling / re-renders). • All errors are wrapped in OpenAIServiceError with a clean message. ) annotationsN) lru_cache)OptionalListDictAny)cfg)logenEnglishhiHindimrMarathitaTamilteTeluguknKannadaml MalayalamguGujaratibnBengalipaPunjabiurUrduesSpanishfrFrenchdeGermanjaJapanesezhChinesearArabic PortugueseRussianItalianKorean IndonesianDutchTurkish)ptruitkoidnltrzDict[str, str]SUPPORTED_LANGUAGES generalroleplay interviewcoaching assessmentlearningsales onboarding announcementceZdZdZy)OpenAIServiceErrorzGRaised for any recoverable OpenAI failure (clean, user-facing message).N)__name__ __module__ __qualname____doc__s/Users/priyanka/Documents/AI_Agent_Assessment/ai-video-generator-gols-v2-new/backend/app/services/openai_service.pyrGrG7sQrMrGcttStjsyt5t ddlm}tjtj d}tjrtj|d<|di|atjdtjtjdddtS#t$r"}tjd|daYd}~4d}~wwxYw#1swYtSxYw) zEReturn a cached OpenAI client, or None if no key / SDK not installed.Nr)OpenAI)api_keytimeoutbase_urlz.[openai] client initialised (model=%s, tts=%s)z"[openai] failed to init client: %srL)_clientr OPENAI_API_KEY _client_lockopenairPOPENAI_TIMEOUTOPENAI_BASE_URLr infoOPENAI_TEXT_MODELOPENAI_TTS_MODEL Exceptionerror)rPkwargses rN _get_clientra@s     ? )"11"11*&&),)<)B   Ns/C-BB?? C*C% C-%C**C--C;ctduS)z*True when OpenAI is configured and usable.N)rarLrMrN is_enabledrcZs = $$rMcRtj|xsdjdS)Nr r )r;getlower)codes rN lang_namerh_s"  " "DLD#7#7#99 EErM)maxsizecht}| td |jjj |d|dd|dg||}|j dj jxsdjS#t$r)}tjd |td |d}~wwxYw) zBLRU-cached chat call. Args must be hashable (all strings/numbers).N.OpenAI is not configured (set OPENAI_API_KEY).systemrolecontentusermodelmessages temperature max_tokensrz#[openai] chat completion failed: %sOpenAI request failed: ) rarGchat completionscreatechoicesmessagerpstripr]r r^)rsrmrqrurvclientrespr`s rN _cached_chatrds]F ~ !QRR @{{&&--!f5D1$!.  Q''//52<<>> @ 7; #:1#!>??@sA%A?? B1$B,,B1cttj|j|j|tjn||tj S|SN)rr r[r~OPENAI_TEMPERATUREOPENAI_MAX_TOKENS)rmrqrurvs rN_chatr{sZ    "-"5;!+!3   :D  rMuYou are a professional scriptwriter for AI talking-avatar videos used on a platform similar to HeyGen and Tavus. You write natural, human-sounding spoken dialogue meant to be read aloud by a realistic avatar. Rules: write ONLY the words the avatar will speak — no stage directions, no headings, no markdown, no bullet symbols, no speaker labels unless a dialogue is explicitly requested. Use short, breathable sentences and natural punctuation that helps text-to-speech pacing.c Dddddddddd d }|j||d S) Nz'Write a clear, engaging spoken message.zRWrite a realistic role-play scene the avatar performs, staying fully in character.zIWrite interview questions and natural framing the avatar asks one by one.zOWrite a warm, motivating coaching session with practical, encouraging guidance.z>Write assessment questions with brief spoken framing for each.zQWrite a concise, easy-to-follow micro-lesson that teaches the topic step by step.zCWrite a persuasive but trustworthy sales pitch focused on benefits.zFWrite a friendly onboarding/welcome message that orients a new person.z&Write a crisp, confident announcement.r<r=)re) content_typebriefss rN_content_briefrs<?jagVi[^@ F ::lF9$5 66rMr= professionalmediumnatural)rlanguagetonelengthemotionc dddd}|j||d}tdt|dt|d|d |d |d }t ||S) z:Generate an avatar-ready spoken script from a topic/brief.zabout 40-70 wordszabout 90-150 wordszabout 200-300 words)shortrlongr z Write the entire output in z. Tone: z. Emotional delivery: z. Target length: .)re _SYSTEM_BASErrhr) promptrrrrr length_mapwordsrms rNgenerate_scriptrs~/:N/1J NN6:h#7 8E .>,789&&/&9%:;,WI5FugQ P   rM)rrcJtdt|d|d}t||S)z .%%.x%8$9G0 1   rM)rcPtdt|d}t|d|d|S)z3Rewrite a script following a free-form instruction.zY Rewrite the script according to the user's instruction. Output the spoken text only, in rz INSTRUCTION: z SCRIPT: r)r instructionrrms rNrewrite_scriptrsB .++4X+>*?q B  = ]6(K LLrMcHtdt|d}t||dS)Nzk Summarize the following into a short spoken version that keeps the key points. Output only spoken text in r333333?rur)rrrms rNsummarize_scriptrs7 .AAJ8AT@UUV X  S 11rMcHtdt|d}t||dS)z@Translate to a target language, preserving spoken/natural style.z Translate the script into zw. Keep it natural and idiomatic for a spoken avatar, not a literal translation. Return only the translated spoken text.rrr)rtarget_languagerms rNtranslate_scriptrs7 .4Y5O4PQ2 3 S 11rMcJtd|dt|d}t||S)Nz+ Personalize the script for this audience: zO. Adjust examples, vocabulary and tone accordingly. Output only spoken text in rr)raudiencerrms rNpersonalize_scriptrs: .DXJO&&/&9%:! =   rM)rrcNtd|dt|d}t||dS)z Clean a raw script into avatar/TTS-optimized spoken text: expand abbreviations, add natural pauses (commas), remove markup/emoji. Used automatically by the pipeline before TTS when OpenAI is enabled. z[ Rewrite the text so it is optimal for text-to-speech and realistic avatar delivery with a z emotional tone: expand abbreviations and numbers into spoken words, add natural commas for breathing, remove any emojis, markdown, URLs or symbols. Keep the meaning and language (z'). Return only the cleaned spoken text.g?rr)rrrrms rNoptimize_for_speechrsD .,,395#8,--T  V  S 11rMcts t|Sd} t||ddddjj }|dd}|t vr|SdS#t $rt|cYSwxYw) zy)z?Offline fallback: detect a few Indian scripts by Unicode range.)i i )i i )i i )i i )i i )i i )i i )i i )ii)iNi)i@0i0)ii) r rrrrrrrrr+r)r'r7r )orditems)rrangeschorglor s rNrr stF G$llnND(2rQ}"} - rM)rrmct}| td|xstdt|d} |jj j tjd|dg|tjtj}|jdjjxsd jS#t$r)}t!j"d |td |d}~wwxYw) zv Maintain conversational context. `messages` is a list of {"role": "user"|"assistant", "content": "..."}. NrlzR You are having a natural conversation that will be spoken by an avatar. Reply in z with spoken text only.rmrnrrrrwz[openai] converse failed: %srx)rarGrrhryrzr{r r[rrr|r}rpr~r]r r^)rtrrmrsysrr`s rNconverser&s ]F ~ !QRR   .""+H"5!66M O @{{&&--'''C8D8D..,, .   Q''//52<<>> @ 0!4 #:1#!>??@sB C C2 $C--C2cx|r|S|xsdjdk(rtjStjS)z9Map a gender ('male'/'female') to an OpenAI TTS voice id.rwmale)rfr OPENAI_VOICE_MALEOPENAI_VOICE_FEMALE)genderoverrides rNvoice_for_genderrCs:%+\r$8$8$:f$D3 %  $ $%rMfemale?)rvoicerrspeedc t}|yt||}dt|d|d} tdt dt |} g} tjd fD]} | s| | vs | j| d} | D]} t|||d | }d |vr| |d <|jjjjdi|5}|j|dddt j"j%|dkDrt'j(d||||yt'j,d| y#t $rd} YwxYw#1swYqxYw#t $r$}|} t'j*d||Yd}~d}~wwxYw)z Generate speech audio (mp3) with OpenAI TTS. Returns True on success, False if OpenAI is unavailable (so the pipeline can fall back to gTTS). Raises OpenAIServiceError only on a hard API error. NFz Speak in z with a z, natural human tone suitable for a realistic talking-avatar video. Clear articulation, appropriate emotional inflection and pacing.g?g@rztts-1mp3)rsrinputresponse_formatrzgpt-4o instructionsiz3[openai] TTS ok (model=%s, voice=%s, lang=%s) -> %sTu3[openai] TTS model '%s' failed (%s) — trying nextuL[openai] All OpenAI TTS attempts failed (%s) — pipeline falls back to gTTSrL)rarrhmaxminfloatr]r r\appenddictaudiospeechwith_streaming_responser{stream_to_fileospathgetsizer rZwarningr^)rout_pathrrrrrrchosenrspd models_to_trymlast_errrsr_rr`s rNsynthesize_speechrKs]F ~ fe ,F Ih'( :7 8 #s3e -. M""G , -'   #-H Y%)6 %S&F5 )5~&C$$<<CCMfMQU##H-Nwwx(4/N(<0$II\ ? &NM  YH KKMuVW X X YsCD?A EE!AE? E EE E F &FF c,tjyr)r cache_clearrLrMrN clear_cachersrMctj}|j|j|j|j dS)N)hitsmissessizerj)r cache_inforrcurrsizerj)rZs rN cache_statsrs4  " " $DIIMMdll <rs 0# ,, ')'W'*. '=A7'('Y'*. '=A:' )'Y'+/'>B9' ( 'X '+/ '>B9 '  ( ' * 9 \9 '^ RR y~~ 4% F  3@#@14@9<@@.*.&* & # /2  <  7$"! !! !  !  !! !*48- ! !25 !FJM6:22GK!9='02!$2582" ' 8AE%)@"@.1@:%: ::  :  :  :: : :|