Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import pipeline
 from huggingface_hub import InferenceClient
+from language_directions import *
 # from IPython.display import Audio as IPythonAudio
 
 playground = gr.Blocks()
@@ -61,10 +62,13 @@ def translate(input_text, source, target):
     except KeyError:
       return "", f"Error: Translation direction {source_readable} to {target} is not supported by Helsinki Translation Models"
 
-def summarize(input):
-    output = summary_pipe(input)
+def summarize_translate(input_text, target_lang):
+    output = summary_pipe(input_text)
     summary_origin = output[0]['summary_text']
-    summary_translated = translate(summary_origin,'en','fr')
+    source = 'en'
+    target_lang_dict = get_all_source_languages()
+    target = target_lang_dict[target_lang]
+    summary_translated = translate(summary_origin,source,target)
     return summary_origin, summary_translated[0]
 
 def merge_tokens(tokens):
@@ -136,18 +140,22 @@ with playground:
                     gr.Markdown("""
                                 ## Text Summarization and Translation
                                 ### Summarize the paragraph and translate it into other language.
-                                > pipeline: summarization, model: [sshleifer/distilbart-cnn-12-6](https://huggingface.co/sshleifer/distilbart-cnn-12-6)
-                                > pipeline: translation, model: [Helsinki-NLP/opus-mt-en-fr](https://huggingface.co/Helsinki-NLP/opus-mt-en-fr)
+                                - pipeline: summarization, model: [sshleifer/distilbart-cnn-12-6](https://huggingface.co/sshleifer/distilbart-cnn-12-6)
+                                - pipeline: translation, model: [Helsinki-NLP/opus-mt-en-fr](https://huggingface.co/Helsinki-NLP/opus-mt-en-fr)
                                 """)
                     
                 with gr.Column(scale=1):
                     text_pipeline_button = gr.Button(value="Start Process", variant="primary")
+                    
 
             with gr.Row():
                 with gr.Column():
                     source_text = gr.Textbox(label="Text to summarize", lines=13)                    
                 with gr.Column():
                     summary_textoutput = gr.Textbox(lines=3, placeholder="", label="Text Summarization")
+                    target_language_dropdown = gr.Dropdown( choices=["Chinese", "French", "Spanish"],
+                                                            value="Chinese",
+                                                            label="Translate to Language")
                     translated_textbox = gr.Textbox(lines=3, placeholder="", label="Translated Result")
                     Text_Clear_button = gr.ClearButton(components=[source_text, summary_textoutput, translated_textbox], value="Clear")
             
@@ -158,7 +166,7 @@ with playground:
                             "Tower Bridge is a Grade I listed combined bascule, suspension, and, until 1960, cantilever bridge[1] in London, built between 1886 and 1894, designed by Horace Jones and engineered by John Wolfe Barry with the help of Henry Marc Brunel.[2] It crosses the River Thames close to the Tower of London and is one of five London bridges owned and maintained by the City Bridge Foundation, a charitable trust founded in 1282. The bridge was constructed to connect the 39 per cent of London's population that lived east of London Bridge, while allowing shipping to access the Pool of London between the Tower of London and London Bridge. The bridge was opened by Edward, Prince of Wales and Alexandra, Princess of Wales on 30 June 1894."
                     ], inputs=[source_text], outputs=[summary_textoutput, translated_textbox], run_on_click=True, cache_examples=True, fn=summarize)
  
-            text_pipeline_button.click(summarize, inputs=[source_text], outputs=[summary_textoutput, translated_textbox])
+            text_pipeline_button.click(summarize_translate, inputs=[source_text, target_language_dropdown], outputs=[summary_textoutput, translated_textbox])
             
         ## ================================================================================================================================
         ## Find entities