qi ndZddlmZddlmZmZddlmZmZddl m Z ddl m Z ddl mZGdd e Zy ) zImage generation service implementation. Provides base functionality for AI-powered image generation services that convert text prompts into images. )abstractmethod)AsyncGeneratorOptional)Frame TextFrame)FrameDirection) AIService)ImageGenSettingscleZdZdZdddeeffdZedede e dffdZ d e d e ffd Z xZS) ImageGenServicea Base class for image generation services. Processes TextFrames by using their content as prompts for image generation. Subclasses must implement the run_image_gen method to provide actual image generation functionality using their specific AI service. N)settingsr c @t|dd|xs ti|y)zInitialize the image generation service. Args: settings: The runtime-updatable settings for the image generation service. **kwargs: Additional arguments passed to the parent AIService. r N)super__init__r )selfr kwargs __class__s P/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/image_service.pyrzImageGenService.__init__s1  " !    promptreturnc Kyw)aGenerate an image from a text prompt. This method must be implemented by subclasses to provide actual image generation functionality using their specific AI service. Args: prompt: The text prompt to generate an image from. Yields: Frame: Frames containing the generated image (typically ImageRawFrame or URLImageRawFrame). Nr)rrs r run_image_genzImageGenService.run_image_gen.s  sframe directioncKt|||d{t|tr}|j ||d{|j d{|j |j|jd{|jd{y|j ||d{y777n7>7(7w)aProcess frames for image generation. TextFrames are used as prompts for image generation, while other frames are passed through unchanged. Args: frame: The frame to process. direction: The direction of frame processing. N) r process_frame isinstancer push_framestart_processing_metricsprocess_generatorrtextstop_processing_metrics)rrrrs rrzImageGenService.process_frame>sg#E9555 eY '//%3 3 3//1 1 1((););EJJ)GH H H..0 0 0//%3 3 3 6 4 1 H 0 3siCC)CCCC1C C  C#C $C>C?CCC C CC)__name__ __module__ __qualname____doc__rr rrstrrrrrr __classcell__)rs@rr r sdBF  H-=$>    #  .2M    44>44rr N)r(abcrtypingrrpipecat.frames.framesrr"pipecat.processors.frame_processorrpipecat.services.ai_servicer pipecat.services.settingsr r rrrr1s* +2=16:4i:4r