Update app.py

app.py

@@ -1,31 +1,23 @@
 import streamlit as st
 from transformers import pipeline
 import torch
+import time
 
-# Function to load the model with error handling and ensure it's available
+# Function to load the model with error handling and retries
 @st.cache_resource
 def load_model():
     try:
-        # Attempt to load a model
         st.write("Attempting to load the emotion analysis model...")
-        emotion_analyzer = pipeline("text-classification", model="bhadresh-savani/bert-base-uncased-emotion", device=0 if torch.cuda.is_available() else -1)
+        # Using a smaller model for quick load times
+        emotion_analyzer = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta", device=0 if torch.cuda.is_available() else -1)
         st.write("Model loaded successfully!")
         return emotion_analyzer
     except Exception as e:
         st.write(f"Error loading the model: {e}")
         return None
 
-# Initialize the model (with caching to prevent reloads)
-emotion_analyzer = load_model()
-
-# Check if the model is loaded
-if emotion_analyzer is None:
-    st.warning("The emotion analysis model could not be loaded. Please try again.")
-else:
-    st.success("Emotion model is ready for predictions!")
-
 # Function to predict emotion for a single response
-def predict_emotion_single(response):
+def predict_emotion_single(response, emotion_analyzer):
     if emotion_analyzer is None:
         st.error("Model not loaded. Please try reloading the app.")
         return {"Error": "Emotion analyzer model not initialized. Please check model loading."}
@@ -54,11 +46,27 @@ questions = [
 # Initialize a dictionary to store responses
 responses = {}
 
-# Ask each question and get response
+# Initialize the emotion analysis model with retries
+emotion_analyzer = None
+max_retries = 3
+retry_delay = 5  # seconds
+
+# Try loading the model with retries
+for attempt in range(max_retries):
+    emotion_analyzer = load_model()
+    if emotion_analyzer:
+        break
+    if attempt < max_retries - 1:
+        st.warning(f"Retrying model load... Attempt {attempt + 2}/{max_retries}")
+        time.sleep(retry_delay)
+    else:
+        st.error("Model failed to load after multiple attempts. Please try again later.")
+
+# Function to handle responses and emotion analysis
 for i, question in enumerate(questions, start=1):
     user_response = st.text_input(f"Question {i}: {question}")
     if user_response:
-        analysis = predict_emotion_single(user_response)
+        analysis = predict_emotion_single(user_response, emotion_analyzer)
         responses[question] = (user_response, analysis)
         st.write(f"**Your Response**: {user_response}")
         st.write(f"**Emotion Analysis**: {analysis}")
@@ -71,9 +79,4 @@ if st.button("Clear Responses"):
 if st.button("Submit Responses"):
     if responses:
         st.write("-- Emotion Analysis Results ---")
-        for i, (question, (response, analysis)) in enumerate(responses.items(), start=1):
-            st.write(f"\n**Question {i}:** {question}")
-            st.write(f"Your Response: {response}")
-            st.write(f"Emotion Analysis: {analysis}")
-    else:
-        st.write("Please answer all the questions before submitting.")
+        for i, (question, (response