Update app.py

app.py

@@ -34,11 +34,15 @@ if user_input:
     st.session_state.history.append(tokenizer.encode(user_input + tokenizer.eos_token, return_tensors="pt"))
     st.write(f"Bot: {response}")
 
-# Show chat history
+# Show chat history (safer decoding)
 if st.session_state.history:
+    st.subheader("📝 Chat History")
     for i in range(len(st.session_state.history) - 1, -1, -1):
-        user_msg = tokenizer.decode(st.session_state.history[i], skip_special_tokens=True)
-        st.write(f"You: {user_msg}")
+        try:
+            user_msg = tokenizer.decode(st.session_state.history[i][0].tolist(), skip_special_tokens=True)
+            st.write(f"You: {user_msg}")
+        except Exception as e:
+            st.warning(f"Could not decode history message: {e}")
 
 # --- File Upload and Processing ---
 st.subheader("📁 Upload a File for AI to Read")