add translation

app.py

@@ -2,6 +2,8 @@
 # https://gist.github.com/brandon8863/3ee552870a066eba463d37f0b39a547a
 import gradio as gr
 import webpages.blip_image_captioning as blip_image_captioning 
+import webpages.text_generation as text_generation 
+
 import home as home
 
  
@@ -15,6 +17,8 @@ def routs():
                 with gr.Column() as result:
                     gr.Button("🏠 Home", link="/?page=home")
                     gr.Button("👀 Find out what in the photo", link="/?page=blip_image_captioning")
+                    gr.Button("📝 Text Generation", link="/?page=text_generation")
+
  
 def handePages(local_state):
      with gr.Column(scale=30):
@@ -29,6 +33,8 @@ def handePages(local_state):
                     return home.get(), local_state
                 elif local_state  == "blip_image_captioning":
                     return blip_image_captioning.get(local_state), local_state
+                elif local_state  == "text_generation":
+                    return text_generation.get(local_state), local_state
                 else:
                     return (
                         home.get_not_found_page(local_state),

flagged/log.csv

@@ -0,0 +1,2 @@
+text,output,flag,username,timestamp
+how are you,كيف حالك,,,2024-09-16 23:42:11.972065

webpages/text_generation.py

@@ -0,0 +1,82 @@
+# Use a pipeline as a high-level helper
+from transformers import pipeline
+import gradio as gr
+import torch
+import gc
+
+translator = pipeline("translation", model="facebook/nllb-200-distilled-600M",  torch_dtype=torch.bfloat16)
+
+lang_list = ["English", "عربى", "French"]
+lang_code = {"English":"eng_Latn", "عربى" : "arz_Arab", "French" : "fra_Latn"}
+ 
+from_lang="eng_Latn"
+to_lang="arz_Arab"
+
+def claer():
+    del translator
+    gc.collect()
+
+def transalate(text) :
+    # claer()
+    text_translated = translator(text,
+                             src_lang=from_lang,
+                             tgt_lang=to_lang)
+    print(to_lang)
+
+    return  text_translated[0]['translation_text']
+
+def form():
+        return gr.Interface(transalate,
+                        inputs="textbox",
+                        outputs="text")
+def rs_change_from(c):
+     from_lang = lang_code[c]
+     print(from_lang)
+
+def rs_change_to(c):
+     to_lang = lang_code[c]
+     print(to_lang)
+
+def get(local_state):
+    with gr.Column() as result:
+        gr.HTML("<br/>")
+        gr.Markdown("## Text Generation")
+        dropdownFrom =  gr.Dropdown(
+           lang_code,  label="Transalte From", info="Will add more  later!"
+        )
+        dropdownFrom.select(rs_change_from,dropdownFrom)
+        dropdownTo = gr.Dropdown(
+           lang_list, label="Transalte To", info="Will add more  later!"
+        )
+        dropdownTo.select(rs_change_to,dropdownTo)
+
+        form()
+        gr.HTML("<br/><br/><br/><br/>")
+        
+    return result
+
+
+
+
+
+
+# - Afrikaans: afr_Latn
+# - Chinese: zho_Hans
+# - Egyptian Arabic: arz_Arab
+# - French: fra_Latn
+# - German: deu_Latn
+# - Greek: ell_Grek
+# - Hindi: hin_Deva
+# - Indonesian: ind_Latn
+# - Italian: ita_Latn
+# - Japanese: jpn_Jpan
+# - Korean: kor_Hang
+# - Persian: pes_Arab
+# - Portuguese: por_Latn
+# - Russian: rus_Cyrl
+# - Spanish: spa_Latn
+# - Swahili: swh_Latn
+# - Thai: tha_Thai
+# - Turkish: tur_Latn
+# - Vietnamese: vie_Latn
+# - Zulu: zul_Latn