Spaces:
Runtime error
Runtime error
| 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") | |
| 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 |