Update utils.py

utils.py

@@ -471,47 +471,6 @@ def load_split_to_original_mapping(directory="chroma/kkg", filename="mapping.pkl
 
 
 
-    
-
-"""
-
-def save_vectorstore(vectorstore, directory="chroma/kkg", splits_filename="splits_and_metadata.pkl", vectorstore_filename="vectorstore.pkl"):
-    # Erstellen des Verzeichnisses, falls es nicht existiert
-    if not os.path.exists(directory):
-        os.makedirs(directory)
-    
-    # Speichern der Splits und Metadaten
-    splits_filepath = os.path.join(directory, splits_filename)
-    with open(splits_filepath, "wb") as f:
-        pickle.dump((vectorstore.documents, SPLIT_TO_ORIGINAL_MAPPING), f)
-    
-    # Speichern des Vektorstores (ohne die SQLite-Verbindung)
-    vectorstore_filepath = os.path.join(directory, vectorstore_filename)
-    with open(vectorstore_filepath, "wb") as f:
-        pickle.dump(vectorstore, f)
-
-    # Hochladen der gespeicherten Dateien zum Hugging Face Space
-    upload_file_to_huggingface(splits_filepath, f"{directory}/{splits_filename}")
-    upload_file_to_huggingface(vectorstore_filepath, f"{directory}/{vectorstore_filename}")
-
-def load_vectorstore(directory="chroma/kkg", splits_filename="splits_and_metadata.pkl", vectorstore_filename="vectorstore.pkl"):
-    # Laden der Splits und Metadaten
-    splits_filepath = os.path.join(directory, splits_filename)
-    vectorstore_filepath = os.path.join(directory, vectorstore_filename)
-    
-    if os.path.exists(splits_filepath) and os.path.exists(vectorstore_filepath):
-        with open(splits_filepath, "rb") as f:
-            documents, SPLIT_TO_ORIGINAL_MAPPING = pickle.load(f)
-        
-        with open(vectorstore_filepath, "rb") as f:
-            vectorstore = pickle.load(f)
-            vectorstore.documents = documents  # Setze die Dokumente im Vektorstore
-            return vectorstore, SPLIT_TO_ORIGINAL_MAPPING
-    return None, None
-"""
-
-
-
 # Beispiel-Upload-Funktion
 def upload_file_to_huggingface(file_path, upload_path):
     api.upload_file(
@@ -769,29 +728,32 @@ def download_link(doc):
 
 
 
+
+
+#################################################
+#File Liste beim Tab für File-Upload schön darstellen
+#################################################
 def display_files():
     files = os.listdir(DOCS_DIR_PDF)
     files_table = "<table style='width:100%; border-collapse: collapse;'>"
-    files_table += "<tr style='background-color: #930BBA; color: white; font-weight: bold; font-size: larger;'><th>Dateiname</th><th>Größe (KB)</th></tr>"
+    files_table += "<tr style='background-color: #930BBA; color: white; font-weight: bold; font-size: larger;'><th>Dateiname - PDF-Ordner</th></tr>"
     for i, file in enumerate(files):
         file_path = os.path.join(DOCS_DIR_PDF, file)
         file_size = os.path.getsize(file_path) / 1024  # Größe in KB
         row_color = "#4f4f4f" if i % 2 == 0 else "#3a3a3a"  # Wechselnde Zeilenfarben
         files_table += f"<tr style='background-color: {row_color}; border-bottom: 1px solid #ddd;'>"
-        files_table += f"<td><b>{download_link(file)}</b></td>"
-        files_table += f"<td>{file_size:.2f}</td></tr>"
+        files_table += f"<td><b>{download_link(file)}</b></td></tr>"
     files_table += "</table>"
 
     files = os.listdir(DOCS_DIR_WORD)
     files_table += "<table style='width:100%; border-collapse: collapse;'>"
-    files_table += "<tr style='background-color: #930BBA; color: white; font-weight: bold; font-size: larger;'><th>Dateiname</th><th>Größe (KB)</th></tr>"
+    files_table += "<tr style='background-color: #930BBA; color: white; font-weight: bold; font-size: larger;'><th>Dateiname - Word-Ordner</th></tr>"
     for i, file in enumerate(files):
         file_path = os.path.join(DOCS_DIR_WORD, file)
         file_size = os.path.getsize(file_path) / 1024  # Größe in KB
         row_color = "#4f4f4f" if i % 2 == 0 else "#3a3a3a"  # Wechselnde Zeilenfarben
         files_table += f"<tr style='background-color: {row_color}; border-bottom: 1px solid #ddd;'>"
-        files_table += f"<td><b>{download_link(file)}</b></td>"
-        files_table += f"<td>{file_size:.2f}</td></tr>"
+        files_table += f"<td><b>{download_link(file)}</b></td></tr>"
     files_table += "</table>"
     return files_table