Update app.py

app.py

@@ -32,7 +32,7 @@ def get_system_prompt():
 
 def generate_response(input_text):
     # If it's the first interaction, add the system prompt to the conversation history
-    if not st.session_state['history']:
+    if len(st.session_state['history']) == 0:
         system_prompt = get_system_prompt()
         st.session_state['conversation'].append(f"System: {system_prompt}")
         system_input_ids = tokenizer.encode(system_prompt + tokenizer.eos_token, return_tensors='pt').to(device)
@@ -41,8 +41,8 @@ def generate_response(input_text):
     # Encode the new user input, add end of string token
     new_user_input_ids = tokenizer.encode(input_text + tokenizer.eos_token, return_tensors='pt').to(device)
 
-    # If there is conversation history, append the new input to it
-    if st.session_state['history']:
+    # Append the new user input to the history
+    if len(st.session_state['history']) > 0:
         history_tensor = torch.tensor(st.session_state['history']).unsqueeze(0).to(device)
         bot_input_ids = torch.cat([history_tensor, new_user_input_ids], dim=-1)
     else: