Update app.py

app.py

@@ -2,20 +2,22 @@ import streamlit as st
 from transformers import pipeline
 import torch
 import time
-from datetime import datetime
-import random
+
+# App configuration
+st.set_page_config(
+    page_title="Aloha Emotion App 🌴",
+    page_icon="🌺",
+    layout="centered",
+    initial_sidebar_state="expanded"
+)
 
 # Function to load the model with error handling and retries
 @st.cache_resource
 def load_model():
     try:
-        st.write("🌺 Loading the emotion analysis model...")
-        emotion_analyzer = pipeline(
-            "text-classification", 
-            model="j-hartmann/emotion-english-distilroberta", 
-            device=0 if torch.cuda.is_available() else -1
-        )
-        st.success("🤙 Model loaded successfully!")
+        st.info("🌺 Loading the emotion analysis model...")
+        emotion_analyzer = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta", device=0 if torch.cuda.is_available() else -1)
+        st.success("✅ Model loaded successfully!")
         return emotion_analyzer
     except Exception as e:
         st.error(f"Error loading the model: {e}")
@@ -24,116 +26,80 @@ def load_model():
 # Function to predict emotion for a single response
 def predict_emotion_single(response, emotion_analyzer):
     if emotion_analyzer is None:
-        st.error("🌧 Model not loaded. Please try reloading the app.")
+        st.error("❌ Model not loaded. Please try reloading the app.")
         return {"Error": "Emotion analyzer model not initialized. Please check model loading."}
+    
     try:
+        # Analyze the emotion of the response
         result = emotion_analyzer([response])
         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}"}
 
-# Function to get a Hawaiian motivational quote
-def get_hawaiian_quote():
-    quotes = [
-        "Aloha is the way to joy and peace.",
-        "No rain, no rainbows.",
-        "Happiness is a choice, and today is a great day to smile.",
-        "He pōmaikaʻi ke ola – Life is a blessing.",
-    ]
-    return random.choice(quotes)
-
 # Streamlit App Layout
-st.set_page_config(page_title="Emotion Prediction App 🌈", page_icon="🌺", layout="wide")
+st.markdown("<h1 style='text-align: center; color: #2E8B57;'>Aloha Emotion Analyzer 🌺</h1>", unsafe_allow_html=True)
+st.write("**Welcome to the Emotion Analyzer!** Feel the tropical vibes and explore your emotions.")
 
-# Add a Hawaiian-themed background
-st.markdown(
-    """
-    <style>
-    body {
-        background-image: url('https://cdn.pixabay.com/photo/2016/02/18/22/16/palm-trees-1209540_1280.jpg');
-        background-size: cover;
-        color: white;
-    }
-    .stTextInput, .stButton { background-color: rgba(255, 255, 255, 0.8); border-radius: 10px; }
-    </style>
-    """,
-    unsafe_allow_html=True
-)
+# Custom background
+page_bg_img = '''
+<style>
+body {
+    background-color: #F5F5DC;
+    background-image: url("https://www.hawaiimagazine.com/wp-content/uploads/2022/06/aloha-background.jpg");
+    background-size: cover;
+}
+</style>
+'''
+st.markdown(page_bg_img, unsafe_allow_html=True)
 
-# Sidebar for Navigation
-st.sidebar.title("Navigate 🌴")
-st.sidebar.markdown("Choose an option:")
-options = st.sidebar.radio("", ["Home", "History", "Tips"])
+# Define questions for the user
+questions = [
+    "🌞 How are you feeling today?",
+    "🌴 Describe your mood in a few words.",
+    "🌊 What was the most significant emotion you felt this week?",
+    "🌺 How do you handle stress or challenges?",
+    "🌈 What motivates you the most right now?"
+]
+
+# Initialize a dictionary to store responses
+responses = {}
 
 # Initialize the emotion analysis model with retries
-emotion_analyzer = load_model()
+emotion_analyzer = None
+max_retries = 3
+retry_delay = 5  # seconds
 
-if options == "Home":
-    st.title("🌈 Aloha! Welcome to the Behavior Prediction App 🌺")
-    st.write("Enter your thoughts or feelings, and let the app predict your emotional states.")
-    
-    # Define questions for the user
-    questions = [
-        "How are you feeling today?",
-        "Describe your mood in a few words.",
-        "What was the most significant emotion you felt this week?",
-        "How do you handle stress or challenges?",
-        "What motivates you the most right now?",
-    ]
-    
-    # Initialize a dictionary to store responses
-    responses = {}
-    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, emotion_analyzer)
-            responses[question] = (user_response, analysis)
-            st.write(f"**Your Response**: {user_response}")
-            st.write(f"**Emotion Analysis**: {analysis}")
-            
-            # Show motivational quote
-            st.markdown(f"💡 Hawaiian Wisdom: *{get_hawaiian_quote()}*")
-            
-            # Save results (timestamped)
-            with open("emotion_history.txt", "a") as file:
-                file.write(f"{datetime.now()} | {question}: {user_response} | Analysis: {analysis}\n")
-    
-    # Buttons for clearing or submitting responses
-    if st.button("Clear Responses"):
-        st.experimental_rerun()
+# 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.")
 
-    if st.button("Submit Responses"):
-        if responses:
-            st.write("-- 🌟 Final 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.")
+# 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, emotion_analyzer)
+        responses[question] = (user_response, analysis)
+        st.write(f"**Your Response:** {user_response}")
+        st.write(f"**Emotion Analysis:** {analysis}")
 
-elif options == "History":
-    st.title("📜 Your Emotional History")
-    try:
-        with open("emotion_history.txt", "r") as file:
-            history = file.readlines()
-            if history:
-                for line in history:
-                    st.write(line.strip())
-            else:
-                st.write("No history found. Start analyzing your emotions today!")
-    except FileNotFoundError:
-        st.write("No history file found. Start analyzing your emotions today!")
+# Buttons for clearing or submitting responses
+if st.button("Clear Responses"):
+    st.experimental_rerun()
 
-elif options == "Tips":
-    st.title("🌟 Hawaiian Emotional Wellness Tips")
-    st.markdown(
-        """
-        - **Spend Time in Nature**: Hawaii’s natural beauty can help ground you emotionally.
-        - **Practice Aloha**: Show love and respect to yourself and others.
-        - **Embrace ‘Ohana**: Stay connected with family and friends during emotional highs and lows.
-        - **Enjoy Hula or Surfing**: Engaging in cultural or physical activities can boost your mood.
-        - **Live by 'Pono'**: Strive for balance and righteousness in life.
-        """
-    )
+if st.button("Submit Responses"):
+    if responses:
+        st.markdown("<h3 style='color: #2E8B57;'>🌟 Final Emotion Analysis Results:</h3>", unsafe_allow_html=True)
+        for i, (question, (response, analysis)) in enumerate(responses.items(), start=1):
+            st.write(f"**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.")