Update UI

app.py

@@ -111,13 +111,40 @@ def categorize(url):
             return {"error_message": error}
 
 # Main App
-url = st.text_input("enter your CNN's URL here")
+with st.expander("See explanation"): # Explain to user why this project is only worked for CNN domain
+    st.write(
+        '''
+            This project works best with CNN articles right now. 
+            Our web crawler is like a special tool for CNN's website. 
+            It can't quite understand other websites because they're built differently
+        '''
+    )
+
+url = st.text_input("Paste your CNN Article's URL Here")
+
 
 if url:
     result = categorize(url)
     article_content = result.get('Article_Content')
     st.text_area("Article Content", value=article_content, height=400) # render the article content as textarea element
+    st.divider()  # 👈 Draws a horizontal rule
     st.json({
         "Logistic": result.get("Logistic_Predicted"),
         "SVC": result.get("SVM_Predicted")
-    })
+    })
+st.divider()  # 👈 Draws a horizontal rule
+
+# Category labels and corresponding counts
+categories = ["Sport", "Health", "Entertainment", "Politics", "Business"]
+counts = [5638, 4547, 2658, 2461, 1362]
+
+# Create the bar chart
+st.bar_chart(data=dict(zip(categories, counts)))
+
+# Optional: Add a chart title
+st.title("Training Data Category Distribution")
+
+# Optional: Display additional information
+st.write("Here's a breakdown of the number of articles in each category:")
+for category, count in zip(categories, counts):
+  st.write(f"- {category}: {count}")