Update app.py

app.py

@@ -2,81 +2,138 @@ import streamlit as st
 from transformers import pipeline
 import torch
 import time
+from datetime import datetime
+import random
 
 # Function to load the model with error handling and retries
 @st.cache_resource
 def load_model():
     try:
-        st.write("Attempting to load the emotion analysis model...")
-        # 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!")
+        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!")
         return emotion_analyzer
     except Exception as e:
-        st.write(f"Error loading the model: {e}")
+        st.error(f"Error loading the model: {e}")
         return None
 
 # 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.title("Behavior Prediction App")
-st.write("Enter your thoughts or feelings, and let the app predict your emotional states.")
+st.set_page_config(page_title="Emotion Prediction App 🌈", page_icon="🌺", layout="wide")
 
-# 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?"
-]
+# 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
+)
 
-# Initialize a dictionary to store responses
-responses = {}
+# Sidebar for Navigation
+st.sidebar.title("Navigate 🌴")
+st.sidebar.markdown("Choose an option:")
+options = st.sidebar.radio("", ["Home", "History", "Tips"])
 
 # Initialize the emotion analysis model with retries
-emotion_analyzer = None
-max_retries = 3
-retry_delay = 5  # seconds
+emotion_analyzer = load_model()
 
-# 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 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()
 
-# 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}")
+    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.")
 
-# Provide button to clear input fields
-if st.button("Clear Responses"):
-    st.experimental_rerun()
+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!")
 
-# Display results once all responses are filled
-if st.button("Submit Responses"):
-    if responses:
-        st.write("-- Emotion Analysis Results ---")
-        for i, (question, (response
+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.
+        """
+    )