Update app.py

app.py

@@ -126,7 +126,7 @@ lang_id = [
 d_lang = lang_id[21]
 #d_lang_code = d_lang.code
 
-def trans_page(input,trg):
+def trans_page(input,input2,trg):
     src_lang = d_lang.code
     for lang in lang_id:
             if lang.name == trg:
@@ -140,21 +140,22 @@ def trans_page(input,trg):
     else:
         translated_text=input
         pass
-    """    
-    if trg_lang != src_lang:
+
         
+    if trg_lang != src_lang:
+        new_lang = []
         tokenizer.src_lang = src_lang
-        with torch.no_grad():
-            #lang_tr = lang_id
-            encoded_input = tokenizer(lang_id, return_tensors="pt").to(device)
-            generated_tokens = model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
-            translated_text1 = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
+        for langs in lang_id:
+            with torch.no_grad():
+                encoded_input = tokenizer(lang.name, return_tensors="pt").to(device)
+                generated_tokens = model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
+                trans_langs = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
+            new_lang.append(trans_langs)
     else:
-        translated_text1=input1
+        new_lang=lang_id
         pass
-    #return translated_text,gr.Dropdown.update(choices=list(translated_text1.keys()))
-    """
-    return translated_text
+        
+    return translated_text,gr.Dropdown.update(choices=[l.name for l in new_lang]),gr.Dropdown.update(choices=[l.name for l in new_lang])
 
 def trans_to(input,src,trg):
     for lang in lang_id:
@@ -185,7 +186,7 @@ with gr.Blocks() as transbot:
         gr.Column()
         with gr.Column():
             with gr.Row():
-                t_space = gr.Dropdown(label="Translate Space", choices=[l.name for l in lang_id], value="English")
+                t_space = gr.Dropdown(label="Translate Space to:", choices=[l.name for l in lang_id], value="English")
                 #t_space = gr.Dropdown(label="Translate Space", choices=list(lang_id.keys()),value="English")
                 t_submit = gr.Button("Translate Space")
         gr.Column()
@@ -207,7 +208,7 @@ with gr.Blocks() as transbot:
                     message = gr.Textbox(label="Prompt",placeholder="Enter Prompt",lines=4)
                     translated = gr.Textbox(label="Translated",lines=4,interactive=False)
         gr.Column()
-    t_submit.click(trans_page,[md,t_space],[md])
+    t_submit.click(trans_page,[md,t_space],[md,lang_from,lang_to])
     
     submit.click(trans_to, inputs=[message,lang_from,lang_to], outputs=[translated])
 transbot.queue(concurrency_count=20)