Update app.py

app.py

@@ -49,7 +49,16 @@ if uploaded_file:
             # Summarization
             st.info("📋 Summarizing conversation...")
             summarizer = pipeline("summarization", model="philschmid/bart-large-cnn-samsum")
-            summary = summarizer(raw_transcript, max_length=256, min_length=60, do_sample=False)
+
+        # Static Session Context Recall
+        st.info("🧠 Referencing prior session context...")
+        past_sessions = [
+            {"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([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}"""
+            summary = summarizer(prompt_input, max_length=256, min_length=60, do_sample=False)
 
             # Emotion tagging
             st.info("🎭 Extracting emotional tones...")
@@ -64,7 +73,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)")
@@ -153,7 +172,17 @@ with tab1:
 with tab2:
     st.subheader("📋 Contextual Summary")
     if 'summary' in locals():
-        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)
     else:
         st.warning("Summary not available.")