hiddlZddlZddlZddlZddlZddlmZmZmZm Z ddl m Z m Z ddl m Z ddlmZddlmZddlmZddlmZmZdd lmZej2eZGd d Zy) N)DictAnyOptionalList)datetimetimezone) AzureOpenAI)Session)Image)BytesIO) ChatThread ChatMessage) StorePersonaceZdZdZdZddededeedefdZ dded ed ede e eeffd Z dd ed eedefdZ dded ede e eefd eede eeff dZ ddededededeed eede eeffdZ dd edede efdZded ede e eefdefdZddedede eeffdZ d dededededeed eedede eeffdZy)!ChatbotLLMServicezVLLM service for the hwGpt chatbot with store persona integration and image generation.cttjdtjddtjd|_tjdd|_tjd|_tjd |_tjd d |_tjjtjjtjjtd d d }tjj|d ddd|_ tj|jdy)NAZURE_OPENAI_API_KEYAZURE_OPENAI_API_VERSIONz2024-02-15-previewAZURE_OPENAI_ENDPOINT)api_key api_versionazure_endpointAZURE_OPENAI_DEPLOYMENTzgpt-4 IDEOGRAM_URLIDEOGRAM_API_KEYBASE_URLz(https://hubwallet-dev.dreamztesting.com/z.. menu_designdesignerstaticgenerated_imagesT)exist_ok)r osgetenvclientdeployment_namerrrpathabspathjoindirname__file__ STATIC_PATHmakedirs)selfBASE_DIRs C/var/www/html/hubwallet-dev/src/marketing/apps/hwGpt/llm_service.py__init__zChatbotLLMService.__init__s!II45 "<>RS99%<=  "yy)BGLIIn5 " *< = *.XY 77??277<<0I4QUW[#\]77<<-XWij D$$t4Ndbstore_id branch_idreturnc N t|tr|dkr%tjd|dt |y|;t|tr|dkr&tjd|dt |d}|j t jt j|k(t jdk(}|r"|jt j|k(}|j}|syd|jxsd d |jxsd d |jxsd d |jxsd d |j xsd d|j"xsd d }|j%S#t&$r+}tjdt)|Yd}~yd}~wwxYw)z*Get store persona context for LLM prompts.rzInvalid store_id: z, type: zGNo specific store persona available. Please provide general assistance.NzInvalid branch_id: TzB Store Brand Persona: - Industry: z Not specifiedz$ - Target Audience: z - Brand Voice: z" - Content Style: z% - Analysis Summary: z% - Confidence Score: ao Based on this persona, you should: 1. Match the brand voice and tone 2. Use industry-specific terminology when appropriate 3. Tailor responses to the target audience 4. Maintain the established content style 5. Provide contextually relevant information zError getting store persona: zDUnable to retrieve store persona. Please provide general assistance.) isinstanceintloggererrortypequeryrfilterr3 is_activer4firstindustry audience_type brand_voice content_styleanalysis_summaryconfidence_scorestrip Exceptionstr)r-r2r3r4r<personapersona_contextes r/get_store_persona_contextz+ChatbotLLMService.get_store_persona_context,s/ Zh,A  1(8DNCSTU`$jC.HIYZN 29+Xd9oEVWX HH\*11%%1&&$.E  \%;%;y%HIkkmG`#$--@AB$$+$9$9$L_#MN ' 3 3 FGH"")"7"7"J?!KL%%,%=%=%P$QR%%,%=%=%P$QRO"#((* * Z LL8QA BY Zs$9E0B:E07A8E00 F$9!FF$ thread_id max_messagescz t|tr|dkrtjd|dgS|j t j t j|k(t jdk(jt jjj|j}|jg}|D]*}|j|j |j"d,|S#t$$r-}tj'dt)|gcYd}~Sd}~wwxYw)z/Build conversation context from thread history.rzInvalid thread_id: z, returning empty contextFrolecontentz%Error building conversation context: N)r7r8r9warningr<rr=rM is_deletedorder_by created_atdesclimitallreverseappendrQrRrGr:rH)r-r2rMrNmessagesconversation_contextmessagerKs r/build_conversation_contextz,ChatbotLLMService.build_conversation_context_s i-a!4YK?XYZ xx ,33%%2&&%/h{--2245eeL6I##%     #% # $++#LL&-  ( '  LL@QI JI s#/DCD D: "D5/D:5D: store_personathread_contextc<d|d|r|ndd}|jS)z ? "N 300$ Ky)""##$# H FF8  HHJ!# #J/#'OOM:&??<8F FF6N!) X\\ :F  IIK JJx JJv  KKB8;;-O`agajaj`kl m "++'//$MM"*"5"5"?"?"A !!))%~~"KK"("3"3"="="?#)#5#5"("3"3  'HKMHZ]`H`t)What type of content should I create for your target audience?z-How can I improve my social media engagement?z8What marketing campaigns should I plan for this quarter?z$How can I optimize my marketing ROI?)r$ryrzr{r%r|r^rRrFsplit startswithrGr9r:rH)r-r`rsuggestions_promptrs suggestionssrKs r/get_chat_suggestionsz&ChatbotLLMService.get_chat_suggestionsIs ( &   L "O ${{''33::**#)6HIJ ;H#**1-55==CCEKKDQK.9|QWWYq||\zO{1779|K|r? "}  LL>s1vhG H  s<BCC&C8C CC D$D D Dc 0 d}|r/|dd}dj|Dcgc]}|dd|dc}}d|d |d |d }|jjjj |j d d dd|dgdd}|j djjj} | dz } tjd| | Scc}w#t$r1} tjdt| d|dcYd} ~ Sd} ~ wwxYw)z]Generate a detailed image prompt for marketing content based on user input and store persona.rcNrrQrrRz Create a detailed image prompt for marketing content based on the following information: Store Persona: z4 User Request: zC Recent Conversation Context: a Requirements: - Create a professional marketing image suitable for social media, ads, or promotional materials - The image should align with the store's brand persona and target audience - Focus on visual appeal and marketing effectiveness - Include relevant product/service elements if mentioned - Use appropriate colors, style, and composition for the brand - Make it suitable for digital marketing use (social media, website, ads) IMPORTANT: - Do NOT include any text, logos, or written content in the image - Focus purely on visual elements and composition - Make it high-quality and professional - Ensure it's appropriate for marketing use Generate a detailed prompt (max 300 characters) that can be used to create this marketing image. rizYou are an expert at creating detailed image prompts for marketing content. Focus on visual elements, composition, and brand alignment.rPrji,rkrrzj Professional marketing image, no text, no logos, high quality, suitable for social media and advertising.z"Generated marketing image prompt: zError generating image prompt: z!Professional marketing image for z>, no text, high quality, suitable for social media advertising)r(r$ryrzr{r%r|r^rRrFr9rwrGr:rH) r-rfr`rgr]recent_messagesrpromptrs image_promptrKs r/generate_marketing_image_promptz1ChatbotLLMService.generate_marketing_image_promptysy; D#% #"6rs";'+yyBQ2;>s6{m2c)n%562($ O N " "# F:{{''33::**%3|}#7;H$++A.66>>DDFL I IL KK<\NK L c2f D LL:3q6(C D6|nECD D Ds.CCB'CC D$&D DDr session_idc  ttj}d|d|d}tjj |j |}|j r |jsddddStj|j d|jid|fd d d d  }tjd|j|jdk7rdd|jddS|j}|jdrt!|ddk(rddddS|ddjd}|sddddStj|d} | jdk7rdd| jddSt#j$t'| j(} | j+|tjd||j,j/dr |j,n|j,dz} | d|} tjd| d| ||ddS#t0$r:} tj3dt5| dt5| ddcYd} ~ Sd} ~ wwxYw) z9Generate an image using Ideogram API and save it locally.hwgpt_marketing__z.pngFzIdeogram API not configuredN)rrr: image_urlzApi-Key)N1x1)NON)NzText, text overlay, logos, brands, written content, words, letters, typography. No text allowed in the image. Strictly no text. No text in the image. No text in the image. No text in the image. No text in the image. No text in the image.)r aspect_ratio magic_promptnegative_prompt)headersfilesz%Received response from Ideogram API: rzIdeogram API error: datarz&No image data returned by Ideogram APIurlz%No image URL returned by Ideogram APIx)timeoutzFailed to download image: zImage saved locally at: /zstatic/generated_images/z Image URL: T)rrr local_pathfilenamer:zError generating image: )r8timer"r&r(r+rrrequestspostr9rw status_coderjsonrrxr openr rRsaverendswithrGr:rH)r-rr timestampimage_filenameimage_filepathrs response_datageneration_urlimage_responseimgbase_urlrrKs r/generate_image_with_ideogramz.ChatbotLLMService.generate_image_with_ideogramsX DIIK(I/ |1YKtLNWW\\$*:*:NKN$$D,A,A$:!% }}!!t44 $\2$1$0(_  H KK?@T@T?UV W##s*$3HMM?C!% %MMOM $$V,M&4I0Ja0O$E!% +621599%@N!$D!%&\\.#FN))S0$9.:T:T9UV!%**W^%;%;<=C HH^ $ KK2>2BC D)- (>(>s(Ct}}Y\I\H#*$<^A-H'0A8H')7H'!H'8H'9B-H'' I*0/I%I*%I*c K |j|||}g} t|tr|dkDr|j||} |j ||| } |j | |} | drd|d} n d|d| dd} t ||d |d } |j| |jt ||d | d | jdr d| dzdzndd| jdr| jdnd}|j||rm|jtjtj|k(j}|r(tj t"j$|_|j)|j+| |j+|t,j/d| jd|jd| j| j0| j2| j4| j6j9d|j|j0|j2|j4|j6j9d| | |t;|dkDr |dddzdS|dS#t<$rJ}t,j?dtA||jCdtA|ddcYd}~Sd}~wwxYww) zLProcess an image generation request and return both text response and image.rrrz9I've generated a marketing image based on your request: 'z'. The image has been created to align with your store's brand persona and is suitable for your marketing campaigns. You can use this image for social media posts, advertisements, or other promotional materials.z'I understand you'd like an image for: 'z9'. However, I encountered an issue generating the image: r:zE. Let me help you with other marketing strategies or try again later.rj image_requestrrz rz![Generated Image]()rcrN)rMrrQrRrrz4Processed image generation request: User message ID rT)rrRrQrrVrr)rrrfrimage_generationrrMrz+Error processing image generation request: FzcI apologize, but I encountered an error processing your image generation request. Please try again.rv)"rLr7r8r_rrrrrrr<r r=rr?rrrrrrrr9rwrRrQrrVrrxrGr:rHr)r-r2rrfr3r4rMrr`rgr image_result text_responserrrrKs r/ process_image_generation_requestz2ChatbotLLMService.process_image_generation_requests\  ::2xSM$& )S)i!m'+'F'Fr9'U$ ??m-AL  <<\:VLI&"[\h[ij}!~ "I,XQR^_fRgQhhm!n ##$, H FF8  HHJ!# $oTR^RbRbcnRo,|K/HH3Nuwxz.;G;K;KI;V,**;7\` F FF6N*-44Z]]i5OPVVX(0 X\\(BF% IIK JJx JJv  KKNx{{m[lmsmvmvlwx y "++'//$MM$,$9$9"*"5"5"?"?"A !!))%~~"KK$*$7$7"("3"3"="="?  %1 ,&HKMHZ]`H`t)2 7 ! 7 c3h 7~$(#'l ll l  l C= lC=l c3hlb!... c .`BDBDBD#4S>2 BD BDHZZ#Z^bcfhkck^lZD$(#'#g gg g  g C= gC=gg c3hgr1r)r"rloggingrrtypingrrrrrropenair sqlalchemy.ormr PILr ior src.marketing.apps.hwGpt.modelr r src.marketing.apps.persona.modelr getLoggerrr9rrr1r/rsM  ,,'"B9   8 $m m r1