Update app.py

app.py

@@ -1,9 +1,62 @@
 import streamlit as st
 from transformers import pipeline
 
-# Set up the app title and description
-st.title("Personality Predictor")
-st.write("Enter some text about yourself or someone else, and this app will analyze it to predict personality-related sentiment.")
+# Set up the app title and description with custom styling
+st.set_page_config(page_title="Personality Predictor", page_icon=":guardsman:", layout="wide")
+
+# Apply custom CSS styles
+st.markdown("""
+    <style>
+        .title {
+            font-size: 36px;
+            font-weight: bold;
+            color: #2C3E50;
+            text-align: center;
+        }
+        .description {
+            font-size: 20px;
+            color: #34495E;
+            text-align: center;
+            margin-bottom: 20px;
+        }
+        .input-area {
+            font-size: 18px;
+            padding: 15px;
+            border-radius: 8px;
+            border: 1px solid #BDC3C7;
+            background-color: #ECF0F1;
+        }
+        .btn {
+            background-color: #2980B9;
+            color: white;
+            font-size: 18px;
+            font-weight: bold;
+            border-radius: 8px;
+            padding: 10px 20px;
+            cursor: pointer;
+        }
+        .btn:hover {
+            background-color: #3498DB;
+        }
+        .result-card {
+            padding: 20px;
+            background-color: #F5F5F5;
+            border-radius: 8px;
+            border: 1px solid #BDC3C7;
+            margin-top: 30px;
+        }
+        .footer {
+            text-align: center;
+            margin-top: 40px;
+            color: #7F8C8D;
+            font-size: 14px;
+        }
+    </style>
+""", unsafe_allow_html=True)
+
+# Set up the app's title and description
+st.markdown('<div class="title">Personality Predictor</div>', unsafe_allow_html=True)
+st.markdown('<div class="description">Enter text to predict personality-related traits. This app analyzes text and provides insights into the sentiment.</div>', unsafe_allow_html=True)
 
 # Load the Hugging Face pre-trained model
 @st.cache_resource
@@ -13,20 +66,21 @@ def load_model():
 # Initialize the model
 classifier = load_model()
 
-# Text input area
-user_input = st.text_area("Write something here:", placeholder="E.g., I enjoy social gatherings and meeting new people.")
+# Text input area with custom styling
+user_input = st.text_area("Write something here:", placeholder="E.g., I enjoy social gatherings and meeting new people.", height=200, key="input_area", help="Type a sentence or paragraph describing a personality trait.", max_chars=500)
 
-# Predict personality traits
-if st.button("Predict Personality"):
+# Action button with custom style
+if st.button("Predict Personality", key="predict_button", help="Click to predict the personality traits based on your input"):
     if user_input.strip():
         with st.spinner("Analyzing text..."):
             results = classifier(user_input)
+            st.markdown('<div class="result-card">', unsafe_allow_html=True)
             st.subheader("Analysis Results")
             for result in results:
                 st.write(f"**Label:** {result['label']}, **Score:** {result['score']:.2f}")
+            st.markdown('</div>', unsafe_allow_html=True)
     else:
-        st.warning("Please enter some text to predict personality.")
+        st.warning("Please enter some text to predict personality.", icon="⚠️")
 
 # Footer
-st.write("---")
-st.write("This app uses a sentiment analysis model from Hugging Face to simulate personality prediction.")
+st.markdown('<div class="footer">This app uses sentiment analysis to simulate personality prediction. Developed with ❤️ for learning purposes.</div>', unsafe_allow_html=True)