Update frontend.py

frontend.py

@@ -151,7 +151,32 @@ with tab1:
         st.download_button("⬇️ Download Audio", audio.read(), "summary.mp3")
 
         st.markdown("### 🔁 Ask a Follow-Up")
-        sample_questions = ["What made the user upset?", "Any feature complaints?", "How urgent is this?"]
+        
+        # 💡 Smarter Follow-Up Suggestion Logic
+        sentiment = data["sentiment"]["label"].lower()
+        churn = data.get("churn_risk", "")
+        pain = data.get("pain_points", [])
+
+        if sentiment == "positive" and churn == "Low Risk":
+            sample_questions = [
+                "What features impressed the user?",
+                "Would they recommend the product?",
+                "What delighted the customer most?"
+            ]
+        elif sentiment == "negative" or churn == "High Risk" or pain:
+            sample_questions = [
+                "What made the user upset?",
+                "Any feature complaints?",
+                "Is this user likely to churn?"
+            ]
+        else:
+            sample_questions = [
+                "What are the key takeaways?",
+                "How can this review guide product changes?",
+                "Is there any concern raised?"
+            ]
+
+
         selected_q = st.selectbox("💡 Suggested Questions", ["Type your own..."] + sample_questions)
         custom_q = selected_q if selected_q != "Type your own..." else st.text_input("🔍 Follow-up Question")