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


app = FastAPI()


class RequestBody(BaseModel):
    prompt: str


@app.post("/api/v1")
async def generate_response(request_body: RequestBody):
    prompt = request_body.prompt
    
    def is_russian_word(word):
        return all('а' <= char <= 'я' or 'А' <= char <= 'Я' for char in word)

    def detect_russian_text(text):
        words = re.findall(r'\b\w+\b', text)
        if not words:
            return False
        russian_word_count = sum(1 for word in words if is_russian_word(word))
        russian_ratio = russian_word_count / len(words)
        return russian_ratio > 0.5
    
    if detect_russian_text(prompt):
        try:
            all_chunk = GoogleTranslator(source='ru', target='en').translate(prompt)
        except:
            all_chunk = prompt
            
    return {"response": all_chunk}