Update app.py

app.py

@@ -42,19 +42,25 @@ def get_content_score(text, top_intents):
         score += 0.2
     return min(score + 0.1, 1.0)
 
+# Updated reply generator with intent-specific follow-up
 def generate_reply(input_text, intent):
-    prompt = (
-        f"You are a telecom customer service agent. Based on the customer's message: '{input_text}'\n"
-        f"The detected issue is: '{intent}'.\n"
-        f"Generate a reply in 3 short sentences:\n"
-        f"1. Start with a polite sentence showing understanding.\n"
-        f"2. Offer a clear, actionable solution to the issue.\n"
-        f"3. End with a short, friendly question offering further help.\n"
-        f"Each sentence should be no more than 100 characters.\n"
-    )
-    result = text_generator(prompt)[0]['generated_text']
-    cleaned = re.sub(r'(\. ){2,}', '. ', result)
-    return cleaned.strip()
+    # Defined intent-specific closing prompts
+    intent_closings = {
+        "top up balance": "Would you like to see recharge options and prices?",
+        "reactivate service": "Shall I help you reactivate your number now?",
+        "report service outage": "Would you like me to file a service report for you?",
+        "change mobile plan": "Need me to compare available plans for you?",
+        "ask for billing support": "Would you like me to check your latest bill details?",
+        "cancel subscription": "Should I guide you through the cancellation process?",
+        "check account status": "Would you like an overview of your account status?",
+        "upgrade device": "Would you like me to show available upgrade options?"
+    }
+
+    closing = intent_closings.get(intent.lower(), "Is there anything else I can help you with?")
+    opening = "Thanks for reaching out. I understand your concern."
+    action = f"Here's how we can assist with '{intent.lower()}'."
+    reply = f"{opening} {action} {closing}"
+    return reply
 
 # UI
 st.set_page_config(page_title="Customer Support Assistant", layout="centered")