Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 import os
 import uuid
 import re
+from deep_translator import GoogleTranslator
 
 
 app = FastAPI()
@@ -17,6 +18,22 @@ class RequestBody(BaseModel):
 async def generate_response(request_body: RequestBody):
     prompt = request_body.prompt
         
-    all_chunk = GoogleTranslator(source='ru', target='en').translate(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}