api/routers/generate_image.py

import logging
from fastapi import APIRouter
from api.stored_data import stored_data
from src.genai.brainstroming_agent.utils.utils import ImageGenerator
from api.backup_prompts import image_gen_backup_prompt

router = APIRouter()
app_logger = logging.getLogger("app_logger")
error_logger = logging.getLogger("error_logger")

@router.get("/generate-image") 
def generate_image_endpoint():
    try:
        image = ImageGenerator().generate_image(str(stored_data.get('final_story',image_gen_backup_prompt))
                            ,str(stored_data.get('business_details'))
                            ,str(stored_data.get('refined_ideation',image_gen_backup_prompt)))
        stored_data['generated_image']=image
        app_logger.info('The image is Generated')
        return {
            'response':image
        }
    except Exception as e:
        error_logger.error(f'Unable to generate image: {e}')
        raise