Update app.py

app.py

@@ -6,12 +6,12 @@ from all_models import models
 from io import BytesIO
 from PIL import Image
 from fastapi import FastAPI, Request
-from translatepy import Translator
+from deep_translator import GoogleTranslator
 
 css_code = os.getenv("DazDinGo_CSS")
 
 # Initialize translator
-translator = Translator()
+translator = GoogleTranslator(source='auto', target='en')
 
 # Load models
 models_load = {}
@@ -27,7 +27,7 @@ def gen_image(model_str, prompt):
     if model_str == 'NA':
         return None
     # Translate prompt to English {noise} {klir}
-    translated_prompt = str(translator.translate(prompt, 'English'))
+    translated_prompt = translator.translate(prompt)
     noise = str(randint(0, 4294967296))
     klir = '| ultra detail, ultra elaboration, ultra quality, perfect'
     return models_load[model_str](f'{translated_prompt}')
@@ -52,7 +52,7 @@ async def api_generate(request: Request):
         return {"error": "Model not found"}
     
     # Translate prompt to English for API endpoint too
-    translated_prompt = str(translator.translate(prompt, 'English'))
+    translated_prompt = translator.translate(prompt)
     image = gen_image(model, translated_prompt)
     if image is None:
         return {"error": "Image generation failed"}
@@ -112,7 +112,7 @@ def make_me():
             
             def generate_wrapper(model_str, prompt):
                 # Translate prompt to English
-                translated_prompt = str(translator.translate(prompt, 'English'))
+                translated_prompt = translator.translate(prompt)
                 image = gen_image(model_str, translated_prompt)
                 if image is None:
                     return None, {"error": "Generation failed"}