Update app.py

app.py

@@ -1,83 +1,135 @@
+# app.py
 import streamlit as st
+from groq import Groq
 from textblob import TextBlob
-# The transformers import is here in case you want to integrate a pre-trained model later.
-# from transformers import pipeline
+import re
+import time
 
-def get_roast(user_text):
-    """
-    Generates a roast based on the user's input.
-    (You can replace or extend this logic by integrating a model like Mixtral-8x7B.)
-    """
-    if "procrastinate" in user_text.lower():
-        return "Ah, the art of doing nothing! Do you charge Netflix for your couch imprint? 🛋️"
-    elif "always" in user_text.lower() or "never" in user_text.lower():
-        return "Wow, painting your world in extremes? Maybe it's time to add some shades of gray!"
-    else:
-        return "Is that a problem or a lifestyle choice? Time to get serious... or maybe not."
+# Initialize Groq client
+client = Groq(api_key=st.secrets["GROQ_API_KEY"])
 
-def analyze_text(user_text):
-    """
-    Analyzes the input text for sentiment and detects basic cognitive distortions.
-    For now, it counts occurrences of words like 'always' or 'never'.
-    """
-    blob = TextBlob(user_text)
-    sentiment = blob.sentiment.polarity  # Ranges from -1 (negative) to 1 (positive)
-    distortions = sum(word in user_text.lower() for word in ["always", "never"])
-    return sentiment, distortions
+# Configure Streamlit
+st.set_page_config(page_title="🤣 Roast Master Therapist", layout="wide")
 
-def calculate_resilience_score(sentiment, distortions):
-    """
-    Calculates a resilience score based on sentiment and the number of detected distortions.
-    The score is capped between 0 and 100.
-    """
-    score = 100
-    # Adjust score by sentiment (scaled)
-    score += int(sentiment * 20)
-    # Penalize for cognitive distortions
-    score -= distortions * 10
-    # Ensure score stays within bounds
-    score = max(0, min(score, 100))
-    return score
+# Custom CSS for animations
+st.markdown("""
+<style>
+@keyframes laugh {
+  0% { transform: scale(1); }
+  50% { transform: scale(1.2); }
+  100% { transform: scale(1); }
+}
 
-def get_reframe_tips(score):
-    """
-    Provides reframe tips based on the resilience score.
-    """
-    if score < 50:
-        return "Remember: small steps lead to big changes. Try breaking tasks into manageable chunks and celebrate every little victory!"
-    elif score < 75:
-        return "You're on your way! Consider setting specific goals and challenge those negative thoughts with evidence."
-    else:
-        return "Keep up the great work! Your resilience is inspiring – maybe share some of that energy with someone who needs it!"
+.laugh-emoji {
+  animation: laugh 0.5s ease-in-out infinite;
+  font-size: 3em;
+  text-align: center;
+}
 
-def main():
-    st.title("🤖 Roast Master Therapist Bot")
-    st.write("Share your problem, and let the roast and resilience tips begin!")
+.progress-bar {
+  height: 20px;
+  border-radius: 10px;
+  background: #ff4b4b;
+  transition: width 0.5s ease-in-out;
+}
+</style>
+""", unsafe_allow_html=True)
+
+def analyze_cbt(text):
+    """Analyze text for cognitive distortions using CBT principles"""
+    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
+    }
     
-    # User input
-    user_input = st.text_area("What's troubling you?", placeholder="e.g., I procrastinate on everything...")
+    # Deduct points for cognitive distortions
+    analysis['resilience_score'] -= analysis['absolutes'] * 5
+    analysis['resilience_score'] = max(analysis['resilience_score'], 0)
     
-    if st.button("Get Roast and Tips"):
-        if user_input.strip():
-            # Generate roast response
-            roast = get_roast(user_input)
-            st.markdown("### Roast:")
-            st.write(roast)
-            
-            # Analyze user input
-            sentiment, distortions = analyze_text(user_input)
-            resilience_score = calculate_resilience_score(sentiment, distortions)
-            tips = get_reframe_tips(resilience_score)
-            
-            # Display analysis and tips
-            st.markdown("### Resilience Analysis:")
-            st.write(f"**Resilience Score:** {resilience_score}/100")
-            st.write(tips)
+    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? 🛋️" """
+    
+    response = client.chat.completions.create(
+        model="mixtral-8x7b-32768",
+        messages=[
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": prompt}
+        ],
+        temperature=0.9,
+        max_tokens=100
+    )
+    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']
+        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)
+        
+        # Display CBT tips
+        st.subheader("💡 Growth Tips")
+        for tip in get_cbt_tips(analysis):
+            st.markdown(f"- {tip}")
             
-            # Fun Streamlit animation!
-            st.balloons()
-        else:
-            st.warning("Please share something so we can get roasting!")
+        # Add disclaimer
+        st.divider()
+        st.caption("Disclaimer: This is not real therapy - it's therapy with jokes! Always consult a professional for serious concerns.")
 
-if __name__ == '__main__':
-    main()
+# Sidebar information
+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
+    
+    **Psychological Basis:**
+    - Cognitive Behavioral Therapy (CBT)
+    - Humor as emotional distancing
+    - Positive reframing techniques
+    """)
+    st.image("https://i.imgur.com/7Q4X4yN.png", width=200)