alexkueck commited on
Commit
aa90626
·
verified ·
1 Parent(s): ab259c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -24
app.py CHANGED
@@ -391,32 +391,32 @@ async def upload_pdf(file):
391
 
392
 
393
  """
394
- def upload_pdf(file):
395
- if file is None:
396
  return None, "Keine Datei hochgeladen."
 
 
 
 
 
 
 
 
 
 
 
 
 
397
 
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"PDF '{filename}' erfolgreich hochgeladen."
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
- upload_pdf_file = gr.File(label="PDF- oder Word-Datei hochladen")
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_file.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)
 
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)