Update app.py

app.py

@@ -12,7 +12,7 @@ import io
 client = Groq(api_key=st.secrets["GROQ_API_KEY"])
 
 # Configure Streamlit
-st.set_page_config(page_title="🤣 Roast Master Pro", layout="wide", page_icon="🤖")
+st.set_page_config(page_title="🤣 Roast Master Pro", layout="center", page_icon="🤖")
 
 # Custom CSS
 st.markdown("""
@@ -63,10 +63,11 @@ def analyze_personality(responses):
     }
     
     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))
+        if response:  # Handle empty 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)
@@ -74,59 +75,66 @@ def analyze_personality(responses):
 
 def generate_personality_report(analysis):
     report = {
-        'type': "",
-        'strengths': [],
-        'tips': [],
-        'quote': ""
+        'type': "The Balanced Visionary 🦉",
+        'strengths': ["Adaptability", "Critical thinking"],
+        'tips': [
+            "🎯 Practice mindful decision-making",
+            "🧘♂️ Balance work and relaxation",
+            "📚 Learn something new daily",
+            "🤝 Cultivate meaningful relationships",
+            "🌱 Embrace gradual improvement",
+            "🎉 Celebrate small victories",
+            "🧠 Challenge cognitive distortions",
+            "💡 Reframe negative thoughts",
+            "🔄 Embrace flexible thinking",
+            "🌟 Believe in growth potential"
+        ],
+        'quote': "The only limit to your growth is your imagination!"
     }
     
-    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
+    try:
+        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"
+            ]
 
-def generate_roast(prompt):
-    response = client.chat.completions.create(
-        model="mixtral-8x7b-32768",
-        messages=[
-            {"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.95
-    )
-    return response.choices[0].message.content
+        # Generate inspirational quote
+        quote_response = client.chat.completions.create(
+            model="mixtral-8x7b-32768",
+            messages=[{"role": "user", "content": f"Generate inspirational quote about {report['type']}"}]
+        )
+        report['quote'] = quote_response.choices[0].message.content
+        
+    except Exception as e:
+        st.error(f"Error generating report: {str(e)}")
+    
+    return report
 
 def create_pdf_report(report):
     buffer = io.BytesIO()
@@ -191,10 +199,10 @@ if not st.session_state.show_report:
             if st.button("Next ➡️"):
                 st.session_state.responses.append(user_input)
                 st.session_state.current_q += 1
-                st.experimental_rerun()
+                st.rerun()
     else:
         st.session_state.show_report = True
-        st.experimental_rerun()
+        st.rerun()
 else:
     # Personality analysis
     analysis = analyze_personality(st.session_state.responses)
@@ -238,7 +246,7 @@ else:
     
     if st.button("🔄 Start Over"):
         st.session_state.clear()
-        st.experimental_rerun()
+        st.rerun()
 
 # Sidebar
 with st.sidebar:
@@ -256,4 +264,3 @@ with st.sidebar:
     - 📊 Personality assessment
     - 📥 PDF report download
     """)
-    st.image("https://i.imgur.com/5X7nKNy.png", width=250)