Update pages/she_melted_mascara.py

pages/she_melted_mascara.py

@@ -29,31 +29,125 @@ CATEGORIES = {
     },
     "🧠 Mental Health": {
         "color": "#DDA0DD",
-        "prompt": "You are a caring friend and art therapist helping with mental health. Use gentle, supportive Gen Z language.",
+        "prompt": """You are a caring friend and art therapist helping with mental health. 
+        Use gentle, supportive Gen Z language. Focus on validation, understanding, 
+        and providing resources when appropriate.""",
         "description": "Safe space for mental health chat & art sharing 💭"
     }
 }
 
+def get_art_analysis(image_data):
+    """Get art analysis from Gemini Vision"""
+    try:
+        # Convert PIL Image to bytes
+        if isinstance(image_data, Image.Image):
+            img_byte_arr = io.BytesIO()
+            image_data.save(img_byte_arr, format='PNG')
+            img_byte_arr = img_byte_arr.getvalue()
+        else:
+            img_byte_arr = image_data
+
+        # Create vision model
+        model = genai.GenerativeModel('gemini-1.5-flash')
+        
+        # Prepare the image for analysis
+        image_parts = [
+            {
+                "mime_type": "image/png",
+                "data": img_byte_arr
+            }
+        ]
+        
+        prompt = """You are an empathetic art therapist analyzing artwork. 
+        Provide a detailed, caring analysis using Gen Z language and emojis.
+        
+        Please analyze:
+        1. 🎨 Colors & Vibes
+        - What emotions do the colors give off?
+        - What's the overall mood?
+        
+        2. 💫 Art Elements
+        - What catches your eye?
+        - What might these elements mean emotionally?
+        
+        3. 💕 Emotional Support
+        - Validate the feelings you see
+        - Share some encouraging words
+        
+        4. 🌟 Growth & Reflection
+        - Ask a gentle question about their feelings
+        - Suggest a supportive activity
+        
+        Use caring, relatable language that teens connect with."""
+
+        # Generate content
+        response = model.generate_content([prompt, image_parts[0]])
+        
+        return response.text
+
+    except Exception as e:
+        print(f"Error in art analysis: {str(e)}")
+        return """I couldn't fully analyze your art bestie, but I'm here to support you! 💕
+                Would you like to tell me more about what you created? I'm all ears! ✨"""
+                
+# Add this function alongside your existing get_art_analysis function
+def get_art_analysis(image_data):
+    """Get art analysis from Gemini Vision"""
+    try:
+        # Convert PIL Image to bytes
+        if isinstance(image_data, Image.Image):
+            img_byte_arr = io.BytesIO()
+            image_data.save(img_byte_arr, format='PNG')
+            img_byte_arr = img_byte_arr.getvalue()
+        else:
+            img_byte_arr = image_data
+
+        # Create vision model
+        model = genai.GenerativeModel('gemini-1.5-flash')
+        
+        # Prepare the image for analysis
+        image_parts = [
+            {
+                "mime_type": "image/png",
+                "data": img_byte_arr
+            }
+        ]
+        
+        prompt = """You are an empathetic art therapist analyzing artwork. 
+        Provide a caring analysis using Gen Z language and emojis.
+        
+        🎨 Art Analysis:
+        - What emotions and mood do you see in this art?
+        - What catches your eye and what might it mean?
+        - Share some supportive and encouraging words
+        
+        Keep your response caring and supportive, using language teens can relate to."""
+
+        # Generate content
+        response = model.generate_content([prompt, image_parts[0]])
+        
+        return response.text
+
+    except Exception as e:
+        print(f"Error in art analysis: {str(e)}")
+        return """I couldn't fully analyze your art bestie, but I'm here to support you! 💕
+                Would you like to tell me more about what you created? I'm all ears! ✨"""
 def get_ai_response(message, category, image=None):
     """Get supportive response from Gemini"""
     try:
         if image:
-            model = genai.GenerativeModel('gemini-pro-vision')
-            prompt = f"""As an empathetic art therapist for teens, analyze this drawing:
-            1. Describe the emotions expressed
-            2. Note any significant elements or symbols
-            3. Provide gentle, supportive feedback
-            4. Ask a caring question about their feelings
-            Use Gen Z friendly language and be very supportive."""
-            response = model.generate_content([prompt, image])
+            # If image is provided, use art analysis instead
+            return get_art_analysis(image)
         else:
-            model = genai.GenerativeModel('gemini-pro')
+            # Regular chat response
+            model = genai.GenerativeModel('gemini-pro')  # Note: Using stable version
             prompt = f"{CATEGORIES[category]['prompt']}\nUser: {message}\nRespond with empathy and support:"
             response = model.generate_content(prompt)
-        return response.text
+            return response.text
     except Exception as e:
+        print(f"Error in get_ai_response: {str(e)}")
         return "I'm here for you bestie! Let's try chatting again? 💕"
-
+    
 def show_page():
     st.title("💧 She Melted Mascara")
     st.write("Your safe space to let it all out! No filter needed here bestie 💕")
@@ -69,8 +163,7 @@ def show_page():
         st.session_state.community_posts = []
     if 'view' not in st.session_state:
         st.session_state.view = 'categories'
-
-    # Layout
+        # Layout
     col1, col2 = st.columns([1, 2])
 
     # Left Column Navigation
@@ -148,6 +241,7 @@ def show_page():
 
                     with tab2:
                         st.write("Express yourself through art 🎨")
+                        st.write("Share your artwork and get supportive analysis ✨")
                         uploaded_file = st.file_uploader(
                             "Upload your drawing",
                             type=['png', 'jpg', 'jpeg']
@@ -155,43 +249,84 @@ def show_page():
                         
                         if uploaded_file:
                             image = Image.open(uploaded_file)
-                            st.image(image, caption="Your artwork")
+                            st.image(image, caption="Your artwork 🎨")
                             
                             share_option = st.radio(
                                 "Would you like to:",
-                                ["Keep private & get AI feedback", "Share with community"],
+                                ["Get private analysis ✨", "Share with community 💕"],
                                 key="art_share_option"
                             )
                             
-                            if st.button("💫 Process Artwork"):
-                                if share_option == "Keep private & get AI feedback":
-                                    # Add to private chat
-                                    st.session_state.chat_history[category].append({
-                                        "role": "user",
-                                        "content": "I made this drawing to express my feelings...",
-                                        "image": image,
-                                        "timestamp": datetime.now().strftime("%I:%M %p")
-                                    })
-                                    
+
+                            if st.button("✨ Analyze My Art"):
+                                with st.spinner("Analyzing your artwork with care and empathy... 💫"):
                                     # Get AI analysis
-                                    response = get_ai_response(None, category, image)
-                                    st.session_state.chat_history[category].append({
-                                        "role": "assistant",
-                                        "content": response,
-                                        "timestamp": datetime.now().strftime("%I:%M %p")
-                                    })
-                                else:
-                                    # Share with community
-                                    st.session_state.community_posts.insert(0, {
-                                        "category": category,
-                                        "content": "Sharing my feelings through art...",
-                                        "image": image,
-                                        "timestamp": datetime.now().strftime("%I:%M %p"),
-                                        "hugs": 0,
-                                        "support": 0,
-                                        "comments": []
-                                    })
-                                st.rerun()
+                                    analysis = get_ai_response(None, category, image)
+                                    
+                                    if share_option == "Get private analysis ✨":
+                                        # Display analysis directly
+                                        st.markdown("### 🎨 Art Analysis")
+                                        st.markdown(analysis)
+                                        
+                                        # Add to chat history
+                                        st.session_state.chat_history[category].append({
+                                            "role": "user",
+                                            "content": "I created this artwork to express my feelings...",
+                                            "image": image,
+                                            "timestamp": datetime.now().strftime("%I:%M %p")
+                                        })
+                                        
+                                        st.session_state.chat_history[category].append({
+                                            "role": "assistant",
+                                            "content": analysis,
+                                            "timestamp": datetime.now().strftime("%I:%M %p")
+                                        })
+                                        
+                                        # Simple follow-up option
+                                        if st.button("Share more about your art? 💫"):
+                                            st.markdown("I'd love to hear more about what inspired this piece! What were you feeling while creating it? 💕")
+                                    
+                                    else:  # Share with community
+                                        # Add to community posts
+                                        st.session_state.community_posts.insert(0, {
+                                            "category": category,
+                                            "content": "Expressing my feelings through art...",
+                                            "image": image,
+                                            "support_message": analysis,
+                                            "timestamp": datetime.now().strftime("%I:%M %p"),
+                                            "hugs": 0,
+                                            "support": 0,
+                                            "comments": []
+                                        })
+                                        st.success("Thank you for sharing your art! The community is here for you 💕")
+                                        
+                                        # Follow-up options
+                                        st.markdown("### Would you like to... 💭")
+                                        col1, col2 = st.columns(2)
+                                        with col1:
+                                            if st.button("Share more about this 💫"):
+                                                followup_msg = "I'm here to listen and understand. Would you like to tell me more about what inspired this artwork? 💕"
+                                                st.session_state.chat_history[category].append({
+                                                    "role": "assistant",
+                                                    "content": followup_msg,
+                                                    "timestamp": datetime.now().strftime("%I:%M %p")
+                                                })
+                                                st.markdown(followup_msg)
+                                        
+                                        with col2:
+                                            if st.button("Get support tips 🌟"):
+                                                support_msg = get_ai_response(
+                                                    "Based on this artwork, what helpful coping strategies would you suggest?",
+                                                    category
+                                                )
+                                                st.session_state.chat_history[category].append({
+                                                    "role": "assistant",
+                                                    "content": support_msg,
+                                                    "timestamp": datetime.now().strftime("%I:%M %p")
+                                                })
+                                                st.markdown(support_msg)
+                                    
+                        
 
                 else:
                     # Regular chat interface for other categories