Update app.py

app.py

@@ -369,10 +369,10 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
 ########################################
 # Hochladen von Dateien und Vektorstore neu erstellen
 # Beispiel-Upload-PDF-Funktion
-def upload_pdf(files):
+def upload_pdf(files, status):
     status_message = ""
     if not files:
-        status_message = " Keine Dateien zum Hochladen"
+        status_message = " Keine Dateien zum Hochladen!  "
     else:
         for file in files:
             try: 
@@ -400,20 +400,22 @@ def upload_pdf(files):
                 logging.error(f"Error uploading file {file.name}: {e}")
                 status_message = "Nicht alle Dateien konnten hochgeladen werden..."
                 
-        status_message = "Hochladen der Dateien abgeschlossen!"
+        status_message = "Hochladen der Dateien abgeschlossen!  "
 
-    return gr.Textbox(label="Status", visible = True), display_files()
+    return gr.Textbox(label="Status", visible = True), display_files(), status_message
 
 # Nachdem alle Dateien hochgeladen wurden, den Vektorstore neu laden    
-def update_vectorstore():
+def update_vectorstore(status):
     try:
         ############################################
         #Vektorstore neu....
         ############################################
         create_vectorstore()
-        status_message = "Vektorstore wurde erneuert."
+        messaage = status + "Vektorstore wurde erneuert"
+        return messaage, messaage
     except Exception as e:
-        status_message = "Fehler beim Erneuern des Vektorstores."
+        messaage = status + "Fehler beim Erneuern des Vektorstores"
+        return messaage, messaage
         
 #File Input automatisch nach upload Prozess resetten
 def reset_file_input():
@@ -430,9 +432,11 @@ def hide_status():
 def show_status():
     return  gr.HTML(value="", label="Status", visible=True)  ##gr.Textbox( label="Status", visible = True)
 
-def show_text_status():
-    return gr.HTML(value='<div style="text-align: center; color: red;">System erfolgreich aktualisiert!</div>', label="Status", visible=True)
-
+def show_text_status(status):
+    if status.contains("Vektorstore wurde erneuert"):
+        return gr.HTML(value='<div style="text-align: center; color: red;">System erfolgreich aktualisiert!</div>', label="Status", visible=True), ""
+    else:
+        return gr.HTML(value='<div style="text-align: center; color: red;">System konnte nicht aktualisiert werden!</div>', label="Status", visible=True), ""
 
 ########################################
 # Bot- test gegen schädliche Bots die die Anwendung testen...
@@ -462,7 +466,7 @@ def custom_css():
 #################################################################################################    
 print ("Start GUI Hauptanwendung")
 with open("custom.css", "r", encoding="utf-8") as f:
-    customCSS = f.read()
+    customCSS = f.read() 
 
 #Add Inputs für Tab 2
 additional_inputs = [
@@ -478,6 +482,7 @@ with gr.Blocks(css=customCSS, theme=themeAlex) as demo:
     # history parallel zu chatbot speichern - da in chatbot bei Bildern zum Anzeigen in der GUI die Bilder speziell formatiert werden, 
     # für die Übergabe an die ki aber der Pfad zum Bild behalten werden muss - was in der history der Fall ist!
     history = gr.State([])
+    """
     uploaded_file_paths= gr.State([])
     history3 = gr.State([])
     uploaded_file_paths3= gr.State([])
@@ -493,9 +498,10 @@ with gr.Blocks(css=customCSS, theme=themeAlex) as demo:
     attached_file_history = gr.State(None)
     attached_file3 = gr.State(None)
     attached_file_history3 = gr.State(None)
+    """
     status_display = gr.State("")
-    status_display2 = gr.State("")
-    status_display3 = gr.State("")
+    #status_display2 = gr.State("")
+    #status_display3 = gr.State("")
     ################################################
     # Tab zum Chatbot mit Text oder Bildeingabe
     ################################################
@@ -660,9 +666,9 @@ with gr.Blocks(css=customCSS, theme=themeAlex) as demo:
     #2ter Tab
     #renew_button.click(fn=upload_pdf, inputs=upload_pdf_files, outputs=[output_text, file_list])
     # Hochladen der Dateien und dann Vektorstore aktualisieren
-    renew_button.click(fn=upload_pdf, inputs=upload_pdf_files, outputs=[output_text, file_list]).then(
-            fn=update_vectorstore, inputs=None, outputs=output_text).then(
-        fn=reset_file_input, inputs=None, outputs=upload_pdf_files).then(fn=show_text_status, inputs=None, outputs=output_text) #.then(fn=hide_status, inputs=None, outputs=output_text, show_progress="hidden")
+    renew_button.click(fn=upload_pdf, inputs=[upload_pdf_files], outputs=[output_text, file_list]).then(
+            fn=update_vectorstore, inputs=status_display, outputs=[output_text, status_display]).then(
+        fn=reset_file_input, inputs=None, outputs=[upload_pdf_files]).then(fn=show_text_status, inputs=None, outputs=[output_text, status_display]) #.then(fn=hide_status, inputs=None, outputs=output_text, show_progress="hidden")
     demo.load(display_files, outputs=file_list)
 
 demo.title = "KKG-Suche"