Update app.py

app.py

@@ -49,11 +49,10 @@ if uploaded_file:
             # Static Session Context Recall
             st.info("🔍 Loading previous session context...")
             past_sessions = [
-                "Session 1: Patient shared ongoing stress due to work deadlines. High anxiety noted.",
-                "Session 2: Therapist suggested breathing exercises. Patient reported partial improvement.",
-                "Session 3: Discussion around sleep disturbance and journaling as a method to track mood."
+                {"date": "2024-04-15", "coping": "walking", "emotion": "anxiety", "notes": "high workload"},
+                {"date": "2024-04-22", "coping": "journaling", "emotion": "stress", "notes": "difficulty sleeping"}
             ]
-            rag_context = "\n".join(past_sessions)
+            rag_context = "\n".join([f"Session {i+1}: {s['coping']}, {s['emotion']}, {s['notes']}" for i, s in enumerate(past_sessions)])
 
             prompt_input = f"""Previous session context:\n{rag_context}\n\nCurrent session:\n{raw_transcript}"""
 
@@ -68,7 +67,29 @@ if uploaded_file:
             emotion_scores = emotion_model(raw_transcript)
 
             # Layout with Tabs
-            tab1, tab2, tab3 = st.tabs(["📝 Transcript", "📋 Summary", "💬 Emotions"])
+            
+import matplotlib.pyplot as plt
+
+# Add trends tab to UI
+tab1, tab2, tab3, tab4 = st.tabs(["📝 Transcript", "📋 Summary", "💬 Emotions", "📈 Trends"])
+
+# Plot session trend (static mock data)
+with tab4:
+    st.subheader("📈 Emotional Trends Over Time")
+
+    session_dates = ["2024-04-01", "2024-04-08", "2024-04-15", "2024-04-22"]
+    anxiety_scores = [70, 65, 55, 40]
+    sadness_scores = [30, 20, 25, 15]
+
+    fig, ax = plt.subplots()
+    ax.plot(session_dates, anxiety_scores, label='Anxiety', marker='o')
+    ax.plot(session_dates, sadness_scores, label='Sadness', marker='o')
+    ax.set_title("Emotional Trends Over Time")
+    ax.set_ylabel("Score (%)")
+    ax.set_xlabel("Session Date")
+    ax.legend()
+    st.pyplot(fig)
+
 
             with tab1:
                 st.subheader("📝 Speaker-Simulated Transcript")
@@ -76,7 +97,17 @@ if uploaded_file:
 
             with tab2:
                 st.subheader("📋 Contextual Summary")
-                st.write(summary[0]["summary_text"])
+                
+            # Insight Tracking based on previous sessions
+            insights = []
+            if "music" in raw_transcript.lower():
+                if any("walking" in s["coping"] for s in past_sessions):
+                    insights.append("Patient previously mentioned walking as a helpful coping mechanism. This time, music is highlighted instead.")
+            if "sleep" in raw_transcript.lower():
+                insights.append("Sleep continues to be a recurring theme across sessions.")
+            final_output = f"{summary[0]['summary_text']}\n\nContextual Observations:\n" + "\n".join(insights)
+            st.write(final_output)
+    
 
             with tab3:
                 st.subheader("💬 Emotional Insights (Overall)")