Update app.py

app.py

@@ -70,9 +70,9 @@ def main():
         st.title("Document Content")
         file_extension = os.path.splitext(uploaded_file.name)[1].lower()
         if file_extension == ".docx":
-            response, document_text = process_document(uploaded_file, "")
-            # if document_text is not None:
-            #     st.text_area("Document Text", value=document_text, height=300)
+            _, document_text = process_document(uploaded_file, "")
+            if document_text is not None:
+                st.text_area("Document Text", value=document_text, height=300)
         elif file_extension == ".pdf":
             images = pdf_to_images(uploaded_file.getvalue())
             if images:
@@ -88,7 +88,9 @@ def main():
         if uploaded_file is not None:
             response, _ = process_document(uploaded_file, query)
             if response:
-                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.")