Update app.py

app.py

@@ -37,17 +37,20 @@ def generate_response(history, user_input):
     """
     Generates a response from the model based on the chat history and user input.
     """
-    # Append user input to the chat history
-    history.append({"role": "user", "content": user_input})
+    # Append user input to chat history
+    chat_history.append(("User", user_input))
 
-    # Build messages for the model
-    messages = [{"role": "system", "content": SYSTEM_INSTRUCTION}] + history
+    # Prepare messages for the model
+    messages = [{"role": "system", "content": SYSTEM_INSTRUCTION}] + [
+        {"role": "user", "content": msg[1]} if msg[0] == "User" else {"role": "assistant", "content": msg[1]}
+        for msg in chat_history
+    ]
 
-    # Tokenize input for the model
+    # Tokenize the input for the model
     text = apply_chat_template(messages)
     model_inputs = tokenizer([text], return_tensors="pt").to(device)
 
-    # Generate response
+    # Generate the model's response
     generated_ids = model.generate(
         **model_inputs,
         max_new_tokens=512
@@ -57,13 +60,11 @@ def generate_response(history, user_input):
     ]
     response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
 
-    # Append the assistant's response to history
-    history.append({"role": "assistant", "content": response})
-
-    # Format the conversation for display
-    formatted_history = format_chat_history(history)
+    # Append AI response to chat history
+    chat_history.append(("MathTutor", response))
 
-    return formatted_history, history
+    # Return updated chat history
+    return chat_history
 
 def format_chat_history(history):
     """