Update app.py

app.py

@@ -31,8 +31,15 @@ def load_keybert_model():
     return KeyBERT(model="all-MiniLM-L6-v2")
 
 # ─── BlenderBot Response Pipeline ───────────────────────────────────────────
+from transformers import Conversation
+
 @st.cache_resource
 def load_response_pipeline():
+    # Use conversational pipeline for interactive-style replies
+    return pipeline(
+        "conversational",
+        model="facebook/blenderbot-400M-distill"
+    ):
     # High-level helper using BlenderBot 400M Distill
     return pipeline(
         "text2text-generation",
@@ -100,19 +107,17 @@ def main():
     # Generate appropriate reply
     response_pipeline = load_response_pipeline()
     if max_label in ["Positive", "Very Positive"]:
-        prompt = (
-            f"You are a friendly customer success representative. The customer said: \"{review}\". "
-            "Write a warm, appreciative reply celebrating their positive experience."
-        )
+        user_input = f"The customer said: \"{review}\". Write a warm, appreciative two-sentence reply celebrating their positive experience."
     else:
-        prompt = (
-            f"You are a helpful customer support specialist. The customer said: \"{review}\". "
-            f"Identified issues: {', '.join([kw for kw, _ in keywords])}. "
+        user_input = (
+            f"The customer said: \"{review}\". Identified issues: {', '.join([kw for kw, _ in keywords])}. "
             "First, ask 1-2 clarifying questions to better understand their situation. "
-            "Then, provide two concrete suggestions or next steps to address these issues, grounded in their feedback."
+            "Then provide two concrete suggestions or next steps to address these issues."
         )
-    result = response_pipeline(prompt)
-    reply = result[0]['generated_text'].strip()
+    conv = Conversation(user_input)
+    response = response_pipeline(conv)
+    # Grab the latest generated response
+    reply = response.generated_responses[-1]
 
     st.subheader("Generated Reply")
     st.write(reply)
@@ -120,3 +125,4 @@ def main():
 
 if __name__ == '__main__':
     main()
+