Upload 2 files

pages/table.py

@@ -1,12 +1,9 @@
 import streamlit as st
-from transformers import pipeline
 from datetime import datetime
 import pandas as pd
 from pathlib import Path
 import os
-from huggingface_hub import login
 
-login(token = "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")
 
 st.set_page_config(
     page_title="Hochgeladene Dokumente",
@@ -36,7 +33,7 @@ def newIndex():
     return newIndex
 
 def generate_files():
-    folder = './files'
+    folder = 'C:/Users/keckm/Documents/GitHub/Streamlit/files'
     files = os.listdir(folder)
     for file in files:
         file_path = Path(folder, file)
@@ -53,13 +50,20 @@ def generate_files():
             FILES.append(file_info)
     return FILES
 
-def delete_entry(key):
-    folder = './files'
-    file = FILES[key]['Titel']
-    os.remove(Path(folder, file))
-    FILES.pop(key)
-    generate_files()
+def delete_row_and_file(index):
+    folder = 'C:/Users/keckm/Documents/GitHub/Streamlit/files'
+    file_name = FILES[index]['Titel']
+    file_path = Path(folder, file_name)
 
+    # Delete the file
+    try:
+        os.remove(file_path)
+        st.success("Datei erfolgreich gelöscht.")
+    except FileNotFoundError:
+        st.warning("Datei nicht gefunden.")
+
+    # Drop the row from the DataFrame
+    FILES.pop(index)
 
 def display_table():
         hide_table_row_index = """ 
@@ -68,24 +72,24 @@ def display_table():
             tbody th {display:none}
             </style>
             """
-        
-
 # Inject CSS with Markdown
         st.markdown(hide_table_row_index, unsafe_allow_html=True)
 
-# Display a static table
-col1, col2 = st.columns([5, 1])
-with col1:
-    table1 = pd.DataFrame.from_dict(generate_files())
-    st.table(table1.style.format(subset=['Titel']))
-with col2:
-    key = 0
-    for entry in FILES:
-       # st.button('Löschen', on_click = delete_entry(key), key = key, use_container_width = True)
-        st.button('Löschen', key = key, use_container_width = True)
-        key = key + 1
+        # Display a static table
+        table1 = pd.DataFrame.from_dict(generate_files())
+        table = st.table(table1)
+        return table1
 
-display_table()
+    #.style.format(subset=['Titel']))
 
+col1, col2 = st.columns([0.7, 0.3])
+with col1:
+    display_table()
+
+with col2:
+    for index in range(len(FILES)):
+        delete_button = st.button("Löschen", key = index)
+        if delete_button:
+            delete_row_and_file(index)
 
 

pages/upload.py

@@ -2,17 +2,12 @@ import streamlit as st
 from pathlib import Path
 from pages.table import generate_files
 from pages.bot import create_vectorstore_and_store
-
-st.set_page_config(
-    page_title="Dokumentenupload",
-    layout="wide",
-    initial_sidebar_state="expanded",
-)
+#from bot.py import *
 
 st.markdown("# Dokumentenupload")
 st.sidebar.markdown("# Dokumentenupload")
 
-UPLOAD_FOLDER = './files' # Ordner für Dateiablage
+UPLOAD_FOLDER = 'C:/Users/keckm/Documents/GitHub/Streamlit/files' # Ordner für Dateiablage
 
 def save_file(uploaded_file):
     save_folder = UPLOAD_FOLDER
@@ -22,7 +17,6 @@ def save_file(uploaded_file):
     if save_path.exists():
         st.success(f'Dokument {uploaded_file.name} wurde erfolgreich auf den Server hochgeladen.')
         create_vectorstore_and_store()
-
 # file uploader
 with st.form('upload_form'):
     uploaded_file = st.file_uploader("Wählen Sie eine Datei aus.", type = [ ".pdf"])