Update app.py

app.py

@@ -1,8 +1,9 @@
-#!/usr/bin/env python
-
 import os
 import random
 import uuid
+from urllib.parse import quote
+from requests import get
+from bs4 import BeautifulSoup
 
 import gradio as gr
 import numpy as np
@@ -25,6 +26,8 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 NUM_IMAGES_PER_PROMPT = 1
 
+valid_languages = {'fon', 'fr', 'yo', 'en'}
+
 if torch.cuda.is_available():
     pipe = DiffusionPipeline.from_pretrained(
         "playgroundai/playground-v2.5-1024px-aesthetic",
@@ -36,9 +39,9 @@ if torch.cuda.is_available():
     if ENABLE_CPU_OFFLOAD:
         pipe.enable_model_cpu_offload()
     else:
-        pipe.to(device)    
+        pipe.to(device)
         print("Loaded on Device!")
-    
+
     if USE_TORCH_COMPILE:
         pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
         print("Model Compiled!")
@@ -56,9 +59,28 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     return seed
 
 
+def translate_to_english(phrase, src_lang):
+    if src_lang == 'en':
+        return phrase
+
+    dest_lang = 'en'
+    encoded_phrase = quote(phrase)
+    url = f"https://translate.glosbe.com/{src_lang}-{dest_lang}/{encoded_phrase}"
+    response = get(url)
+
+    if response.status_code == 200:
+        soup = BeautifulSoup(response.text, 'html.parser')
+        translation_div = soup.find('div', class_='w-full h-full bg-gray-100 h-full border p-2 min-h-25vh sm:min-h-50vh whitespace-pre-wrap break-words')
+        translation = translation_div.text if translation_div else "Translation not found"
+        return translation
+    else:
+        return "Error: Unable to translate"
+
+
 @spaces.GPU(enable_queue=True)
 def generate(
-    prompt: str,
+    phrase: str,
+    input_lang: str,
     negative_prompt: str = "",
     use_negative_prompt: bool = False,
     seed: int = 0,
@@ -73,9 +95,14 @@ def generate(
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)
 
+    if input_lang != 'en':
+        prompt = translate_to_english(phrase, input_lang)
+    else:
+        prompt = phrase
+
     if not use_negative_prompt:
         negative_prompt = None  # type: ignore
-    
+
     images = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
@@ -116,6 +143,7 @@ with gr.Blocks(css=css) as demo:
     )
     with gr.Group():
         with gr.Row():
+            input_lang = gr.Dropdown(choices=list(valid_languages), value='en', label='Input Language')
             prompt = gr.Text(
                 label="Prompt",
                 show_label=False,
@@ -168,7 +196,7 @@ with gr.Blocks(css=css) as demo:
 
     gr.Examples(
         examples=examples,
-        inputs=prompt,
+        inputs=[prompt, input_lang],
         outputs=[result, seed],
         fn=generate,
         cache_examples=CACHE_EXAMPLES,
@@ -190,6 +218,7 @@ with gr.Blocks(css=css) as demo:
         fn=generate,
         inputs=[
             prompt,
+            input_lang,
             negative_prompt,
             use_negative_prompt,
             seed,
@@ -197,10 +226,10 @@ with gr.Blocks(css=css) as demo:
             height,
             guidance_scale,
             randomize_seed,
-        ],
-        outputs=[result, seed],
-        api_name="run",
-    )
+       ],
+       outputs=[result, seed],
+       api_name="run",
+   )
 
 if __name__ == "__main__":
-    demo.queue(max_size=20).launch()
+   demo.queue(max_size=20).launch()