Update app.py
Browse files
app.py
CHANGED
@@ -391,32 +391,32 @@ async def upload_pdf(file):
|
|
391 |
|
392 |
|
393 |
"""
|
394 |
-
def upload_pdf(
|
395 |
-
if
|
396 |
return None, "Keine Datei hochgeladen."
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
397 |
|
398 |
-
|
399 |
-
|
400 |
|
401 |
-
|
402 |
-
|
403 |
-
|
404 |
-
|
405 |
-
upload_path = f"chroma/kkg/pdf/{filename}"
|
406 |
-
elif file_extension == ".docx":
|
407 |
-
upload_path = f"chroma/kkg/word/{filename}"
|
408 |
-
else:
|
409 |
-
upload_path = f"chroma/kkg/{filename}"
|
410 |
-
|
411 |
-
# Datei zum Hugging Face Space hochladen
|
412 |
-
upload_file_to_huggingface(file.name, upload_path)
|
413 |
-
|
414 |
-
############################################
|
415 |
-
#Vektorstore neu....
|
416 |
-
############################################
|
417 |
-
#create_vectorstore()
|
418 |
|
419 |
-
return f"
|
420 |
|
421 |
|
422 |
#Vektorstore erst auf ButtonKlick erneuern, vorher eventuell mehrere Dateien hochladen
|
@@ -589,14 +589,14 @@ with gr.Blocks(css=customCSS, theme=themeAlex) as demo:
|
|
589 |
|
590 |
|
591 |
with gr.Tab("Datei hochladen"):
|
592 |
-
|
593 |
output_text = gr.Textbox(label="Status")
|
594 |
renew_button = gr.Button("Vektorstore erneuern", elem_id="renew_button")
|
595 |
file_list = gr.HTML(elem_id="file_list", show_label=False)
|
596 |
|
597 |
# Automatisches Ausführen der Upload-Funktion, wenn eine Datei hochgeladen wird
|
598 |
#upload_pdf_file.change(fn=upload_pdf, inputs=upload_pdf_file, outputs=output_text)
|
599 |
-
|
600 |
renew_button.click(fn=renew_vectorstore, outputs=output_text)
|
601 |
#gr.HTML(update=display_files, elem_id="file_list", show_label=False)
|
602 |
demo.load(display_files, outputs=file_list)
|
|
|
391 |
|
392 |
|
393 |
"""
|
394 |
+
def upload_pdf(files):
|
395 |
+
if files is None:
|
396 |
return None, "Keine Datei hochgeladen."
|
397 |
+
for file in files:
|
398 |
+
# Extrahieren des Dateinamens aus dem vollen Pfad
|
399 |
+
filename = os.path.basename(file.name)
|
400 |
+
|
401 |
+
# Extrahieren der Dateiendung
|
402 |
+
file_extension = os.path.splitext(filename)[1]
|
403 |
+
# Bestimmen des Upload-Pfads basierend auf der Dateiendung
|
404 |
+
if file_extension == ".pdf":
|
405 |
+
upload_path = f"chroma/kkg/pdf/{filename}"
|
406 |
+
elif file_extension == ".docx":
|
407 |
+
upload_path = f"chroma/kkg/word/{filename}"
|
408 |
+
else:
|
409 |
+
upload_path = f"chroma/kkg/{filename}"
|
410 |
|
411 |
+
# Datei zum Hugging Face Space hochladen
|
412 |
+
upload_file_to_huggingface(file.name, upload_path)
|
413 |
|
414 |
+
############################################
|
415 |
+
#Vektorstore neu....
|
416 |
+
############################################
|
417 |
+
#create_vectorstore()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
418 |
|
419 |
+
return f"Dateien erfolgreich hochgeladen."
|
420 |
|
421 |
|
422 |
#Vektorstore erst auf ButtonKlick erneuern, vorher eventuell mehrere Dateien hochladen
|
|
|
589 |
|
590 |
|
591 |
with gr.Tab("Datei hochladen"):
|
592 |
+
upload_pdf_files = gr.Files(label="PDF- oder Word-Dateien hochladen", file_count="multiple")
|
593 |
output_text = gr.Textbox(label="Status")
|
594 |
renew_button = gr.Button("Vektorstore erneuern", elem_id="renew_button")
|
595 |
file_list = gr.HTML(elem_id="file_list", show_label=False)
|
596 |
|
597 |
# Automatisches Ausführen der Upload-Funktion, wenn eine Datei hochgeladen wird
|
598 |
#upload_pdf_file.change(fn=upload_pdf, inputs=upload_pdf_file, outputs=output_text)
|
599 |
+
upload_pdf_files.change(fn=upload_pdf, inputs=upload_pdf_file, outputs=[output_text, file_list], postprocess=display_files)
|
600 |
renew_button.click(fn=renew_vectorstore, outputs=output_text)
|
601 |
#gr.HTML(update=display_files, elem_id="file_list", show_label=False)
|
602 |
demo.load(display_files, outputs=file_list)
|