Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from transformers import pipeline
 import random
 from datetime import datetime
+import logging
 
 # Initialize models with smaller, faster alternatives
 sentiment_analyzer = pipeline(
@@ -10,6 +11,9 @@ sentiment_analyzer = pipeline(
     device=-1  # Force CPU usage
 )
 
+# Configure logging
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
+
 # Pre-defined prompts and affirmations for different sentiments
 PROMPT_TEMPLATES = {
     "POSITIVE": [
@@ -21,6 +25,11 @@ PROMPT_TEMPLATES = {
         "- What can you learn from this challenging situation?",
         "- What small step could you take to feel better?",
         "- Who or what helps you feel supported during difficult times?"
+    ],
+    "NEUTRAL": [
+        "- What does this experience teach you about balance?",
+        "- How does this experience fit into your overall life story?",
+        "- What is something small you are grateful for today?"
     ]
 }
 
@@ -34,6 +43,11 @@ AFFIRMATIONS = {
         "This too shall pass, and I am growing stronger.",
         "I trust in my ability to handle challenging situations.",
         "Every experience is teaching me something valuable."
+    ],
+    "NEUTRAL": [
+        "I appreciate the calmness of the present moment.",
+        "I am in harmony with life’s natural flow.",
+        "Balance is a gift I cultivate every day."
     ]
 }
 
@@ -42,11 +56,11 @@ class JournalCompanion:
         self.entries = []
     
     def get_prompts(self, sentiment):
-        prompts = PROMPT_TEMPLATES.get(sentiment, PROMPT_TEMPLATES["POSITIVE"])
+        prompts = PROMPT_TEMPLATES.get(sentiment, PROMPT_TEMPLATES["NEUTRAL"])
         return "\n\nReflective Prompts:\n" + "\n".join(prompts)
 
     def get_affirmation(self, sentiment):
-        affirmations = AFFIRMATIONS.get(sentiment, AFFIRMATIONS["POSITIVE"])
+        affirmations = AFFIRMATIONS.get(sentiment, AFFIRMATIONS["NEUTRAL"])
         return random.choice(affirmations)
 
     def analyze_entry(self, entry_text):
@@ -59,7 +73,7 @@ class JournalCompanion:
             sentiment = sentiment_result["label"].upper()
             sentiment_score = sentiment_result["score"]
         except Exception as e:
-            print("Error during sentiment analysis:", e)
+            logging.error("Error during sentiment analysis: %s", e)
             return (
                 "An error occurred during analysis. Please try again.",
                 "Error",
@@ -86,18 +100,28 @@ class JournalCompanion:
     def get_monthly_insights(self):
         if not self.entries:
             return "No entries yet to analyze."
-            
-        total_entries = len(self.entries)
-        positive_entries = sum(1 for entry in self.entries if entry["sentiment"] == "POSITIVE")
+
+        current_month = datetime.now().month
+        monthly_entries = [entry for entry in self.entries if datetime.fromisoformat(entry["timestamp"]).month == current_month]
+        
+        total_entries = len(monthly_entries)
+        if total_entries == 0:
+            return "No entries this month to analyze."
+
+        positive_entries = sum(1 for entry in monthly_entries if entry["sentiment"] == "POSITIVE")
+        neutral_entries = sum(1 for entry in monthly_entries if entry["sentiment"] == "NEUTRAL")
+        negative_entries = total_entries - positive_entries - neutral_entries
         
         try:
             percentage_positive = (positive_entries / total_entries * 100)
-            percentage_negative = ((total_entries - positive_entries) / total_entries * 100)
+            percentage_neutral = (neutral_entries / total_entries * 100)
+            percentage_negative = (negative_entries / total_entries * 100)
             
             insights = f"""Monthly Insights:
             Total Entries: {total_entries}
             Positive Entries: {positive_entries} ({percentage_positive:.1f}%)
-            Negative Entries: {total_entries - positive_entries} ({percentage_negative:.1f}%)
+            Neutral Entries: {neutral_entries} ({percentage_neutral:.1f}%)
+            Negative Entries: {negative_entries} ({percentage_negative:.1f}%)
             """
             return insights
         except ZeroDivisionError:
@@ -227,17 +251,4 @@ def create_journal_interface():
         submit_btn.click(
             fn=journal.analyze_entry,
             inputs=[entry_input],
-            outputs=[result_message, sentiment_output, prompt_output, affirmation_output]
-        )
-        
-        insights_btn.click(
-            fn=journal.get_monthly_insights,
-            inputs=[],
-            outputs=[insights_output]
-        )
-
-    return interface
-
-if __name__ == "__main__":
-    interface = create_journal_interface()
-    interface.launch()
+            outputs=[result