Update app.py

app.py

@@ -23,7 +23,7 @@ def translate(text, src_lang, tgt_lang):
     result = translation_pipeline(text)
     return result[0]['translation_text']
 
-def file_translate(sorce_file_path, pred_file_path):
+def file_translate(sorce_file_path, src_lang, tgt_lang, pred_file_path):
     sorce_list = []
     with open(sorce_file_path, "r", encoding="utf-8") as sorce_file:
         for line in sorce_file:
@@ -31,7 +31,7 @@ def file_translate(sorce_file_path, pred_file_path):
 
     pred_list = []
     for line in sorce_list:
-        pred_list.append(translate(line, list(langs_dict.keys())[0], list(langs_dict.keys())[1]))
+        pred_list.append(translate(line, src_lang, tgt_lang))
 
     with open(pred_file_path, "w", encoding="utf-8") as output_file:
         for translation in pred_list:
@@ -58,6 +58,8 @@ if __name__ == '__main__':
     fn=file_translate,
     inputs=[
         gr.components.File(label="Input File"),
+        gr.components.Dropdown(label="زبان مبدا", choices=list(langs_dict.keys())),
+        gr.components.Dropdown(label="زبان مقصد", choices=list(langs_dict.keys())),
         gr.components.Textbox(label="Output File Name (optional)"),
     ],
     outputs=[