Update app.py

app.py

@@ -1,141 +1,90 @@
 # app.py
 import streamlit as st
 from groq import Groq
-from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
-import random
-from datetime import datetime
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 import io
+import random
+from datetime import datetime
 
-# Initialize components
+# Initialize Groq client
 try:
     groq_client = Groq(api_key=st.secrets["GROQ_API_KEY"])
 except KeyError:
-    st.error("GROQ_API_KEY missing in secrets!")
+    st.error("GROQ_API_KEY missing in secrets! Add it in Hugging Face settings.")
     st.stop()
 
 # Load personality model
 try:
-    personality_model = AutoModelForSequenceClassification.from_pretrained(
+    model = AutoModelForSequenceClassification.from_pretrained(
         "KevSun/Personality_LM", 
         ignore_mismatched_sizes=True
     )
-    personality_tokenizer = AutoTokenizer.from_pretrained("KevSun/Personality_LM")
+    tokenizer = AutoTokenizer.from_pretrained("KevSun/Personality_LM")
 except Exception as e:
     st.error(f"Model loading error: {str(e)}")
     st.stop()
 
 # Configure Streamlit
-st.set_page_config(page_title="🧠 PsychBot Pro", layout="wide", page_icon="🤖")
+st.set_page_config(page_title="🧠 Mind Mapper Pro", layout="wide", page_icon="🤖")
 
 # Custom CSS
 st.markdown("""
 <style>
-@keyframes float {
-  0% { transform: translateY(0px); }
-  50% { transform: translateY(-20px); }
-  100% { transform: translateY(0px); }
+@keyframes rainbow {
+  0% { color: #ff0000; }
+  20% { color: #ff8000; }
+  40% { color: #ffff00; }
+  60% { color: #00ff00; }
+  80% { color: #0000ff; }
+  100% { color: #ff00ff; }
 }
 .personality-title {
-  animation: float 3s ease-in-out infinite;
+  animation: rainbow 3s infinite;
+  font-size: 2.5em !important;
 }
 .social-post {
-  border: 1px solid #e0e0e0;
+  border: 2px solid #4CAF50;
   border-radius: 15px;
   padding: 20px;
   margin: 15px 0;
-  background: #f8f9fa;
-}
-.platform-selector {
-  display: flex;
-  gap: 10px;
-  margin: 20px 0;
-}
-.response-box {
-  border-left: 3px solid #4CAF50;
-  padding: 10px;
-  margin: 10px 0;
-}
-.question-card {
-  padding: 20px;
-  border-radius: 15px;
-  margin: 10px 0;
-  background: #f8f9fa;
+  background: #f8fff9;
 }
 </style>
 """, unsafe_allow_html=True)
 
-# Enhanced question bank with mix of funny/serious questions
+# Personality traits
+OCEAN_TRAITS = ["agreeableness", "openness", "conscientiousness", "extraversion", "neuroticism"]
+
+# Enhanced question bank with mix of serious and funny questions
 QUESTION_BANK = [
-    {"text": "If you were a kitchen appliance, what would you be? 🔪", "type": "funny", "trait": "openness"},
-    {"text": "How do you handle unexpected changes? 🌀", "type": "serious", "trait": "neuroticism"},
-    {"text": "What's your spirit animal and why? 🦁", "type": "funny", "trait": "agreeableness"},
-    {"text": "Describe your ideal work environment 🖥️", "type": "serious", "trait": "conscientiousness"},
-    {"text": "If your life was a movie title, what would it be? 🎬", "type": "funny", "trait": "openness"},
-    {"text": "How do you approach conflict resolution? ✌️", "type": "serious", "trait": "agreeableness"},
-    {"text": "What superhero power matches your personality? 🦸", "type": "funny", "trait": "extraversion"},
-    {"text": "How do you prioritize tasks? 📅", "type": "serious", "trait": "conscientiousness"}
+    {"text": "If you were a kitchen appliance, which would you be and why? 🥄", "trait": "openness"},
+    {"text": "Describe your ideal Sunday vs your actual last Sunday 📆", "trait": "conscientiousness"},
+    {"text": "How would you survive a zombie apocalypse? 🧟", "trait": "neuroticism"},
+    {"text": "What's your spirit animal during tax season? 🦁", "trait": "agreeableness"},
+    {"text": "Plan a perfect date... with yourself! 💃", "trait": "extraversion"},
+    {"text": "If emotions were weather patterns, what's your forecast? ⛅", "trait": "neuroticism"},
+    {"text": "What would your childhood toy say about you? 🧸", "trait": "openness"},
+    {"text": "Describe your email inbox as a movie title 📧", "trait": "conscientiousness"},
+    {"text": "How do you react when someone says 'We need to talk'? 😬", "trait": "agreeableness"},
+    {"text": "What's your superpower in group projects? 🦸", "trait": "extraversion"}
 ]
 
-def initialize_session():
-    """Initialize all session state variables"""
-    defaults = {
-        'questions': random.sample(QUESTION_BANK, 5),
-        'current_q': 0,
-        'show_post': False,
-        'responses': [],
-        'personality': None
-    }
-    for key, value in defaults.items():
-        if key not in st.session_state:
-            st.session_state[key] = value
-
-def analyze_personality(text):
-    """Predict personality traits using LM"""
-    inputs = personality_tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=512)
+def analyze_psychology(text):
+    """Analyze personality traits using the specified model"""
+    encoded_input = tokenizer(text, return_tensors='pt', padding=True, truncation=True, max_length=64)
     with torch.no_grad():
-        outputs = personality_model(**inputs)
-    probs = torch.nn.functional.softmax(outputs.logits, dim=-1)
-    traits = ["openness", "conscientiousness", "extraversion", "agreeableness", "neuroticism"]
-    return {trait: float(prob) for trait, prob in zip(traits, probs[0])}
-
-
-
+        outputs = model(**encoded_input)
+    predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
+    return {trait: score.item() for trait, score in zip(OCEAN_TRAITS, predictions[0])}
 
 def generate_social_post(platform, traits):
-    """Generate platform-specific social post"""
-    platform_prompts = {
-        "instagram": {
-            "emoji": "📸",
-            "prompt": "Create an Instagram post with 3 emojis and 2 hashtags about personal growth:"
-        },
-        "linkedin": {
-            "emoji": "💼",
-            "prompt": "Create a professional LinkedIn post about self-improvement:"
-        },
-        "facebook": {
-            "emoji": "👥",
-            "prompt": "Create a friendly Facebook post with 2 emojis:"
-        },
-        "whatsapp": {
-            "emoji": "💬",
-            "prompt": "Create a casual WhatsApp status with 2 emojis:"
-        }
-    }
-    
-    # Define the style for each platform separately
-    styles = {
-        'instagram': 'visually appealing with hashtags',
-        'linkedin': 'professional and inspiring',
-        'facebook': 'friendly and engaging',
-        'whatsapp': 'casual and fun'
-    }
-    
-    prompt = f"""{platform_prompts[platform]['prompt']}
-Based on these personality traits: {traits}
-Include {platform_prompts[platform]['emoji']} emoji and make it {styles[platform]}"""
+    """Generate platform-specific social media post"""
+    prompt = f"""Create a {platform} post about personal growth based on these personality scores:
+{traits}
+Include relevant emojis and make it {platform}-appropriate. Keep it under 280 characters."""
     
     response = groq_client.chat.completions.create(
         model="mixtral-8x7b-32768",
@@ -144,30 +93,60 @@ Include {platform_prompts[platform]['emoji']} emoji and make it {styles[platform
     )
     return response.choices[0].message.content
 
-
-
-
-
-
-# Initialize session state
-initialize_session()
+def create_pdf_report(report):
+    """Generate downloadable PDF report"""
+    buffer = io.BytesIO()
+    p = canvas.Canvas(buffer, pagesize=letter)
+    
+    p.setFont("Helvetica-Bold", 16)
+    p.drawString(100, 750, "🌟 Psychological Profile Report 🌟")
+    p.setFont("Helvetica", 12)
+    
+    y_position = 700
+    content = [
+        f"Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M')}",
+        "",
+        "Personality Traits:",
+        *[f"- {trait.upper()}: {score:.2f}" for trait, score in report['traits'].items()],
+        "",
+        "Personalized Quote:",
+        report['quote'],
+    ]
+    
+    for line in content:
+        p.drawString(100, y_position, line)
+        y_position -= 15
+    
+    p.save()
+    buffer.seek(0)
+    return buffer
+
+# Session state management
+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 'selected_questions' not in st.session_state:
+    st.session_state.selected_questions = random.sample(QUESTION_BANK, 5)
 
 # Main UI
-st.title("🧠 PsychBot Pro")
-st.markdown("### Your AI Personality Companion 🤖💬")
+st.title("🧠 Mind Mapper Pro")
+st.markdown("### Your Advanced Psychology Assistant 🔍✨")
+
+# Progress tracker
+progress = st.session_state.current_q / 5
+st.markdown(f"""
+<div style="background: #f0f2f6; border-radius: 15px; padding: 5px;">
+    <div style="width: {progress*100}%; height: 25px; border-radius: 15px; background: linear-gradient(90deg, #FF6B6B 0%, #4ECDC4 100%);"></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 < 5:
+    q = st.session_state.selected_questions[st.session_state.current_q]
     
     with st.chat_message("assistant"):
-        st.markdown(f"""
-        <div class="question-card">
-            <h4>{q['text']}</h4>
-            <p>{'🌟 Fun question!' if q['type'] == 'funny' else '🤔 Serious reflection'}</p>
-        </div>
-        """, unsafe_allow_html=True)
-        
+        st.markdown(f"### {q['text']}")
         user_input = st.text_input("Your response:", key=f"q{st.session_state.current_q}")
         
         if st.button("Next ➡️"):
@@ -175,59 +154,60 @@ if st.session_state.current_q < len(st.session_state.questions):
             st.session_state.current_q += 1
             st.rerun()
 else:
-    # Generate personality report
-    if not st.session_state.personality:
-        combined_text = " ".join(st.session_state.responses)
-        st.session_state.personality = analyze_personality(combined_text)
-    
-    traits = st.session_state.personality
-    st.balloons()
+    # Generate report
+    combined_text = "\n".join(st.session_state.responses)
+    traits = analyze_psychology(combined_text)
     
-    # Personality report section
-    st.markdown(f"## <div class='personality-title'>🌟 Your Personality Report</div>", unsafe_allow_html=True)
+    report = {
+        "traits": traits,
+        "quote": generate_social_post("general", traits)  # Initial quote
+    }
     
-    # Personality visualization
-    cols = st.columns(5)
-    sorted_traits = sorted(traits.items(), key=lambda x: x[1], reverse=True)
-    for i, (trait, score) in enumerate(sorted_traits):
-        cols[i].metric(label=trait.upper(), value=f"{score*100:.0f}%")
+    st.balloons()
+    st.markdown(f"## <div class='personality-title'>🌟 Your Personality Profile 🌟</div>", unsafe_allow_html=True)
     
-    # Social post generation
-    st.markdown("---")
-    st.markdown("### 📱 Social Media Post Generator")
+    # Trait Visualization
+    with st.expander("🧩 Personality Breakdown"):
+        cols = st.columns(5)
+        for i, (trait, score) in enumerate(report['traits'].items()):
+            cols[i].metric(label=trait.upper(), value=f"{score:.2f}")
+
+    # Social Media Post Generation
+    st.markdown("### 📱 Create Social Media Post")
+    platform = st.selectbox("Choose platform:", ["LinkedIn", "Instagram", "Facebook", "WhatsApp"])
     
-    platforms = ["instagram", "linkedin", "facebook", "whatsapp"]
-    selected = st.radio("Choose platform:", platforms, 
-                       format_func=lambda x: f"{x.capitalize()} {'📸💼👥💬'[platforms.index(x)]}", 
-                       horizontal=True)
+    if st.button("Generate Post 🚀"):
+        post = generate_social_post(platform, report['traits'])
+        st.session_state.post = post
     
-    if st.button("Generate Post ✨"):
-        post = generate_social_post(selected, traits)
+    if 'post' in st.session_state:
         st.markdown(f"""
         <div class="social-post">
-            <h4>🎨 {selected.capitalize()} Post Draft</h4>
-            <div class="response-box">{post}</div>
-            <button onclick="navigator.clipboard.writeText(`{post}`)" 
-                    style="margin-top:10px; background: #4CAF50; color: white; border: none; padding: 8px 15px; border-radius: 5px;">
-                📋 Copy Text
-            </button>
+            {st.session_state.post}
         </div>
         """, unsafe_allow_html=True)
-
-    # Restart conversation
-    if st.button("🔄 Start New Analysis"):
-        for key in list(st.session_state.keys()):
-            del st.session_state[key]
-        st.rerun()
+        st.button("📋 Copy Post", on_click=lambda: st.write(st.session_state.post))
+
+    # PDF Report
+    pdf_buffer = create_pdf_report(report)
+    st.download_button(
+        "📥 Download Full Report", 
+        data=pdf_buffer, 
+        file_name="personality_profile.pdf",
+        mime="application/pdf"
+    )
 
 # Sidebar
 with st.sidebar:
-    st.markdown("## 🌈 Features")
+    st.markdown("## 🌈 Features Overview")
     st.markdown("""
-    - 🎭 Dynamic personality assessment
-    - 🤖 AI-generated social posts
-    - 📊 Visual trait analysis
-    - 💬 Mix of fun/serious questions
-    - 📥 PDF report download
-    """)
-    
+    - **Enhanced Personality Assessment**
+    - **Social Media Post Generator**
+    - **Interactive Question Flow**
+    - **PDF Report Download**
+    
+    **New Features:**
+    - 🎭 Fun psychological questions
+    - 📱 Platform-specific post crafting
+    - 🧠 Advanced personality modeling
+    """)