Update app.py

app.py

@@ -104,12 +104,16 @@ 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."
+        prompt = (
+            f"You are a friendly customer success representative. The customer said: \"{review}\". "
+            "Write a warm, appreciative reply celebrating their positive experience."
+        )
     else:
         prompt = (
             f"You are a helpful customer support specialist. The customer said: \"{review}\". "
-            f"Key topics: {[kw for kw, _ in keywords]}. "
-            "Ask 1-2 clarifying questions to understand their concerns, then suggest specific improvements or next steps."
+            f"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."
         )
     result = response_pipeline(prompt)
     reply = result[0]['generated_text'].strip()
@@ -118,4 +122,6 @@ def main():
     st.write(reply)
 
 if __name__ == '__main__':
+    main() '__main__':
     main()
+