Update app.py

app.py

@@ -8,7 +8,7 @@ def load_model():
         emotion_classifier = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base")
         return emotion_classifier
     except Exception as e:
-        st.error(f"Error loading model: {str(e)}")
+        st.error(f"Error loading model: {str(e)}")  # Error handling
         return None
 
 emotion_classifier = load_model()
@@ -95,6 +95,24 @@ def get_well_being_suggestions(emotion):
 
 # Streamlit UI
 def main():
+    # Allow user to upload an image for the background
+    uploaded_file = st.file_uploader("Upload a background image", type=["png", "jpg", "jpeg"])
+
+    # Check if an image is uploaded
+    if uploaded_file is not None:
+        st.image(uploaded_file, use_column_width=True)
+    else:
+        # If no image uploaded, use a default URL for a peaceful background
+        st.markdown("""
+        <style>
+        .stApp {
+            background-image: url('https://example.com/your-peaceful-image.jpg');
+            background-size: cover;
+            background-position: center;
+        }
+        </style>
+        """, unsafe_allow_html=True)
+
     # Custom Styling for purple background and white text color
     st.markdown("""
     <style>