Update app.py

app.py

@@ -19,10 +19,10 @@ st.subheader("Understand your emotions better with AI-powered predictions!")
 def load_emotion_model():
     try:
         st.info("⏳ Loading the emotion analysis model, please wait...")
-        # Using a publicly available model
+        # Using a publicly available model for emotion analysis
         emotion_analyzer = pipeline(
             "text-classification",
-            model="bhadresh-savani/distilbert-base-uncased-emotion",
+            model="bhadresh-savani/distilbert-base-uncased-emotion",  # A valid public model
             device=0 if torch.cuda.is_available() else -1,  # Use GPU if available
         )
         st.success("✅ Model loaded successfully!")
@@ -37,13 +37,15 @@ emotion_analyzer = load_emotion_model()
 # ---- Function for Predicting Emotion ----
 def predict_emotion(text):
     if emotion_analyzer is None:
-        return {"Error": "Emotion analyzer model not initialized. Please reload the app."}
+        st.error("⚠️ Model not loaded. Please reload the app.")
+        return {"Error": "Emotion analyzer model not initialized. Please try again later."}
     
     try:
         # Analyze emotions
         result = emotion_analyzer([text])
         return {res["label"]: round(res["score"], 4) for res in result}
     except Exception as e:
+        st.error(f"⚠️ Prediction failed: {e}")
         return {"Error": f"Prediction failed: {e}"}
 
 # ---- User Input Section ----
@@ -88,3 +90,11 @@ st.markdown(
     Designed for a fun and intuitive experience! 🌟  
     """
 )
+
+# ---- Error Handling and User Suggestions ----
+if emotion_analyzer is None:
+    st.error("⚠️ We couldn't load the emotion analysis model. Please check your internet connection or try again later.")
+    st.markdown("🔧 **Troubleshooting Steps:**")
+    st.markdown("1. Ensure you have a stable internet connection.")
+    st.markdown("2. If the issue persists, please refresh the page and try again.")
+    st.markdown("3. Check if the model has been updated or is temporarily unavailable.")