Update app.py

app.py

@@ -76,6 +76,9 @@ def interact(user_input, history, interaction_count):
         if tokenizer is None or model is None:
             raise ValueError("Tokenizer or model is not initialized.")
         
+        if interaction_count >= MAX_INTERACTIONS:
+            user_input += " Thank you for your questions. Our session is now over. Goodbye!"
+        
         messages = history + [{"role": "user", "content": user_input}]
         
         # Ensure roles alternate correctly
@@ -92,14 +95,10 @@ def interact(user_input, history, interaction_count):
         
         # Update chat history with generated response
         history.append({"role": "user", "content": user_input})
+        history.append({"role": "assistant", "content": response})
         
         interaction_count += 1
         
-        if interaction_count >= MAX_INTERACTIONS:
-            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"]]
         
         return "", formatted_history, history, interaction_count
@@ -241,4 +240,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()