Update app.py

app.py

@@ -73,12 +73,6 @@ chat_history = []
 def interact(user_input, history, interaction_count):
     global tokenizer, model
     try:
-        if interaction_count >= MAX_INTERACTIONS:
-            response = "Thank 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
-        
         if tokenizer is None or model is None:
             raise ValueError("Tokenizer or model is not initialized.")
         
@@ -102,6 +96,10 @@ def interact(user_input, history, interaction_count):
         history.append({"role": "user", "content": user_input})
         history.append({"role": "assistant", "content": response})
         
+        if interaction_count >= MAX_INTERACTIONS:
+            response += "\n\nThank you for your questions. Our session is now over. Goodbye!"
+            history[-1]["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