Update app.py

app.py

@@ -430,11 +430,11 @@ if __name__ == '__main__':
     st.title('Multipurpose AI Agent')
     #st.markdown("<h1 style='position: fixed; top: 0; left: 0; width: 100%; padding: 10px; text-align: left; color: black;'>Multipurpose AI Agent</h1>", unsafe_allow_html=True)
 
-    data_editor_path = os.path.join(os.getenv('HF_HOME'), 'documents.parquet.br')
+    data_editor_path = os.path.join(os.getenv('HF_HOME'), 'documents')
     
     if 'df' not in st.session_state:
         if os.path.exists(data_editor_path):
-            st.session_state.df = pd.read_parquet(data_editor_path, engine='pyarrow')
+            st.session_state.df = pd.read_parquet(os.path.join(data_editor_path, 'data_editor.parquet.br'), engine='pyarrow')
         else:
             st.session_state.df = pd.DataFrame()
             os.mkdir(data_editor_path)
@@ -676,8 +676,18 @@ if __name__ == '__main__':
                     'time': [datetime.strptime(processing_time, '%d %b %Y, %I:%M %p')],
                     'toggle': True
                 }
-                st.session_state.df = pd.concat([st.session_state.df, pd.DataFrame(data=d)])
-                st.session_state.df.to_parquet(data_editor_path, compression='brotli')
+                
+                st.session_state.df = pd.concat(
+                    [st.session_state.df, pd.DataFrame(data=d)]
+                )
+                st.session_state.df.to_parquet(
+                    os.path.join(
+                        data_editor_path, 
+                        'data_editor.parquet.br'
+                    ), 
+                    compression='brotli', 
+                    engine='pyarrow'
+                )
 
                 elements = partition_pptx(file=uploaded_file)