Update app.py

app.py

@@ -4,132 +4,258 @@ from groq import Groq
 from textblob import TextBlob
 import re
 import time
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
+import io
 
 # Initialize Groq client
 client = Groq(api_key=st.secrets["GROQ_API_KEY"])
 
 # Configure Streamlit
-st.set_page_config(page_title="🤣 Roast Master Therapist", layout="wide")
+st.set_page_config(page_title="🤣 Roast Master Pro", layout="wide", page_icon="🤖")
 
-# Custom CSS for animations
+# Custom CSS
 st.markdown("""
 <style>
-@keyframes laugh {
-  0% { transform: scale(1); }
-  50% { transform: scale(1.2); }
-  100% { transform: scale(1); }
+@keyframes float {
+  0% { transform: translateY(0px); }
+  50% { transform: translateY(-20px); }
+  100% { transform: translateY(0px); }
 }
 
-.laugh-emoji {
-  animation: laugh 0.5s ease-in-out infinite;
-  font-size: 3em;
+.emoji-float {
+  animation: float 3s ease-in-out infinite;
+  font-size: 2.5em;
   text-align: center;
 }
 
 .progress-bar {
-  height: 20px;
-  border-radius: 10px;
-  background: #ff4b4b;
-  transition: width 0.5s ease-in-out;
+  height: 25px;
+  border-radius: 15px;
+  background: linear-gradient(90deg, #FF6B6B 0%, #FFE66D 100%);
+  transition: width 0.8s ease-in-out;
+}
+
+.stButton>button {
+  background: #4CAF50 !important;
+  color: white !important;
+  border-radius: 25px !important;
+  padding: 10px 25px !important;
+  font-size: 1.1em !important;
 }
 </style>
 """, unsafe_allow_html=True)
 
-def analyze_cbt(text):
-    """Analyze text for cognitive distortions using CBT principles"""
+# Personality questions
+QUESTIONS = [
+    {"text": "What's your most 'creative' excuse for procrastinating? 🧠💤", "type": "funny"},
+    {"text": "How do you typically react when things don't go as planned? 🌀", "type": "serious"},
+    {"text": "Describe your ideal day... if adulting didn't exist 🏖️🍹", "type": "funny"},
+    {"text": "What's one thing you'd change about your daily habits? 🔄", "type": "serious"}
+]
+
+def analyze_personality(responses):
     analysis = {
-        'absolutes': len(re.findall(r'\b(always|never|everyone|nobody)\b', text, re.I)),
-        'negative_words': TextBlob(text).sentiment.polarity,
-        'resilience_score': 100  # Start with perfect score
+        'humor_score': 0,
+        'resilience': 100,
+        'absolutes': 0,
+        'negativity': 0
     }
     
-    # Deduct points for cognitive distortions
-    analysis['resilience_score'] -= analysis['absolutes'] * 5
-    analysis['resilience_score'] = max(analysis['resilience_score'], 0)
+    for response in responses:
+        blob = TextBlob(response)
+        analysis['humor_score'] += len(re.findall(r'😂|🤣|😜', response)) * 5
+        analysis['negativity'] += blob.sentiment.polarity
+        analysis['absolutes'] += len(re.findall(r'\b(always|never|everyone|nobody)\b', response, re.I))
     
+    analysis['resilience'] -= analysis['absolutes'] * 3
+    analysis['resilience'] = max(min(analysis['resilience'], 100), 0)
     return analysis
 
-def generate_roast(prompt):
-    """Generate a humorous roast using Groq/Mixtral"""
-    system_prompt = """You are a sarcastic therapist who roasts bad habits in a funny way. 
-    Respond with 1-2 short sentences including emojis. Example: "Ah, the art of doing nothing! 
-    Do you charge Netflix for your couch imprint? 🛋️" """
+def generate_personality_report(analysis):
+    report = {
+        'type': "",
+        'strengths': [],
+        'tips': [],
+        'quote': ""
+    }
+    
+    if analysis['humor_score'] > 15:
+        report['type'] = "The Resilient Jester 🃏"
+        report['strengths'] = ["Stress diffuser", "Creative thinker"]
+        report['tips'] = [
+            "🎯 Use humor to reframe challenges",
+            "🧘♂️ Balance jokes with mindfulness",
+            "📅 Schedule 'serious time' daily",
+            "🎭 Channel creativity productively",
+            "🤔 Challenge 'all-or-nothing' thoughts",
+            "📝 Journal funny observations",
+            "🌅 Start day with positive affirmation",
+            "🎉 Reward small achievements",
+            "🧩 Break tasks into fun-sized chunks",
+            "🔄 Replace 'never' with 'sometimes'"
+        ]
+    else:
+        report['type'] = "The Thoughtful Strategist 🧭"
+        report['strengths'] = ["Analytical mind", "Detail-oriented"]
+        report['tips'] = [
+            "😄 Schedule daily laughter breaks",
+            "🌈 Visualize positive outcomes",
+            "🗣️ Practice positive self-talk",
+            "🎯 Set micro-goals daily",
+            "🧘♀️ Try 5-minute mindfulness",
+            "📚 Read humorous content",
+            "🔄 Reframe 'problems' as 'challenges'",
+            "🎁 Reward progress, not perfection",
+            "🤝 Share worries with friends",
+            "✨ Celebrate small wins"
+        ]
+    
+    report['quote'] = client.chat.completions.create(
+        model="mixtral-8x7b-32768",
+        messages=[{"role": "user", "content": f"Generate inspirational quote about {report['type']"}]
+    ).choices[0].message.content
     
+    return report
+
+def generate_roast(prompt):
     response = client.chat.completions.create(
         model="mixtral-8x7b-32768",
         messages=[
-            {"role": "system", "content": system_prompt},
+            {"role": "system", "content": """You are a comedian therapist. Roast bad habits in funny way with emojis. 
+            Example: "Netflix calling you MVP of couch potatoes? 🥔📺"""},
             {"role": "user", "content": prompt}
         ],
-        temperature=0.9,
-        max_tokens=100
+        temperature=0.95
     )
     return response.choices[0].message.content
 
-def get_cbt_tips(analysis):
-    """Generate personalized CBT tips"""
-    tips = []
-    if analysis['absolutes'] > 0:
-        tips.append("🔍 Notice absolute words like 'always/never' - reality is usually more nuanced!")
-    if analysis['negative_words'] < -0.2:
-        tips.append("🌈 Try reframing negative statements: 'I sometimes...' instead of 'I never...'")
-    if analysis['resilience_score'] < 70:
-        tips.append("💪 Practice thought challenging: 'Is this truly catastrophic or just inconvenient?'")
-    return tips
-
-# Main app interface
-st.title("🤣 Roast Master Therapist")
-st.markdown("### Your hilarious path to emotional resilience!")
-
-user_input = st.text_input("Share your problem or bad habit:", 
-                         placeholder="e.g., 'I always procrastinate...'")
-
-if user_input:
-    with st.spinner("Consulting the comedy gods..."):
-        # Show laughing animation
-        st.markdown('<div class="laugh-emoji">😂</div>', unsafe_allow_html=True)
-        time.sleep(1.5)
-        
-        # Generate and display roast
-        roast = generate_roast(user_input)
-        st.subheader("🔥 Your Roast:")
-        st.write(roast)
-        
-        # Analyze and show results
-        analysis = analyze_cbt(user_input)
-        st.divider()
-        
-        # Resilience score visualization
-        st.subheader("🧠 Resilience Analysis")
-        score = analysis['resilience_score']
+def create_pdf_report(report):
+    buffer = io.BytesIO()
+    p = canvas.Canvas(buffer, pagesize=letter)
+    p.setFont("Helvetica-Bold", 16)
+    p.drawString(100, 750, "🌟 Your Personality Report 🌟")
+    
+    y_position = 700
+    p.setFont("Helvetica", 12)
+    for line in [
+        f"Personality Type: {report['type']}",
+        "",
+        "Top Strengths:",
+        *[f"- {strength}" for strength in report['strengths']],
+        "",
+        "Growth Tips:",
+        *[f"{i+1}. {tip}" for i, tip in enumerate(report['tips'])],
+        "",
+        f"Inspirational Quote: {report['quote']}"
+    ]:
+        p.drawString(100, y_position, line)
+        y_position -= 15
+    
+    p.save()
+    buffer.seek(0)
+    return buffer
+
+# Initialize session state
+if 'current_q' not in st.session_state:
+    st.session_state.current_q = 0
+if 'responses' not in st.session_state:
+    st.session_state.responses = []
+if 'show_report' not in st.session_state:
+    st.session_state.show_report = False
+
+# Main UI
+st.title("🤖 Roast Master Pro")
+st.markdown("### The AI That Roasts You Better Than Your Best Friend 🔥")
+
+# Progress bar
+progress = st.session_state.current_q / len(QUESTIONS)
+st.markdown(f"""
+<div style="background: #f0f2f6; border-radius: 15px; padding: 5px; margin: 20px 0;">
+    <div class="progress-bar" style="width: {progress * 100}%;"></div>
+</div>
+""", unsafe_allow_html=True)
+
+# Conversation flow
+if not st.session_state.show_report:
+    if st.session_state.current_q < len(QUESTIONS):
+        q = QUESTIONS[st.session_state.current_q]
+        with st.chat_message("assistant"):
+            st.markdown(f"### {q['text']}")
+            
+            if q['type'] == "funny":
+                st.markdown('<div class="emoji-float">😂 🎭 🤪</div>', unsafe_allow_html=True)
+            else:
+                st.markdown('<div class="emoji-float">🧠 💡 🔍</div>', unsafe_allow_html=True)
+            
+            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
+                st.experimental_rerun()
+    else:
+        st.session_state.show_report = True
+        st.experimental_rerun()
+else:
+    # Personality analysis
+    analysis = analyze_personality(st.session_state.responses)
+    report = generate_personality_report(analysis)
+    
+    # Show report
+    st.balloons()
+    st.success("🎉 Personality Analysis Complete!")
+    
+    col1, col2 = st.columns(2)
+    with col1:
         st.markdown(f"""
-        <div style="background: #f0f2f6; border-radius: 10px; padding: 10px;">
-            <div class="progress-bar" style="width: {score}%;"></div>
-        </div>
-        <br>Current Resilience Score: {score}/100
-        """, unsafe_allow_html=True)
+        ### �� Your Personality Type
+        # {report['type']}
+        """)
+        st.markdown(f"### 💬 Inspirational Quote:")
+        st.info(f'"{report["quote"]}"')
         
-        # Display CBT tips
-        st.subheader("💡 Growth Tips")
-        for tip in get_cbt_tips(analysis):
-            st.markdown(f"- {tip}")
-            
-        # Add disclaimer
-        st.divider()
-        st.caption("Disclaimer: This is not real therapy - it's therapy with jokes! Always consult a professional for serious concerns.")
+    with col2:
+        st.markdown("### 🏆 Top Strengths")
+        for strength in report['strengths']:
+            st.markdown(f"- 🌟 {strength}")
+        
+        st.markdown("### 📈 Resilience Score")
+        st.metric(label="", value=f"{analysis['resilience']}/100")
+    
+    st.markdown("---")
+    st.markdown("### 🛠️ 10 Growth Tips")
+    cols = st.columns(2)
+    for i, tip in enumerate(report['tips']):
+        cols[i%2].markdown(f"{i+1}. {tip}")
+    
+    # PDF Download
+    pdf = create_pdf_report(report)
+    st.download_button(
+        label="📥 Download Full Report",
+        data=pdf,
+        file_name="personality_report.pdf",
+        mime="application/pdf"
+    )
+    
+    if st.button("🔄 Start Over"):
+        st.session_state.clear()
+        st.experimental_rerun()
 
-# Sidebar information
+# Sidebar
 with st.sidebar:
-    st.header("How It Works")
     st.markdown("""
-    1. Share your bad habit/problem
-    2. Get hilariously roasted 🤣
-    3. Receive psychological insights
-    4. Get personalized growth tips
+    ## 🎮 How to Play
+    1. Answer funny/serious questions
+    2. Get roasted 🔥
+    3. Receive personality analysis
+    4. Get 10 growth tips
+    5. Download/share report!
     
-    **Psychological Basis:**
-    - Cognitive Behavioral Therapy (CBT)
-    - Humor as emotional distancing
-    - Positive reframing techniques
+    **Features:**
+    - 🤖 AI-powered insights
+    - 😂 Humor therapy
+    - 📊 Personality assessment
+    - 📥 PDF report download
     """)
-    st.image("https://i.imgur.com/7Q4X4yN.png", width=200)
+    st.image("https://i.imgur.com/5X7nKNy.png", width=250)