aphilippov/python-server-api

1from fastapi import APIRouter, HTTPException2from .api_model import GenerateWebRequest3from .logger import logger4from .fake_comments_generator import FakeCommentsGenerator5 6router = APIRouter()7fake_comments_generator = FakeCommentsGenerator()8 9 10@router.post("/generate")11async def generate(data: GenerateWebRequest):12    prompt = data.prompt13    history = data.history or ""14 15    prompt = f"{history}\n\n{prompt}"16    print("******history******", history)17 18    try:19        agent_response = fake_comments_generator.run_flow(prompt=prompt)20        logger.info(f"Response data: {agent_response}")21        response = {22            "data": agent_response,23            "status": True24        }25    except Exception as e:26        log_error(e)27        response = {28            "data": str(e),29            "status": False30        }31 32    return response33 34 35def log_error(e: Exception):36    if isinstance(e, HTTPException):37        logger.error(f"An error occurred during answers generation: {e.detail}")38    else:39        logger.error(f"An unexpected error occurred: {e}")40