Update app.py

app.py

@@ -6,9 +6,13 @@ from all_models import models
 from io import BytesIO
 from PIL import Image
 from fastapi import FastAPI, Request
+from translatepy import Translator
 
 css_code = os.getenv("DazDinGo_CSS")
 
+# Initialize translator
+translator = Translator()
+
 # Load models
 models_load = {}
 for model in models:
@@ -22,9 +26,11 @@ app = FastAPI()
 def gen_image(model_str, prompt):
     if model_str == 'NA':
         return None
+    # Translate prompt to English
+    translated_prompt = str(translator.translate(prompt, 'English'))
     noise = str(randint(0, 4294967296))
     klir = '| ultra detail, ultra elaboration, ultra quality, perfect'
-    return models_load[model_str](f'{prompt} {klir} {noise}')
+    return models_load[model_str](f'{translated_prompt} {klir} {noise}')
 
 def image_to_base64(image):
     buffered = BytesIO()
@@ -45,7 +51,9 @@ async def api_generate(request: Request):
     if model not in models:
         return {"error": "Model not found"}
     
-    image = gen_image(model, prompt)
+    # Translate prompt to English for API endpoint too
+    translated_prompt = str(translator.translate(prompt, 'English'))
+    image = gen_image(model, translated_prompt)
     if image is None:
         return {"error": "Image generation failed"}
     
@@ -54,7 +62,8 @@ async def api_generate(request: Request):
     return {
         "status": "success",
         "model": model,
-        "prompt": prompt,
+        "original_prompt": prompt,
+        "translated_prompt": translated_prompt,
         "image_base64": base64_str,
         "image_format": "jpeg"
     }
@@ -102,7 +111,9 @@ def make_me():
             json_output = gr.JSON(label="API Response")
             
             def generate_wrapper(model_str, prompt):
-                image = gen_image(model_str, prompt)
+                # Translate prompt to English
+                translated_prompt = str(translator.translate(prompt, 'English'))
+                image = gen_image(model_str, translated_prompt)
                 if image is None:
                     return None, {"error": "Generation failed"}
                 
@@ -110,7 +121,8 @@ def make_me():
                 response = {
                     "status": "success",
                     "model": model_str,
-                    "prompt": prompt,
+                    "original_prompt": prompt,
+                    "translated_prompt": translated_prompt,
                     "image_base64": base64_str,
                     "image_format": "jpeg"
                 }