Update app.py

app.py

@@ -5,32 +5,72 @@ st.set_page_config(page_title="Mental Wellness Diary Analyzer", page_icon="🧠"
 st.title("🧠 Mental Wellness Diary Analyzer")
 
 # Load a lightweight language model
-generator = pipeline("text-generation", model="distilgpt2", max_new_tokens=200)
+@st.cache_resource
+def load_model():
+    return pipeline("text-generation", model="distilgpt2", max_new_tokens=200)
+
+generator = load_model()
 
 def analyze_journal(entry):
     prompt = f"""
 Analyze the following journal entry:
 "{entry}"
-
 Determine the following:
-1. Emotional tone of the journal.
-2. Any recurring themes or sentiments.
-3. Personalized motivational advice.
-
-Be friendly and encouraging in your tone.
+1. Emotional tone: [Identify the primary emotions]
+2. Recurring themes: [Identify key themes]
+3. Advice: [Provide encouraging advice]
 """
     output = generator(prompt)[0]["generated_text"]
-    return output.split("Determine the following:")[-1].strip()
+    
+    # Extract the analysis part
+    analysis = output.split("Analyze the following journal entry:")[-1].strip()
+    
+    # Format the output in a cleaner way without JSON
+    formatted_analysis = ""
+    if "Emotional tone:" in analysis:
+        emotional_tone = analysis.split("Emotional tone:")[1].split("Recurring themes:")[0].strip()
+        formatted_analysis += f"😊 **Emotional tone:** {emotional_tone}\n\n"
+    
+    if "Recurring themes:" in analysis:
+        themes = analysis.split("Recurring themes:")[1].split("Advice:")[0].strip()
+        formatted_analysis += f"🔍 **Recurring themes:** {themes}\n\n"
+    
+    if "Advice:" in analysis:
+        advice = analysis.split("Advice:")[1].strip()
+        formatted_analysis += f"💡 **Advice:** {advice}"
+    
+    return formatted_analysis if formatted_analysis else analysis
 
-# User input
-journal_entry = st.text_area("✍️ Write your journal entry here:")
+# User interface
+st.markdown("### Share what's on your mind today")
+journal_entry = st.text_area("✍️ Write your journal entry here:", height=150)
 
-if st.button("🧠 Analyze"):
+col1, col2, col3 = st.columns([1, 1, 1])
+with col2:
+    analyze_button = st.button("🧠 Analyze My Entry", use_container_width=True)
+
+if analyze_button:
     if journal_entry.strip():
-        st.info("Analyzing your entry with AI...")
-        result = analyze_journal(journal_entry)
-        st.markdown("### 💬 Analysis Result")
-        st.write(result)
+        with st.spinner("Analyzing your thoughts..."):
+            result = analyze_journal(journal_entry)
+            
+        st.markdown("### 💭 Your Wellness Insights")
+        st.markdown(result)
+        
+        # Add some supportive message
+        st.markdown("---")
+        st.markdown("💪 **Remember**: Writing about your feelings is already a positive step for your mental wellness!")
     else:
-        st.warning("Please enter your journal entry first.")
+        st.warning("Please share your thoughts in the journal entry box first.")
+
+# Add a sample entry option
+with st.expander("Need inspiration? Try a sample entry"):
+    if st.button("Use sample entry"):
+        sample_text = "I feel really stressed and anxious about my work. I haven't been sleeping well, and I'm worried about meeting deadlines."
+        st.session_state["sample_text"] = sample_text
+        st.experimental_rerun()
 
+# Use sample if available in session state
+if "sample_text" in st.session_state:
+    journal_entry = st.session_state["sample_text"]
+    del st.session_state["sample_text"]