Update app.py

app.py

@@ -44,6 +44,7 @@ def regenerate(chatbot, chat_history_state, system_msg, max_tokens, temperature,
 
 
 def chat(history, system_message, max_tokens, temperature, top_p, top_k, repetition_penalty):
+    print(f"Chat function called with history: {history}")
     history = history or []
     
     # Use BASE_SYSTEM_MESSAGE if system_message is empty
@@ -51,7 +52,7 @@ def chat(history, system_message, max_tokens, temperature, top_p, top_k, repetit
     
     # A última mensagem do usuário
     user_prompt = history[-1][0] if history else ""
-    
+    print(f"User prompt used for generation: {user_prompt}")  # Debug print
     # Preparar a entrada para o modelo
     prompt_template = f'''system
 {system_message_to_use.strip()}
@@ -74,13 +75,14 @@ assistant
     # Decodificar a saída
     decoded_output = tokenizer.decode(output[0], skip_special_tokens=True)
     assistant_response = decoded_output.split('assistant')[-1].strip()  # Pegar apenas a última resposta do assistente
-    
+    print(f"Generated assistant response: {assistant_response}")  # Debug print
     # Atualizar o histórico
     if history:
         history[-1][1] += assistant_response
     else:
         history.append(["", assistant_response])
-    
+        
+    print(f"Updated history: {history}")
     return history, history, ""