Update app.py

app.py

@@ -48,9 +48,9 @@ if submit:
     last_time = time.time()
     with st.spinner("Reading Article..."):
         scrap = Scrap(user_input)
-        title, text = scrap.title, scrap.text
 
-    if text:
+    if scrap:
+        title, text = scrap.title, scrap.text
         text = re.sub(r'\n', ' ', text)
 
         with st.spinner("Computing..."):
@@ -80,8 +80,8 @@ if submit:
 
             prediction = np.argmax(result, axis=-1)
             if prediction == 0:
-                st.markdown(f"""<p style="background-color: rgb(254, 242, 242); 
-                color: rgb(153, 27, 27);
+                st.markdown(f"""<p style="background-color: rgb(236, 253, 245); 
+                color: rgb(6, 95, 70);
                 font-size: 20px;
                 border-radius: 7px;
                 padding-left: 12px;
@@ -90,8 +90,8 @@ if submit:
                 line-height: 25px;
                 text-align: center;">This article is <b>{label[prediction]}</b>.</p>""", unsafe_allow_html=True)
             else:
-                st.markdown(f"""<p style="background-color: rgb(236, 253, 245); 
-                color: rgb(6, 95, 70);
+                st.markdown(f"""<p style="background-color: rgb(254, 242, 242); 
+                color: rgb(153, 27, 27);
                 font-size: 20px;
                 border-radius: 7px;
                 padding-left: 12px;
@@ -99,6 +99,7 @@ if submit:
                 padding-bottom: 15px;
                 line-height: 25px;
                 text-align: center;">This article is <b>{label[prediction]}</b>.</p>""", unsafe_allow_html=True)
+                
             
             with st.expander("Related Articles"):
                 for i in sorted[:5]:
@@ -106,4 +107,14 @@ if submit:
                     st.markdown(f"""
                     <small style="text-align: left;">{data["url"][i].split("/")[2]}</small><br>
                     <a href={data["url"][i]} style="text-align: left;">{data["title"][i]}</a>
-                    """, unsafe_allow_html=True)
+                    """, unsafe_allow_html=True)
+    else:
+        st.markdown(f"""<p style="background-color: rgb(254, 242, 242); 
+        color: rgb(153, 27, 27);
+        font-size: 20px;
+        border-radius: 7px;
+        padding-left: 12px;
+        padding-top: 15px;
+        padding-bottom: 15px;
+        line-height: 25px;
+        text-align: center;">Can't scrap article from this link.</p>""", unsafe_allow_html=True)