Update app.py

app.py

@@ -375,10 +375,13 @@ if __name__ == '__main__':
         st.session_state.tooltip = 'The AI answer your questions only considering the documents provided'
     
     client, collection_name, tokenizer, model, llm, dense_model, sparse_model = load_models_and_documents()
+
+    if 'df' not in st.session_state:
+        st.session_state.df = pd.DataFrame([0])
     
     if menu_id == 'Documents':
         editable_df = st.data_editor(
-            pd.DataFrame([]), 
+            st.session_state.df, 
             num_rows="dynamic",
             use_container_width=True,
             column_config={
@@ -417,6 +420,7 @@ if __name__ == '__main__':
                 ) 
             }
         )
+        st.session_state.df = editable_df
 
     conversations_path = os.path.join(os.getenv('HF_HOME'), 'conversations')
 
@@ -545,6 +549,14 @@ if __name__ == '__main__':
                                      include_page_breaks=True
                                     )
                 metadata_dict = {"source": uploaded_file.name}
+
+                name, extension = os.path.splitext(uploaded_file.name)
+                
+                st.session_state.df.append(
+                    {
+                       'document': name, 'type': extension, 'time': 
+                    }
+                )
                 
                 texts, metadatas = [], []
                 for elem in elements: