Update app.py
Browse files
app.py
CHANGED
@@ -369,67 +369,9 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
|
|
369 |
########################################
|
370 |
# Hochladen von Dateien und Vektorstore neu erstellen
|
371 |
# Beispiel-Upload-PDF-Funktion
|
372 |
-
"""
|
373 |
-
def upload_pdf(files):
|
374 |
-
if not files:
|
375 |
-
logging.warning("No files uploaded.")
|
376 |
-
status_message = " Keine Dateien zum Hochladen"
|
377 |
-
else:
|
378 |
-
futures = []
|
379 |
-
with ThreadPoolExecutor() as pool:
|
380 |
-
for file in files:
|
381 |
-
try:
|
382 |
-
# Extrahieren des Dateinamens aus dem vollen Pfad
|
383 |
-
filename = os.path.basename(file.name)
|
384 |
-
|
385 |
-
# Extrahieren der Dateiendung
|
386 |
-
file_extension = os.path.splitext(filename)[1]
|
387 |
-
# Bestimmen des Upload-Pfads basierend auf der Dateiendung
|
388 |
-
if file_extension == ".pdf":
|
389 |
-
upload_path = f"chroma/kkg/pdf/{filename}"
|
390 |
-
elif file_extension == ".docx":
|
391 |
-
upload_path = f"chroma/kkg/word/{filename}"
|
392 |
-
else:
|
393 |
-
upload_path = f"chroma/kkg/{filename}"
|
394 |
-
|
395 |
-
# Entfernen der vorhandenen Datei, falls sie existiert
|
396 |
-
if os.path.exists(upload_path):
|
397 |
-
os.remove(upload_path)
|
398 |
-
|
399 |
-
# Hochladen der Datei
|
400 |
-
future = pool.submit(upload_file_to_huggingface, file.name, upload_path)
|
401 |
-
futures.append(future)
|
402 |
-
except Exception as e:
|
403 |
-
print("Fehler bei file upload...............")
|
404 |
-
|
405 |
-
# Warten, bis alle Uploads abgeschlossen sind
|
406 |
-
for future in futures:
|
407 |
-
try:
|
408 |
-
future.result()
|
409 |
-
status_message = "Dateien fertig hochgeladen"
|
410 |
-
except Exception as e:
|
411 |
-
logging.error(f"Error during file upload: {e}")
|
412 |
-
status_message = "Fehler beim Hochladen der Dateien."
|
413 |
-
|
414 |
-
# Nachdem alle Dateien hochgeladen wurden, den Vektorstore neu laden
|
415 |
-
try:
|
416 |
-
############################################
|
417 |
-
#Vektorstore neu....
|
418 |
-
############################################
|
419 |
-
#create_vectorstore()
|
420 |
-
logging.info("Vektorstore successfully renewed.")
|
421 |
-
status_message += " - Vektorstore wurde erneuert."
|
422 |
-
except Exception as e:
|
423 |
-
logging.error(f"Error renewing Vektorstore: {e}")
|
424 |
-
status_message += " - Fehler beim Erneuern des Vektorstores."
|
425 |
-
|
426 |
-
return status_message, display_files()
|
427 |
-
"""
|
428 |
-
|
429 |
def upload_pdf(files):
|
430 |
status_message = ""
|
431 |
if not files:
|
432 |
-
logging.warning("No files uploaded.")
|
433 |
status_message = " Keine Dateien zum Hochladen"
|
434 |
else:
|
435 |
for file in files:
|
@@ -469,10 +411,8 @@ def update_vectorstore():
|
|
469 |
#Vektorstore neu....
|
470 |
############################################
|
471 |
create_vectorstore()
|
472 |
-
logging.info("Vektorstore successfully renewed.")
|
473 |
status_message = "Vektorstore wurde erneuert."
|
474 |
except Exception as e:
|
475 |
-
logging.error(f"Error renewing Vektorstore: {e}")
|
476 |
status_message = "Fehler beim Erneuern des Vektorstores."
|
477 |
|
478 |
|
@@ -642,28 +582,6 @@ with gr.Blocks(css=customCSS, theme=themeAlex) as demo:
|
|
642 |
renew_button = gr.Button("Dateien hochladen und System aktualisieren", elem_id="renew_button")
|
643 |
file_list = gr.HTML(elem_id="file_list", show_label=False)
|
644 |
|
645 |
-
# Automatisches Ausführen der Upload-Funktion, wenn eine Datei hochgeladen wird
|
646 |
-
#upload_pdf_file.change(fn=upload_pdf, inputs=upload_pdf_file, outputs=output_text)
|
647 |
-
#upload_pdf_files.change(fn=upload_pdf, inputs=upload_pdf_files, outputs=[output_text, file_list]) #, postprocess=display_files)
|
648 |
-
#renew_button.click(fn=upload_pdf, inputs=upload_pdf_files, outputs=[output_text, file_list])
|
649 |
-
#gr.HTML(update=display_files, elem_id="file_list", show_label=False)
|
650 |
-
#demo.load(display_files, outputs=file_list)
|
651 |
-
|
652 |
-
|
653 |
-
"""
|
654 |
-
with gr.Tab("Datei hochladen"):
|
655 |
-
upload_pdf_file = gr.File(label="PDF- oder Word-Datei hochladen")
|
656 |
-
output_text = gr.Textbox(label="Status")
|
657 |
-
#upload_button = gr.Button("Datei hochladen")
|
658 |
-
file_list = gr.HTML(elem_id="file_list", show_label=False)
|
659 |
-
|
660 |
-
#upload_button.click(fn=upload_pdf, inputs=upload_pdf_file, outputs=output_text)
|
661 |
-
# Automatisches Ausführen der Upload-Funktion, wenn eine Datei hochgeladen wird
|
662 |
-
upload_pdf_file.change(fn=upload_pdf, inputs=upload_pdf_file, outputs=output_text)
|
663 |
-
#gr.HTML(update=display_files, elem_id="file_list", show_label=False)
|
664 |
-
demo.load(display_files, outputs=file_list)
|
665 |
-
"""
|
666 |
-
|
667 |
gr.Markdown(description)
|
668 |
|
669 |
######################################
|
|
|
369 |
########################################
|
370 |
# Hochladen von Dateien und Vektorstore neu erstellen
|
371 |
# Beispiel-Upload-PDF-Funktion
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
372 |
def upload_pdf(files):
|
373 |
status_message = ""
|
374 |
if not files:
|
|
|
375 |
status_message = " Keine Dateien zum Hochladen"
|
376 |
else:
|
377 |
for file in files:
|
|
|
411 |
#Vektorstore neu....
|
412 |
############################################
|
413 |
create_vectorstore()
|
|
|
414 |
status_message = "Vektorstore wurde erneuert."
|
415 |
except Exception as e:
|
|
|
416 |
status_message = "Fehler beim Erneuern des Vektorstores."
|
417 |
|
418 |
|
|
|
582 |
renew_button = gr.Button("Dateien hochladen und System aktualisieren", elem_id="renew_button")
|
583 |
file_list = gr.HTML(elem_id="file_list", show_label=False)
|
584 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
585 |
gr.Markdown(description)
|
586 |
|
587 |
######################################
|