Update app.py

app.py

@@ -96,10 +96,9 @@ def interact(user_input, history, interaction_count):
         interaction_count += 1
         
         if interaction_count >= MAX_INTERACTIONS:
-            response = "Thank you for your questions. Our session is now over. Goodbye!"
-            history.append({"role": "assistant", "content": response})
-        else:
-            history.append({"role": "assistant", "content": response})
+            response = response.strip() + "\n\nThank you for your questions. Our session is now over. Goodbye!"
+        
+        history.append({"role": "assistant", "content": response})
         
         formatted_history = [(entry["content"], None) if entry["role"] == "user" else (None, entry["content"]) for entry in history if entry["role"] in ["user", "assistant"]]
         
@@ -242,4 +241,4 @@ with gr.Blocks() as demo:
     send_message_button.click(fn=interact, inputs=[chatbot_input, chat_history_json, interaction_count], outputs=[chatbot_input, chatbot_output, chat_history_json, interaction_count])
     save_button.click(fn=save_comment_score, inputs=[chatbot_output, score_input, comment_input, story_dropdown, user_dropdown, system_prompt_dropdown], outputs=[data_table, comment_input])
 
-demo.launch()
+demo.launch()