from fastapi import FastAPI, Request
from pydantic import BaseModel
import requests
import os
import uuid
import re
from deep_translator import GoogleTranslator


app = FastAPI()


class RequestBody(BaseModel):
    prompt: str


@app.post("/api/v1")
async def generate_response(request_body: RequestBody):
    prompt = request_body.prompt
     
    try:
        all_chunk = GoogleTranslator(source='ru', target='en').translate(prompt)
    except:
        all_chunk = prompt
            
    return {"response": all_chunk}