Update app.py

app.py

@@ -116,7 +116,9 @@ def main():
             response, _ = process_document(uploaded_file, query)
             if response:  # Check if response is not empty
                 # Update conversation history
-                st.write(response)
+                # st.write(response)
+                st.write("You:", query)
+                st.write("AI:", response)
                 update_conversation(query, response)
         else:
             st.sidebar.write("Please upload a document first.")