o i @shdZddlmZddlmZmZddlmZmZddl m Z ddl m Z ddl mZGdd d e Zd S) 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)ImageGenSettingscsdeZdZdZdddeeffddZedede e dffd d Z d e d e ffd d Z ZS)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 s tjdd|p ti|dS)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__r R/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/image_service.pyrs  zImageGenService.__init__promptreturncsdS)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 )rrr r r run_image_gen.szImageGenService.run_image_genframe directioncst||IdHt|tr5|||IdH|IdH|||jIdH| IdHdS|||IdHdS)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)rrrrr rr>s zImageGenService.process_frame)__name__ __module__ __qualname____doc__rr rrstrrrrrr __classcell__r r rrr s r N)r#abcrtypingrrpipecat.frames.framesrr"pipecat.processors.frame_processorrpipecat.services.ai_servicerpipecat.services.settingsr r r r r rs