Update app.py

app.py

@@ -54,21 +54,16 @@ def get_emotion_score(emotion):
         return 0.2
 
 def generate_response(intent, human=True):
-    if human:
-        prompt = (
-            f"You are a telecom customer service agent. For the customer intent '{intent}', generate a 3-sentence response:\n"
-            "[Greeting: polite opener.]\n"
-            "[Middle: State that current plan is Plan X, ¥X/month, and recommend Plan Y with XXGB at ¥Y/month. Use fictional values.]\n"
-            "[End: Ask if they'd like to proceed. One sentence only.]"
-        )
-    else:
-        prompt = (
-            f"You are a helpful telecom assistant. Briefly resolve the intent '{intent}' in one sentence. "
-            "Use placeholders like Plan X, ¥X/month, and keep the tone friendly."
-        )
-    result = text_generator(prompt, max_new_tokens=80, do_sample=False)
+    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)
     return result[0]['generated_text'].strip()
 
+
 # Streamlit config
 st.set_page_config(page_title="Smart Customer Support Assistant", layout="wide")
 st.sidebar.title("📁 Customer Selector")