16 lines
513 B
Python
16 lines
513 B
Python
from image_processor.images_finder import generate_image
|
|
from ppt_generator.models.query_and_prompt_models import (
|
|
ImageAspectRatio,
|
|
ImagePromptWithThemeAndAspectRatio,
|
|
)
|
|
|
|
|
|
async def test_image_generation():
|
|
await generate_image(
|
|
ImagePromptWithThemeAndAspectRatio(
|
|
image_prompt="halloween night at a haunted museum",
|
|
theme_prompt="halloween",
|
|
aspect_ratio=ImageAspectRatio.r_1_1,
|
|
),
|
|
output_path="/tmp/presenton/generated_image.jpg",
|
|
)
|