api/routers/generate_final_story.py

from fastapi import APIRouter
from fastapi.responses import StreamingResponse
from api.stored_data import stored_data
from src.genai.brainstroming_agent.utils.utils import generate_final_story

router= APIRouter()

@router.post("/generate-final-story")
def generate_final_story_endpoint():
    def event_stream():
        for chunk in generate_final_story(stored_data.get('brainstorming_response') or stored_data.get('business_details')):
            yield chunk

    return StreamingResponse(event_stream(), media_type="text/event-stream")