api/routers/orchestration.py

from fastapi import APIRouter, Depends
from pydantic import BaseModel
from api.stored_data import stored_data
from src.genai.orchestration_agent.agent import orchestration_chat
from typing import Optional

class OrchestrationRequest(BaseModel):
    message: str
    image_base64 : Optional[list] = []

router= APIRouter()

@router.post("/orchestration")
def orchestration_endpoint(request:OrchestrationRequest):
    print('Image:',request.image_base64)
    result = orchestration_chat(request.message , request.image_base64)
    if result.image_caption != '':
        stored_data['image_caption']=result.image_caption
    if result.video_idea !='' or result.video_idea != 'null':
        stored_data['refined_ideation']= result.video_idea
    if result.video_story!='' or result.video_story!='null':
        stored_data['final_story']= result.video_story
    print('Idea:',stored_data['refined_ideation'])
    print('Story:', stored_data['final_story'])

    return {'tool_response': result.tool ,
             'message_response': result.query_response, 
             'image_caption':result.image_caption,
             'video_idea': result.video_idea,
             'video_story': result.video_story}