Update frontend.py

frontend.py

@@ -6,7 +6,13 @@ import base64
 from io import BytesIO
 import os
 import plotly.express as px
+from datetime import datetime
+import uuid
 
+# Simulated in-memory storage for churn log (replace with a database or file in production)
+if "churn_log" not in st.session_state:
+    st.session_state.churn_log = []
+    
 st.set_page_config(page_title="ChurnSight AI", page_icon="🧠", layout="wide")
 
 if os.path.exists("logo.png"):
@@ -145,6 +151,19 @@ with tab1:
 
         if "pain_points" in data and data["pain_points"]:
             st.error("🔍 Pain Points: " + ", ".join(data["pain_points"]))
+        # Add churn risk to session log
+        try:
+            st.session_state.churn_log.append({
+                "timestamp": datetime.now(),
+                "product": data.get("product_category", "General"),
+                "churn_risk": data.get("churn_risk", "Unknown"),
+                "session_id": str(uuid.uuid4())
+           })
+           # Keep log lightweight
+           if len(st.session_state.churn_log) > 1000:
+               st.session_state.churn_log = st.session_state.churn_log[-1000:]
+        except Exception as e:
+            st.warning(f"🧪 Logging failed: {e}")
 
         st.subheader("🔊 Audio Summary")
         audio = speak(data["summary"], lang=voice_lang)
@@ -200,3 +219,17 @@ with tab1:
     if st.session_state.followup_answer:
         st.subheader("✅ Answer")
         st.success(st.session_state.followup_answer)
+    # 📈 Show churn trend chart (optional)
+   if st.checkbox("📊 Show Churn Risk Trends", value=False):
+       try:
+           df = pd.DataFrame(st.session_state.churn_log)
+           df["date"] = pd.to_datetime(df["timestamp"]).dt.date
+           trend = df.groupby(["date", "churn_risk"]).size().unstack(fill_value=0).reset_index()
+
+           st.markdown("#### 📅 Daily Churn Trend")
+           fig = px.bar(trend, x="date", y=["High Risk", "Low Risk"], barmode="group", title="Daily Churn Risk Distribution")
+           st.plotly_chart(fig, use_container_width=True)
+
+           st.download_button("⬇️ Export Trend CSV", trend.to_csv(index=False), "churn_trend.csv", mime="text/csv")
+       except Exception as e:
+           st.error(f"❌ Failed to generate trend: {e}")