Update app.py

app.py

@@ -1,12 +1,12 @@
 # Smart Customer Support Assistant (Enhanced UI Version)
-# Note: Core analysis logic remains unchanged
+# Note: Core analysis logic remains unchanged, now with text generation
 
 import streamlit as st
 from transformers import pipeline
 import re
 
 # ------------------------------
-# Load models (same as before)
+# Load models (now includes 3rd: text generation)
 # ------------------------------
 emotion_classifier = pipeline(
     "text-classification",
@@ -14,9 +14,10 @@ emotion_classifier = pipeline(
     return_all_scores=True
 )
 intent_classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+text_generator = pipeline("text2text-generation", model="declare-lab/flan-alpaca-base")
 
 # ------------------------------
-# Candidate tasks / templates
+# Candidate tasks / prompts
 # ------------------------------
 candidate_tasks = [
     "change mobile plan",
@@ -29,27 +30,10 @@ candidate_tasks = [
     "upgrade device"
 ]
 
-intent_solutions = {
-    "top up balance": "Your balance is ¥12. Promo: recharge ¥100 get ¥5 bonus.",
-    "reactivate service": "Service suspended due to unpaid ¥38. Recharge to restore in 30 mins.",
-    "change mobile plan": "You're on Basic (¥68/5GB). Suggest Plus (¥98/20GB).",
-    "check account status": "Data: 3.2GB/5GB. Balance: ¥12. Calls left: 22 mins.",
-    "ask for billing support": "Last bill: ¥96 (Mar). Includes ¥16 overage.",
-    "cancel subscription": "Contract ends: 2025-06-30. No penalty after this date.",
-    "upgrade device": "Eligible for upgrade. New iPhone plan: ¥399/month.",
-    "report service outage": "Signal issues detected (ZIP: XXX). Engineers notified."
-}
-
-intent_closings = {
-    "top up balance": "Proceed with recharge now?",
-    "reactivate service": "Shall I help restart your service?",
-    "report service outage": "Shall I file a service report?",
-    "change mobile plan": "Switch to a better plan?",
-    "ask for billing support": "Show recent billing records?",
-    "cancel subscription": "Guide you through cancellation?",
-    "check account status": "Show usage and balance?",
-    "upgrade device": "See available upgrades?"
-}
+def generate_response(intent):
+    prompt = f"Write a polite and helpful customer service response for this request: '{intent}'"
+    output = text_generator(prompt, max_new_tokens=80, do_sample=True)[0]['generated_text']
+    return output
 
 urgent_emotions = {"anger", "frustration", "anxiety", "urgency", "afraid", "annoyed"}
 moderate_emotions = {"confused", "sad", "tired", "concerned", "sadness"}
@@ -133,7 +117,7 @@ if analyze_clicked and user_input.strip():
 
     if final_score < 0.5 and top_intents:
         intent = top_intents[0]
-        response = f"Thank you for contacting us. I understand your concern. {intent_solutions[intent]} {intent_closings[intent]}"
+        response = generate_response(intent)
         st.session_state.chat.append({"role": "assistant", "content": response})
         st.session_state.system_result = None
         st.session_state.support_required = "🟢 Automated response handled this request."
@@ -167,7 +151,7 @@ if st.session_state.system_result is not None:
 
     st.markdown("#### Detected Customer Needs")
     for intent in st.session_state.system_result['intents']:
-        suggestion = f"Thank you for contacting us. I understand your concern. {intent_solutions[intent]} {intent_closings[intent]}"
+        suggestion = generate_response(intent)
         st.markdown(f"**• {intent.capitalize()}**")
         if st.button(suggestion, key=f"btn_{intent}"):
-            st.session_state.agent_reply = suggestion
+            st.session_state.agent_reply = suggestion