from fastapi import APIRouter, HTTPException
from .api_model import GenerateWebRequest
from .logger import logger
from .fake_comments_generator import FakeCommentsGenerator

router = APIRouter()
fake_comments_generator = FakeCommentsGenerator()


@router.post("/generate")
async def generate(data: GenerateWebRequest):
    prompt = data.prompt
    history = data.history or ""

    prompt = f"{history}\n\n{prompt}"
    print("******history******", history)

    try:
        agent_response = fake_comments_generator.run_flow(prompt=prompt)
        logger.info(f"Response data: {agent_response}")
        response = {
            "data": agent_response,
            "status": True
        }
    except Exception as e:
        log_error(e)
        response = {
            "data": str(e),
            "status": False
        }

    return response


def log_error(e: Exception):
    if isinstance(e, HTTPException):
        logger.error(f"An error occurred during answers generation: {e.detail}")
    else:
        logger.error(f"An unexpected error occurred: {e}")