Update app.py

app.py

@@ -3,7 +3,8 @@ import json
 import requests
 import time
 from newspaper import Article
-
+import nltk
+nltk.download('punkt')
 
 # Page title layout
 c1, c2 = st.columns([0.32, 2])
@@ -13,13 +14,13 @@ with c1:
 
 with c2:
     st.title("Website Article Summarize")
-st.markdown("**Generate summaries of articles from websites using abstractive summarization with Google's Pegasus language model.**")
+st.markdown("**Generate summaries of articles from websites using abstractive summarization with Language Model and Library NewsPaper.**")
 st.caption("Created by Bayhaqy.")
 
 # Sidebar content
 st.sidebar.subheader("About the app")
 st.sidebar.info("This app uses optional 🤗HuggingFace's Model [facebook/bart-large-cnn](https://huggingface.co/facebook/bart-large-cnn) \
-or [pegasus_indonesian_base-finetune](https://huggingface.co/pegasus_indonesian_base-finetune) model.")
+or [pegasus_indonesian_base-finetune](https://huggingface.co/pegasus_indonesian_base-finetune) model and Library NewsPaper.")
 st.sidebar.write("\n\n")
 st.sidebar.markdown("**Get a free API key from HuggingFace:**")
 st.sidebar.markdown("* Create a [free account](https://huggingface.co/join) or [login](https://huggingface.co/login)")
@@ -27,7 +28,7 @@ st.sidebar.markdown("* Go to **Settings** and then **Access Tokens**")
 st.sidebar.markdown("* Create a new Token (select 'read' role)")
 st.sidebar.markdown("* Paste your API key in the text box")
 st.sidebar.divider()
-st.sidebar.write("Please make sure your article is in English/Indonesia, choose the correct model and is not behind a paywall.")
+st.sidebar.write("Please make sure you choose the correct model and is not behind a paywall.")
 st.sidebar.write("\n\n")
 st.sidebar.divider()
 
@@ -55,28 +56,18 @@ if fetch_button:
                 time.sleep(3)
                 st.success('Your article is ready for summarization!')     
 
-                view_button = st.button("View Article")
+                article = Article(url)
+                article.download()
+                article.parse()
+                
+                title = article.title
+                text = article.text
                 
-                if view_button:
-                    article = Article(url)
-                    article.download()
-                    article.parse()
-                    
-                    title = article.title
-                    text = article.text
-                    html = article.html
-                    
-                    st.divider()
-                    st.subheader("Real Article")
-                    st.markdown(f"Your article: **{title}**")
-                    st.markdown(f"**{text}**")
-                    st.divider()
-            
-                    st.divider()
-                    st.subheader("Real Article (HTML)")
-                    st.code(f"Your article: **{title}**",language='html')
-                    st.code(f"**{html}**",language='html')
-                    st.divider()
+                st.divider()
+                st.subheader("Real Article")
+                st.markdown(f"Your article: **{title}**")
+                st.markdown(f"**{text}**")
+                st.divider()
 
         else:
             st.write("Error occurred while fetching article.")
@@ -99,9 +90,6 @@ if selected_api_url == "bart-large-cnn":
 else:
     API_URL = "https://api-inference.huggingface.co/models/thonyyy/pegasus_indonesian_base-finetune"
 
-
-API_URL_TL = "https://api-inference.huggingface.co/models/facebook/mbart-large-50-many-to-many-mmt"
-
 submit_button = st.button("Submit to Summarize")
 
 # Download and parse the article
@@ -109,10 +97,12 @@ if submit_button:
     article = Article(url)
     article.download()
     article.parse()
+    article.nlp()
     
     title = article.title
     text = article.text
     html = article.html
+    summ = article.summary
 
     # HuggingFace API request function summary
     def query_sum(payload):
@@ -134,19 +124,11 @@ if submit_button:
         st.markdown(f"**{summary}**")
 
         st.divider()
-        st.subheader("Real Article")
+        st.subheader("Summary Library NewsPaper")
         st.markdown(f"Your article: **{title}**")
-        st.markdown(f"**{text}**")
-        st.divider()
+        st.markdown(f"**{summ}**")
 
         st.divider()
         st.subheader("Real Article")
         st.markdown(f"Your article: **{title}**")
         st.markdown(f"**{text}**")
-        st.divider()
-
-        st.divider()
-        st.subheader("Real Article (HTML)")
-        st.code(f"Your article: **{title}**",language='html')
-        st.code(f"**{html}**",language='html')
-        st.divider()