3jr& SSKrSSKrSSKJrJrJrJr SSKrSSKJ r \R"\ 5r Sr SrSrSS\\\\4S\\\\R"\\4S \S \\4S jjrS \\R"\\4S \\ R4S jrS\\S \S\S \\4SjrS\\S\\ RS \S\S \\4 SjrSS\\ RS\S \\4SjjrSrS \S \\4Sjrg)N)ListOptionalTupleUnion)Imagea{ You are an expert cinematic director with many award winning movies, When writing prompts based on the user input, focus on detailed, chronological descriptions of actions and scenes. Include specific movements, appearances, camera angles, and environmental details - all in a single flowing paragraph. Start directly with the action, and keep descriptions literal and precise. Think like a cinematographer describing a shot list. Do not change the user input intent, just enhance it. Keep within 150 words. For best results, build your prompts using this structure: Start with main action in a single sentence Add specific details about movements and gestures Describe character/object appearances precisely Include background and environment details Specify camera angles and movements Describe lighting and colors Note any changes or sudden events Do not exceed the 150 word limit! Output the enhanced prompt only. Examples: user prompt: A man drives a toyota car. enhanced prompt: A person is driving a car on a two-lane road, holding the steering wheel with both hands. The person's hands are light-skinned and they are wearing a black long-sleeved shirt. The steering wheel has a Toyota logo in the center and black leather around it. The car's dashboard is visible, showing a speedometer, tachometer, and navigation screen. The road ahead is straight and there are trees and fields visible on either side. The camera is positioned inside the car, providing a view from the driver's perspective. The lighting is natural and overcast, with a slightly cool tone. user prompt: A young woman is sitting on a chair. enhanced prompt: A young woman with dark, curly hair and pale skin sits on a chair; she wears a dark, intricately patterned dress with a high collar and long, dark gloves that extend past her elbows; the scene is dimly lit, with light streaming in from a large window behind the characters. user prompt: Aerial view of a city skyline. enhanced prompt: The camera pans across a cityscape of tall buildings with a circular building in the center. The camera moves from left to right, showing the tops of the buildings and the circular building in the center. The buildings are various shades of gray and white, and the circular building has a green roof. The camera angle is high, looking down at the city. The lighting is bright, with the sun shining from the upper left, casting shadows from the buildings. aIYou are an expert cinematic director with many award winning movies, When writing prompts based on the user input, focus on detailed, chronological descriptions of actions and scenes. Include specific movements, appearances, camera angles, and environmental details - all in a single flowing paragraph. Start directly with the action, and keep descriptions literal and precise. Think like a cinematographer describing a shot list. Keep within 150 words. For best results, build your prompts using this structure: Describe the image first and then add the user input. Image description should be in first priority! Align to the image caption if it contradicts the user text input. Start with main action in a single sentence Add specific details about movements and gestures Describe character/object appearances precisely Include background and environment details Specify camera angles and movements Describe lighting and colors Note any changes or sudden events Align to the image caption if it contradicts the user text input. Do not exceed the 150 word limit! Output the enhanced prompt only. cUR5S:aUR5S::deUS-S- nURSSS5nUR5R 5S-R S5n[ R"U5$)Nruint8)minmaxpermutecpunumpyastyper fromarray)tensor numpy_images S/home/wildlama/comfy/ComfyUI/custom_nodes/ComfyUI-LTXVideo/prompt_enhancer_utils.py tensor_to_pilr:s~ ::<2 &**,!"33 3qjA F^^Aq! $F::<%%'#-55g>K ??; ''promptconditioning_itemsmax_new_tokensreturnc [U[5(aU/OUnUc[UUUU[5nU$USn[ U5n [ U 5[ U5:XdS5e[ UUUUUU U[5nU$)Nrz:Number of conditioning frames must match number of prompts) isinstancestr_generate_t2v_promptT2V_CINEMATIC_PROMPT(_get_first_frames_from_conditioning_itemlen_generate_i2v_promptI2V_CINEMATIC_PROMPT) image_caption_modelimage_caption_processorprompt_enhancer_modelprompt_enhancer_tokenizerrrrpromptsfirst_frame_conditioning_item first_framess rgenerate_cinematic_promptr.Ks%VS11vhvG!& ! %    D N));1(=%? ) < C %   H G H '  # ! %       Nrconditioning_itemc USn[URS5Vs/sHn[XSS2SSS2SS245PM sn$s snf)Nr)rangeshaper)r/ frames_tensoris rr#r#|sW&a(M}**1-. .A mq!QM23.  s"Ar+ system_promptc UVs/sHnSUS.SSU3S./PM nnUVs/sHnURUSSS9PM nnU"USS 9RUR5n [XX5$s snfs snf) Nsystemrolecontentuser user_prompt: FTtokenizeadd_generation_promptptreturn_tensors)apply_chat_templatetodevice_generate_and_decode_prompts) r)r*r+rr5pmessagesmtexts model_inputss rr!r!s  A- 8-s(; <     A "55 T 6     -U4HKK$$L (, % s A+A0r-c $[XU5n[XH5V V s/sHupSUS.SSU SU 3S./PM n n n U V s/sHn URU SSS9PM n n U"U S S 9RUR5n[ X#X5$s sn n fs sn f) Nr7r8r;r<z image_caption: FTr=r@rA)_generate_image_captionsziprCrDrErF)r'r(r)r*r+r-rr5image_captionsrGcrHrIrJrKs rr%r%s.lN0  1DA- 8-s:KA3(O P 1    A "55 T 6     -U4HKK$$L (, % s BB imagesc U/[U5-nU"XBSS9RUR5n[R"5 UR USUSSSSS9nSSS5 UR WS S 9$!,(df  N=f) Nr@rA input_ids pixel_valuesiF)rSrTr do_sample num_beamsTskip_special_tokens)r$rDrEtorchinference_modegenerate batch_decode)r'r(rQr5image_caption_promptsinputs generated_idss rrMrMs +Oc&k9 $db  # #$     +44[)/ 5   # / / SW / XX s A77 Bc6/SQn[R"U5$)z4 Randomly select a scene type to add to the prompt. )z+The scene is captured in real-life footage.z(The scene is computer-generated imagery.z%The scene appears to be from a movie.z'The scene appears to be from a TV show.z"The scene is captured in a studio.)randomchoice)typess r_get_random_scene_typeres E == rc [R"5 UR"S0UDSU0D6n[URU5VVs/sHupVU[ U5SPM nnnUR USS9nSSS5 WV s/sHoS[5S3-PM nn [U5 U$s snnf!,(df  NA=fs sn f)NrTrX .) rZr[r\rNrSr$r]reprint) r)r*rKroutputsrS output_idsr`decoded_promptsrGs rrFrFs    '00  +9  *-\-C-CW)M )M%  s9~' ()M  4@@ tA  ETTOqQ578::OOT /  Us#0B.B( B.?B?(B.. B<)N)z)loggingrbtypingrrrrrZPILr getLogger__name__loggerr"r&rr Tensorintfloatr.r#r!r%rMrerFrirrrxs //    8 $8((.KO. #tCy. ! . !eELL#u,D&E!FG .. #Y.bU\\356 %++#Y     #Y <" #Y " u{{# """ #Y"R. Y  Y Y  #Y Y.  TW #Yr