Update app.py

app.py

@@ -68,6 +68,7 @@ tokenizer, model = load_model(selected_model)
 
 # Chat history and interaction counter
 chat_history = []
+interaction_count = 0
 
 # Function to handle interaction with model
 @spaces.GPU
@@ -77,6 +78,12 @@ def interact(user_input, history, interaction_count):
         if tokenizer is None or model is None:
             raise ValueError("Tokenizer or model is not initialized.")
         
+        interaction_count += 1
+        print(f"Interaction count: {interaction_count}")
+
+        if interaction_count >= MAX_INTERACTIONS:
+            user_input += ". Thank you for the questions. That's all for now. Goodbye!"
+
         messages = history + [{"role": "user", "content": user_input}]
         
         # Ensure roles alternate correctly
@@ -95,13 +102,6 @@ def interact(user_input, history, interaction_count):
         history.append({"role": "user", "content": user_input})
         history.append({"role": "assistant", "content": response})
         
-        interaction_count += 1
-        print(f"Interaction count: {interaction_count}")
-        
-        if interaction_count >= MAX_INTERACTIONS:
-            response += ". Thank you for the questions. That's all for now. 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
     except Exception as e: