Update app.py

app.py

@@ -364,7 +364,8 @@ def upload_pdf(files):
     if not files:
         logging.warning("No files uploaded.")
         return "Keine Dateien hochgeladen.", display_files()
-    
+        
+    futures = []
     with ThreadPoolExecutor() as pool:
         for file in files:
             try:
@@ -382,11 +383,35 @@ def upload_pdf(files):
                     upload_path = f"chroma/kkg/{filename}"
 
                 # Hochladen der Datei
-                pool.submit(upload_file_to_huggingface, file.name, upload_path)
+                future = pool.submit(upload_file_to_huggingface, file.name, upload_path)
+                futures.append(future)
             except Exception as e:
-                logging.error(f"Error uploading file {file.name}: {e}")
-    
-    return "Dateien wurden erfolgreich hochgeladen.", display_files()
+                logging.error(f"Error preparing upload for file {file.name}: {e}")
+
+    # Warten, bis alle Uploads abgeschlossen sind
+    for future in as_completed(futures):
+        try:
+            future.result()
+        except Exception as e:
+            logging.error(f"Error during file upload: {e}")
+
+    # Nachdem alle Dateien hochgeladen wurden, den Vektorstore neu laden
+    try:
+        ############################################
+        #Vektorstore neu....
+        ############################################
+        create_and_save_vectorstore()
+        logging.info("Vektorstore successfully renewed.")
+        status_message = "Dateien wurden erfolgreich hochgeladen und Vektorstore wurde erneuert."
+    except Exception as e:
+        logging.error(f"Error renewing Vektorstore: {e}")
+        status_message = "Fehler beim Erneuern des Vektorstores."
+
+    return status_message, display_files()
+
+
+
+
 
 """
 def upload_pdf(files):
@@ -410,13 +435,13 @@ def upload_pdf(files):
         
             # Datei zum Hugging Face Space hochladen
             upload_file_to_huggingface(file.name, upload_path)
-            ############################################
-            #Vektorstore neu....
-            ############################################
-            #create_vectorstore()
+            
         except Exception as e:
             logging.error(f"Error uploading file {file.name}: {e}")
-        
+    ############################################
+    #Vektorstore neu....
+    ############################################
+    #create_vectorstore()    
     return f"Dateien erfolgreich hochgeladen.", display_files()    
 """
 
@@ -593,15 +618,15 @@ with gr.Blocks(css=customCSS, theme=themeAlex) as demo:
     with gr.Tab("Datei hochladen"):
         upload_pdf_files = gr.Files(label="PDF- oder Word-Dateien hochladen", file_count="multiple")
         output_text = gr.Textbox(label="Status")
-        renew_button = gr.Button("Vektorstore erneuern", elem_id="renew_button")
+        renew_button = gr.Button("Dateien hochladen", elem_id="renew_button")
         file_list = gr.HTML(elem_id="file_list", show_label=False)
 
         
 
         # Automatisches Ausführen der Upload-Funktion, wenn eine Datei hochgeladen wird
         #upload_pdf_file.change(fn=upload_pdf, inputs=upload_pdf_file, outputs=output_text)
-        upload_pdf_files.change(fn=upload_pdf, inputs=upload_pdf_files, outputs=[output_text, file_list]) #, postprocess=display_files)  
-        renew_button.click(fn=renew_vectorstore, outputs=output_text)
+        #upload_pdf_files.change(fn=upload_pdf, inputs=upload_pdf_files, outputs=[output_text, file_list]) #, postprocess=display_files)  
+        renew_button.click(fn=upload_pdf, inputs=upload_pdf_files, outputs=[output_text, file_list])
         #gr.HTML(update=display_files, elem_id="file_list", show_label=False)
         demo.load(display_files, outputs=file_list)