Update app.py

app.py

@@ -19,7 +19,7 @@ def read_article(article):
 # Function to compute sentence similarity based on cosine similarity
 def sentence_similarity(sent1, sent2, stopwords):
     words1 = nltk.word_tokenize(sent1)
-    words2 = nltk.word_tokenize(sent2)
+    words2 = nltk.word_tokenize sent2)
 
     words1 = [word.lower() for word in words1 if word.isalnum()]
     words2 = [word.lower() for word in words2 if word.isalnum()]
@@ -43,7 +43,7 @@ def sentence_similarity(sent1, sent2, stopwords):
 
 # Function to create a similarity matrix of sentences
 def build_similarity_matrix(sentences, stopwords):
-    similarity_matrix = np.zeros((len(sentences), len(sentences)))
+    similarity_matrix = np.zeros((len(sentences), len(sentences))
 
     for i in range(len(sentences)):
         for j in range(len(sentences)):
@@ -76,6 +76,7 @@ def generate_summary(article, top_n=5):
 st.title("Article Summarizer")
 user_article = st.text_area("Enter your article here:")
 translate = st.checkbox("Translate Summary")
+target_language = st.selectbox("Select Target Language", ["English", "French", "Spanish", "German"])
 
 if st.button("Summarize"):
     if user_article:
@@ -84,14 +85,18 @@ if st.button("Summarize"):
         st.write(summary)
         
         if translate:
-            target_language = st.text_input("Enter the target language (e.g., 'fr' for French):")
-            if st.button("Translate"):
-                if target_language:
-                    translator = Translator()
-                    translated_summary = translator.translate(summary, dest=target_language)
-                    st.subheader("Translated Summary:")
-                    st.write(translated_summary.text)
-                else:
-                    st.warning("Please enter a target language code.")
+            if target_language == "English":
+                target_language_code = "en"
+            elif target_language == "French":
+                target_language_code = "fr"
+            elif target_language == "Spanish":
+                target_language_code = "es"
+            elif target_language == "German":
+                target_language_code = "de"
+            
+            translator = Translator()
+            translated_summary = translator.translate(summary, dest=target_language_code)
+            st.subheader("Translated Summary:")
+            st.write(translated_summary.text)
     else:
         st.warning("Please enter an article to summarize.")