Update app.py

app.py

@@ -73,6 +73,12 @@ 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.")
         
@@ -85,11 +91,6 @@ def interact(user_input, history, interaction_count):
             if messages[i-1].get("role") == messages[i].get("role"):
                 raise ValueError("Conversation roles must alternate user/assistant/user/assistant/...")
         
-        if interaction_count >= MAX_INTERACTIONS:
-            user_input += '''.
-            ***
-            Thank you for your questions. Our session is now over. Goodbye!'''
-        
         prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
         
         # Generate response using selected model