Update app.py

app.py

@@ -10,18 +10,37 @@ from datetime import datetime
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 import io
+from transformers import AutoModel, AutoTokenizer
+
+
+
+# Initialize Groq client (Ensure API key is available in Streamlit secrets)
+if "GROQ_API_KEY" in st.secrets:
+    groq_client = Groq(api_key=st.secrets["GROQ_API_KEY"])
+else:
+    st.error("GROQ_API_KEY is missing in Streamlit secrets.")
+
+# Emotion Classification Model
+try:
+    personality_classifier = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base")
+except Exception as e:
+    st.error(f"Error loading emotion classifier: {e}")
+
+# Big Five Personality Model
+try:
+    big5_model = pipeline(
+        "text-classification",
+        model="bardsai/bigfive-base",  # Ensure this model exists
+        top_k=5
+    )
+except Exception as e:
+    st.error(f"Error loading Big Five personality model: {e}")
+
+
+
 
 
-# Initialize components
-groq_client = Groq(api_key=st.secrets["GROQ_API_KEY"])
-personality_classifier = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base")
 
-# Verified working Big Five model (no auth required)
-big5_model = pipeline(
-    "text-classification",
-    model="bardsai/bigfive-base",  # Publicly available model
-    top_k=5
-)
 
 # Configure Streamlit
 st.set_page_config(page_title="🧠 Personality Prodigy", layout="wide", page_icon="🤖")