Update app.py

app.py

@@ -73,15 +73,8 @@ 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]
-    
-    # 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")
-    
-    # Summarize the text using mBART
+def summarize_text(text, target_language="English"):
+    # Summarize the text using mBART (assuming input text is in English)
     inputs = multilingual_summarization_tokenizer(text, return_tensors='pt', padding=True, truncation=True)
     summary_ids = multilingual_summarization_model.generate(
         inputs['input_ids'],
@@ -93,6 +86,8 @@ def summarize_text(text, source_language="English", target_language="English"):
     )
     summary = multilingual_summarization_tokenizer.decode(summary_ids[0], skip_special_tokens=True)
     
+    target_lang_code = LANGUAGES[target_language]
+    
     # Translate summary to the target language if needed
     if target_lang_code != "en_XX":
         summary = translate_text(summary, "en_XX", target_lang_code)
@@ -102,13 +97,12 @@ def summarize_text(text, source_language="English", target_language="English"):
 # Streamlit interface
 st.title("Multi-Language Text Summarization Tool")
 
-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 = st.text_area("Input Text (in English)")
+target_language = st.selectbox("Target Language for Summary", options=list(LANGUAGES.keys()), index=list(LANGUAGES.keys()).index("English"))
 
 if st.button("Summarize"):
     if text:
-        summary = summarize_text(text, source_language, target_language)
+        summary = summarize_text(text, target_language)
         st.subheader("Summary")
         st.write(summary)
     else: