Update app.py

app.py

@@ -7,12 +7,17 @@ import random
 import os
 import urllib.parse
 import hashlib
+from deep_translator import GoogleTranslator
+from langdetect import detect
 
 app = Flask(__name__)
 
 # Максимальные значения для ширины и высоты
-MAX_WIDTH = 850
-MAX_HEIGHT = 850
+#MAX_WIDTH = 1024
+#MAX_HEIGHT = 1024
+
+MAX_WIDTH = 512
+MAX_HEIGHT = 512
 
 # Кэш на 10 минут
 cache = TTLCache(maxsize=100, ttl=600)
@@ -77,6 +82,22 @@ def get_image(prompt):
     # Декодируем URL-кодированный prompt
     prompt = urllib.parse.unquote(prompt)
 
+    # Определяем язык промпта
+    try:
+        language = detect(prompt)
+    except Exception as e:
+        app.logger.error(f"Error detecting language: {e}")
+        abort(500, description="Error detecting language")
+
+    # Переводим промпт, если он не на английском языке
+    if language != 'en':
+        try:
+            translator = GoogleTranslator(source=language, target='en')
+            prompt = translator.translate(prompt)
+        except Exception as e:
+            app.logger.error(f"Error translating prompt: {e}")
+            abort(500, description="Error translating prompt")
+
     # Масштабируем размеры изображения, если они превышают максимальные значения
     width, height = scale_dimensions(width, height, MAX_WIDTH, MAX_HEIGHT)
 
@@ -103,4 +124,4 @@ def health_check():
     return "OK", 200
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860, debug=False)
+    app.run(host='0.0.0.0', port=7860, debug=False)