Update main.py

main.py

@@ -13,6 +13,7 @@ import requests
 import io
 from PIL import Image
 import base64
+from deep_translator import GoogleTranslator
 
 app = FastAPI()
 
@@ -69,7 +70,12 @@ def generate_image(inputs, is_negative, steps, cfg_scale, seed):
         global request_counter
         request_counter += 1
         timestamp = f"{time.time()}_{request_counter}"
-        prompt_with_noise = add_random_noise(inputs) + f" - {timestamp}"
+        
+        # Translate inputs to English
+        translator_to_en = GoogleTranslator(source='auto', target='english')
+        english_inputs = translator_to_en.translate(inputs)
+        
+        prompt_with_noise = add_random_noise(english_inputs) + f" - {timestamp}"
         payload = {
             "inputs": prompt_with_noise,
             "is_negative": is_negative,
@@ -93,6 +99,7 @@ def generate_image(inputs, is_negative, steps, cfg_scale, seed):
 
 
 
+
 def get_prompts(prompt_text):
     if not prompt_text:
         return "Please enter text before generating prompts.رجاء ادخل النص اولا"