Update UI

app.py

@@ -112,7 +112,7 @@ def categorize(url):
 
         
 # Main App
-st.header('Classification Project')
+st.title('Classification Project')
 st.write("Unsure what category a CNN article belongs to? Our clever tool can help! Paste the URL below and press Enter. We'll sort it into one of our 5 categories in a flash! ⚡️")
 
 # Define category information (modify content and bullet points as needed)
@@ -143,7 +143,16 @@ categories = {
         "Stay updated on celebrity news and cultural trends."
     ]
 }
-# Create expanders containing list of category can be classified
+
+# Define model information (modify descriptions as needed)
+models = {
+  "Logistic Regression": "A widely used statistical method for classification problems. It excels at identifying linear relationships between features and the target variable.",
+  "SVC (Support Vector Classifier)": "A powerful machine learning model that seeks to find a hyperplane that best separates data points of different classes. It's effective for high-dimensional data and can handle some non-linear relationships.",
+  "LSTM (Long Short-Term Memory)": "A type of recurrent neural network (RNN) particularly well-suited for sequential data like text or time series. LSTMs can effectively capture long-term dependencies within the data.",
+  "BERT (Bidirectional Encoder Representations from Transformers)": "A powerful pre-trained model based on the Transformer architecture. It excels at understanding the nuances of language and can be fine-tuned for various NLP tasks like text classification."
+}
+
+# Create expanders containing list of categories can be classified
 with st.expander("Category List"):
   # Title for each category
   st.subheader("Available Categories:")
@@ -156,6 +165,17 @@ with st.expander("Category List"):
     for item in content:
       st.write(f"- {item}")
 
+
+# Create expanders containing list of models used in this project
+with st.expander("Available Models"):
+  st.subheader("List of Models:")
+  for model_name in models.keys():
+    st.write(f"- {model_name}")
+  st.write("---")
+  for model_name, description in models.items():
+    st.subheader(model_name)
+    st.write(description)
+        
 # Explain to user why this project is only worked for CNN domain
 with st.expander("Tips", expanded=True):
     st.write(
@@ -172,8 +192,10 @@ if url:
     st.divider() # 👈 Draws a horizontal rule
     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.title('Article Content Fetched')
+    st.text_area("", value=article_content, height=400) # render the article content as textarea element
     st.divider()  # 👈 Draws a horizontal rule
+    st.title('Predicted Results')
     st.json({
         "Logistic": result.get("Logistic_Predicted"),
         "SVC": result.get("SVM_Predicted")