Update app.py

app.py

@@ -17,23 +17,25 @@ except KeyError:
     st.error("GROQ_API_KEY missing in secrets! Add it in Hugging Face settings.")
     st.stop()
 
-# Initialize models with verified working versions
+# Initialize psychological models using verified public models
 try:
-    personality_classifier = pipeline(
+    # Personality insights using zero-shot classification
+    personality_analyzer = pipeline(
+        "zero-shot-classification",
+        model="facebook/bart-large-mnli"
+    )
+    
+    # Emotion detection model
+    emotion_classifier = pipeline(
         "text-classification", 
         model="j-hartmann/emotion-english-distilroberta-base"
     )
-    big5_model = pipeline(
-        "text-classification",
-        model="m3hrdadfi/bigfive-personality-traits",
-        top_k=5
-    )
 except Exception as e:
     st.error(f"Model loading error: {str(e)}")
     st.stop()
 
 # Configure Streamlit
-st.set_page_config(page_title="🧠 Personality Prodigy", layout="wide", page_icon="🤖")
+st.set_page_config(page_title="🧠 Mind Mapper", layout="wide", page_icon="🤖")
 
 # Custom CSS
 st.markdown("""
@@ -58,50 +60,38 @@ st.markdown("""
 </style>
 """, unsafe_allow_html=True)
 
-# Dynamic questions pool
+# Personality traits and questions
+OCEAN_TRAITS = {
+    "openness": ["curious", "creative", "adventurous"],
+    "conscientiousness": ["organized", "responsible", "disciplined"],
+    "extraversion": ["outgoing", "energetic", "sociable"],
+    "agreeableness": ["friendly", "compassionate", "cooperative"],
+    "neuroticism": ["sensitive", "nervous", "emotional"]
+}
+
 QUESTION_BANK = [
-    {"text": "What's your most creative procrastination method? 🎨🛌", "trait": "openness"},
-    {"text": "How do you react when plans change suddenly? 🌪️🤔", "trait": "neuroticism"},
-    {"text": "Describe your ideal weekend vs reality 🏝️🛠️", "trait": "extraversion"},
-    {"text": "What's your 'guilty pleasure' productivity hack? 🍫📈", "trait": "conscientiousness"},
-    {"text": "How do you comfort a stressed friend? 🤗💬", "trait": "agreeableness"},
-    {"text": "What would you do if you were invisible for a day? 👻🎭", "trait": "openness"},
-    {"text": "Your reaction to unexpected criticism? 🥊💡", "trait": "neuroticism"},
-    {"text": "Plan a party vs enjoy quiet night? 🎉📚", "trait": "extraversion"},
-    {"text": "How do you handle missed deadlines? ⌛🤷", "trait": "conscientiousness"},
-    {"text": "Describe your perfect collaboration 👥✨", "trait": "agreeableness"}
+    {"text": "How do you typically approach new experiences? 🌍", "trait": "openness"},
+    {"text": "Describe your ideal weekend vs reality 🏝️", "trait": "conscientiousness"},
+    {"text": "How do you recharge after social events? 🪫", "trait": "extraversion"},
+    {"text": "Your reaction to someone disagreeing with you? 💢", "trait": "agreeableness"},
+    {"text": "How do you handle unexpected changes? 🌀", "trait": "neuroticism"}
 ]
 
-def analyze_big5(text):
-    """Process Big Five personality traits from model output"""
-    results = big5_model(text[:512])
-    trait_map = {
-        'EXTRAVERSION': 'extraversion',
-        'NEUROTICISM': 'neuroticism',
-        'CONSCIENTIOUSNESS': 'conscientiousness',
-        'OPENNESS': 'openness',
-        'AGREEABLENESS': 'agreeableness'
-    }
-    return {trait_map[res['label']]: res['score'] for res in results}
-
-def get_dynamic_questions(responses):
-    """Select questions based on personality patterns"""
-    if not responses:
-        return random.sample(QUESTION_BANK, 5)
-    
-    traits = analyze_big5("\n".join(responses))
-    dominant_trait = max(traits, key=traits.get)
-    return [q for q in QUESTION_BANK if q['trait'] == dominant_trait][:3] + random.sample(QUESTION_BANK, 2)
+def analyze_psychology(text):
+    """Analyze personality traits using zero-shot classification"""
+    candidate_labels = list(OCEAN_TRAITS.keys())
+    result = personality_analyzer(text, candidate_labels, multi_label=True)
+    return {label: score for label, score in zip(result['labels'], result['scores'])}
 
 def analyze_emotions(text):
-    """Detect emotional tone using RoBERTa"""
-    return personality_classifier(text[:512])
+    """Detect emotional tone"""
+    return emotion_classifier(text[:512])
 
 def generate_quote(traits):
-    """Generate personality-specific quote using Groq/Mistral"""
+    """Generate personality-specific quote"""
     prompt = f"""Create a motivational quote for someone with these traits:
 {traits}
-Make it inspirational, funny, and include an emoji."""
+Make it inspirational and include an emoji."""
     
     response = groq_client.chat.completions.create(
         model="mixtral-8x7b-32768",
@@ -114,45 +104,26 @@ def generate_tips(traits):
     """Generate evidence-based psychological tips"""
     tips = []
     if traits.get('openness', 0) > 0.7:
-        tips.extend([
-            "🎨 Try creative cross-training (write a poem about your work)", 
-            "🌍 Learn something new daily about different cultures"
-        ])
+        tips.append("🎨 Try creative cross-training: Write a short story about your day")
     if traits.get('neuroticism', 0) > 0.6:
-        tips.extend([
-            "🧘 Practice box breathing: 4s in, 4s hold, 4s out",
-            "📝 Keep a worry journal to externalize anxieties"
-        ])
-    return tips[:10]
-
-def create_personality_report(responses):
-    """Generate comprehensive personality analysis"""
-    text = "\n".join(responses)
-    big5_results = analyze_big5(text)
-    return {
-        "big5": big5_results,
-        "emotions": analyze_emotions(text),
-        "quote": generate_quote(big5_results),
-        "tips": generate_tips(big5_results)
-    }
+        tips.append("🧘 Practice 4-7-8 breathing: Inhale 4s, hold 7s, exhale 8s")
+    return tips[:5]
 
 def create_pdf_report(report):
     """Generate downloadable PDF report"""
     buffer = io.BytesIO()
     p = canvas.Canvas(buffer, pagesize=letter)
     
-    # Header
     p.setFont("Helvetica-Bold", 16)
-    p.drawString(100, 750, "🌟 Personality Prodigy Report 🌟")
+    p.drawString(100, 750, "🌟 Psychological Profile Report 🌟")
     p.setFont("Helvetica", 12)
     
-    # Content
     y_position = 700
     content = [
         f"Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M')}",
         "",
-        "Big Five Personality Traits:",
-        *[f"- {trait.upper()}: {score:.2f}" for trait, score in report['big5'].items()],
+        "Personality Traits:",
+        *[f"- {trait.upper()}: {score:.2f}" for trait, score in report['traits'].items()],
         "",
         "Emotional Analysis:",
         *[f"- {emo['label']}: {emo['score']:.2f}" for emo in report['emotions']],
@@ -162,7 +133,6 @@ def create_pdf_report(report):
         "",
         "Growth Tips:",
         *[f"{i+1}. {tip}" for i, tip in enumerate(report['tips'])]
-    ]  # Fixed missing closing bracket
     
     for line in content:
         p.drawString(100, y_position, line)
@@ -177,78 +147,75 @@ if 'responses' not in st.session_state:
     st.session_state.responses = []
 if 'current_q' not in st.session_state:
     st.session_state.current_q = 0
-if 'questions' not in st.session_state:
-    st.session_state.questions = get_dynamic_questions([])
 
 # Main UI
-st.title("🧠 Personality Prodigy")
-st.markdown("### Your AI-Powered Mirror for Self-Discovery 🔍✨")
+st.title("🧠 Mind Mapper")
+st.markdown("### Your Personal Psychology Assistant 🔍✨")
 
 # Progress tracker
-progress = st.session_state.current_q / len(st.session_state.questions)
+progress = st.session_state.current_q / len(QUESTION_BANK)
 st.markdown(f"""
 <div style="background: #f0f2f6; border-radius: 15px; padding: 5px;">
     <div class="progress-bar" style="width: {progress*100}%; height: 25px;"></div>
 </div>
 """, unsafe_allow_html=True)
 
-# Dynamic question flow
-if st.session_state.current_q < len(st.session_state.questions):
-    q = st.session_state.questions[st.session_state.current_q]
+# Question flow
+if st.session_state.current_q < len(QUESTION_BANK):
+    q = QUESTION_BANK[st.session_state.current_q]
     
     with st.chat_message("assistant"):
         st.markdown(f"### {q['text']}")
-        st.markdown(f"*What's your answer?* {'🤔' if q['trait'] in ['neuroticism','conscientiousness'] else '😄'}")
-        
         user_input = st.text_input("Your response:", key=f"q{st.session_state.current_q}")
         
         if st.button("Next ➡️"):
             st.session_state.responses.append(user_input)
             st.session_state.current_q += 1
-            
-            # Update questions based on responses
-            if st.session_state.current_q == len(st.session_state.questions):
-                st.session_state.questions = get_dynamic_questions(st.session_state.responses)
-                st.session_state.current_q = 0
-                
             st.rerun()
 else:
-    # Generate and display report
-    report = create_personality_report(st.session_state.responses)
-    st.balloons()
+    # Generate report
+    combined_text = "\n".join(st.session_state.responses)
+    traits = analyze_psychology(combined_text)
+    emotions = analyze_emotions(combined_text)
     
-    st.markdown(f"## <div class='personality-title'>🌟 Your Personality Report 🌟</div>", unsafe_allow_html=True)
+    report = {
+        "traits": traits,
+        "emotions": emotions,
+        "quote": generate_quote(traits),
+        "tips": generate_tips(traits)
+    }
+    
+    st.balloons()
+    st.markdown(f"## <div class='personality-title'>🌟 Your Psychological Profile 🌟</div>", unsafe_allow_html=True)
     
-    # Big Five Traits Visualization
-    with st.expander("🔍 Big Five Personality Traits"):
+    # Trait Visualization
+    with st.expander("🧩 Personality Breakdown"):
         cols = st.columns(5)
-        traits = report['big5']
-        for i, (trait, score) in enumerate(traits.items()):
-            cols[i].metric(label=trait.upper(), value=f"{score:.0%}", 
-                           help=f"{['Low','Moderate','High'][int(score//0.33)]} {trait}")
+        for i, (trait, score) in enumerate(report['traits'].items()):
+            cols[i].metric(label=trait.upper(), value=f"{score:.0%}")
     
     # Emotional Analysis
-    with st.expander("🎭 Emotional Tone Analysis"):
+    with st.expander("🎭 Emotional Landscape"):
         emotion = max(report['emotions'], key=lambda x: x['score'])
         st.markdown(f"**Dominant Emotion**: {emotion['label']} ({emotion['score']:.0%})")
         st.progress(emotion['score'])
     
-    # Personalized Content
+    # Recommendations
     col1, col2 = st.columns(2)
     with col1:
-        st.markdown("### 💬 Your Personality Quote")
+        st.markdown("### 💬 Personalized Wisdom")
         st.success(f'"{report["quote"]}"')
     with col2:
-        st.markdown("### 🛠️ Growth Strategies")
-        for tip in report['tips'][:5]:
+        st.markdown("### 🛠️ Actionable Tips")
+        for tip in report['tips']:
             st.markdown(f"- {tip}")
     
-    # Downloadable Report
+    # PDF Report
     pdf_buffer = create_pdf_report(report)
     st.download_button(
         "📥 Download Full Report", 
         data=pdf_buffer, 
-        file_name="personality_report.pdf",
+        file_name="psych_profile.pdf",
         mime="application/pdf"
     )
 
@@ -256,13 +223,13 @@ else:
 with st.sidebar:
     st.markdown("## 🌈 How It Works")
     st.markdown("""
-    1. Answer dynamic personality questions
-    2. Get real-time psychological analysis
-    3. Receive personalized growth strategies
-    4. Download/share your unique report
+    1. Answer 5 psychology-based questions
+    2. Get instant personality analysis
+    3. Receive emotional insights
+    4. Download personalized report
     
-    **Scientific Backing:**
-    - Big Five Personality Model (OCEAN)
-    - Emotion Recognition (RoBERTa)
-    - Cognitive Behavioral Therapy (CBT)
+    **Science Behind It:**
+    - Zero-shot personality classification
+    - Emotion recognition (RoBERTa)
+    - Cognitive behavioral therapy principles
     """)