Update app.py

app.py

@@ -73,36 +73,37 @@ def translate_text(text, source_lang, target_lang):
     return translated_text
 
 # Summarization function with multi-language support
-def summarize_text(text, source_language="English", target_language="English"):
-    source_lang_code = LANGUAGES[source_language]
-    target_lang_code = LANGUAGES[target_language]
+def summarize_text(text, input_language="English", output_language="English"):
+    input_lang_code = LANGUAGES[input_language]
+    output_lang_code = LANGUAGES[output_language]
     
     # If the input language is not English, translate to English
-    if source_lang_code != "en_XX":
-        text = translate_text(text, source_lang_code, "en_XX")
+    if input_lang_code != "en_XX":
+        text = translate_text(text, input_lang_code, "en_XX")
     
     # Summarize the text using mBART
     inputs = multilingual_summarization_tokenizer(text, return_tensors='pt', padding=True, truncation=True)
     summary_ids = multilingual_summarization_model.generate(inputs['input_ids'], num_beams=4, max_length=200, early_stopping=True)
     summary = multilingual_summarization_tokenizer.decode(summary_ids[0], skip_special_tokens=True)
     
-    # Translate summary to the target language if needed
-    if target_lang_code != "en_XX":
-        summary = translate_text(summary, "en_XX", target_lang_code)
+    # Translate summary to the output language if needed
+    if output_lang_code != "en_XX":
+        summary = translate_text(summary, "en_XX", output_lang_code)
     
     return summary
 
 # Streamlit interface
 st.title("Multi-Language Text Summarization Tool")
+st.write("Enter the text you want to summarize, select the input language, and choose the output language for the summary.")
 
-text = st.text_area("Input Text")
-source_language = st.selectbox("Source Language", options=list(LANGUAGES.keys()), index=list(LANGUAGES.keys()).index("English"))
-target_language = st.selectbox("Target Language", options=list(LANGUAGES.keys()), index=list(LANGUAGES.keys()).index("English"))
+text_input = st.text_area("Input Text")
+input_language = st.selectbox("Input Language", options=list(LANGUAGES.keys()), index=list(LANGUAGES.keys()).index("English"))
+output_language = st.selectbox("Output Language", options=list(LANGUAGES.keys()), index=list(LANGUAGES.keys()).index("English"))
 
 if st.button("Summarize"):
-    if text:
-        summary = summarize_text(text, source_language, target_language)
-        st.subheader("Summary")
+    if text_input:
+        summary = summarize_text(text_input, input_language, output_language)
+        st.write("Summary:")
         st.write(summary)
     else:
-        st.warning("Please enter text to summarize.")
+        st.warning("Please enter some text to summarize.")