api/routers/human_idea_refining.py

from fastapi import APIRouter
from api.stored_data import stored_data
from src.genai.human_refined_ideation.agent import human_refined_idea
from api.schemas.human_idea_refining import RefineIdeationRequest
router= APIRouter()
human_refine_graph = human_refined_idea()

@router.post("/human-idea-refining")
def human_idea_refine_endpoint(request:RefineIdeationRequest):
    stored_data['human_ideation_interactions'].append({"role": "user", "content": request.query})
    response = human_refine_graph.invoke(
        {
        'query': stored_data['human_ideation_interactions'],
        'business_details': stored_data["business_details"],
        'final_ideation': stored_data.get('final_ideation',["","","",""]),
        },config={"configurable": {"thread_id": request.thread_id}}
    )
    stored_data['human_ideation_interactions'].append({"role": "assistant", "content": response['result']})
    stored_data['refined_ideation'] = stored_data['human_ideation_interactions'][-1]['content']
    return {'response' : stored_data['human_ideation_interactions'][-1]['content'] }