from fastapi import FastAPI
from dialogue_generator import (
    generate_response,
    generate_response_from_audio,
    generate_response_from_webcam,
    DialogueOutput,
    TextDialogueInput,
    AudioDialogueInput,
    WebcamDialogueInput
)

app = FastAPI()

@app.post("/process_text", response_model=DialogueOutput)
async def process_dialogue(dialogue: TextDialogueInput):
    response = generate_response(dialogue)
    return response

@app.post("/process_audio", response_model=DialogueOutput)
async def process_dialogue(dialogue: AudioDialogueInput):
    response = generate_response_from_audio(dialogue)
    return response

@app.post("/process_images", response_model=DialogueOutput)
async def process_dialogue(dialogue: WebcamDialogueInput):
    response = generate_response_from_webcam(dialogue)
    return response