Update app.py

app.py

@@ -49,15 +49,22 @@ def get_emotion_score(emotion):
         return 0.2
 
 def generate_response(intent, human=True):
-    prompt = (
-        f"You are a telecom customer service assistant. For the customer intent '{intent}', generate a 3-part response:\n"
-        "[Greeting: polite welcome.]\n"
-        "[Middle: mention the customer is currently on Plan X (¥X/month), and suggest switching to Plan Y with XXGB at ¥Y/month. Use fictional placeholder values.]\n"
-        "[End: ask if they'd like to proceed with the new plan or need more details.]"
-    )
-    result = text_generator(prompt, max_new_tokens=100, do_sample=False)
+    if human:
+        prompt = (
+            f"You are a telecom customer service agent. For the customer intent '{intent}', generate a professional 3-part response:\n"
+            "1. Greeting (e.g., Thank you for contacting us...)\n"
+            "2. Current plan summary (e.g., You're currently on Plan X at ¥X/month. We suggest Plan Y with XXGB for ¥Y/month.)\n"
+            "3. Close with a question (e.g., Would you like to switch?)"
+        )
+    else:
+        prompt = (
+            f"As a telecom assistant, generate a full 3-part structured response for customer intent: '{intent}'.\n"
+            "Include greeting, summary of their current plan and a better offer (fictional), and end with a follow-up question."
+        )
+    result = text_generator(prompt, max_new_tokens=120, do_sample=False)
     return result[0]['generated_text'].strip()
 
+
 # App UI Setup
 st.set_page_config(page_title="Smart Customer Support Assistant", layout="wide")
 st.sidebar.title("📁 Customer Selector")