alexkueck commited on
Commit
4e312ba
1 Parent(s): 0e1a7fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -28
app.py CHANGED
@@ -360,38 +360,35 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
360
  ########################################
361
  # Hochladen von Dateien und Vektorstore neu erstellen
362
  # Beispiel-Upload-PDF-Funktion
363
- """
364
- async def upload_pdf(file):
365
- if file is None:
366
- return None, "Keine Datei hochgeladen."
367
-
368
- # Extrahieren des Dateinamens aus dem vollen Pfad
369
- filename = os.path.basename(file.name)
370
 
371
- # Extrahieren der Dateiendung
372
- file_extension = os.path.splitext(filename)[1]
373
- # Bestimmen des Upload-Pfads basierend auf der Dateiendung
374
- if file_extension == ".pdf":
375
- upload_path = f"chroma/kkg/pdf/{filename}"
376
- elif file_extension == ".docx":
377
- upload_path = f"chroma/kkg/word/{filename}"
378
- else:
379
- upload_path = f"chroma/kkg/{filename}"
380
-
381
- # Asynchrones Hochladen der Datei
382
- loop = asyncio.get_event_loop()
383
  with ThreadPoolExecutor() as pool:
384
- await loop.run_in_executor(pool, upload_file_to_huggingface, file.name, upload_path)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
385
 
386
- # Asynchrones Erstellen des Vektorstores
387
- #vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING werden aktualisiert
388
- await loop.run_in_executor(pool, create_vectorstore)
389
-
390
- return f"PDF '{filename}' erfolgreich hochgeladen."
391
 
392
-
393
  """
394
-
395
  def upload_pdf(files):
396
  if not files:
397
  return "Keine Dateien hochgeladen.", display_files()
@@ -421,7 +418,7 @@ def upload_pdf(files):
421
  logging.error(f"Error uploading file {file.name}: {e}")
422
 
423
  return f"Dateien erfolgreich hochgeladen.", display_files()
424
-
425
 
426
 
427
  #Vektorstore erst auf ButtonKlick erneuern, vorher eventuell mehrere Dateien hochladen
 
360
  ########################################
361
  # Hochladen von Dateien und Vektorstore neu erstellen
362
  # Beispiel-Upload-PDF-Funktion
363
+ def upload_pdfs(files):
364
+ if not files:
365
+ logging.warning("No files uploaded.")
366
+ return "Keine Dateien hochgeladen.", display_files()
 
 
 
367
 
 
 
 
 
 
 
 
 
 
 
 
 
368
  with ThreadPoolExecutor() as pool:
369
+ for file in files:
370
+ try:
371
+ # Extrahieren des Dateinamens aus dem vollen Pfad
372
+ filename = os.path.basename(file.name)
373
+
374
+ # Extrahieren der Dateiendung
375
+ file_extension = os.path.splitext(filename)[1]
376
+ # Bestimmen des Upload-Pfads basierend auf der Dateiendung
377
+ if file_extension == ".pdf":
378
+ upload_path = f"chroma/kkg/pdf/{filename}"
379
+ elif file_extension == ".docx":
380
+ upload_path = f"chroma/kkg/word/{filename}"
381
+ else:
382
+ upload_path = f"chroma/kkg/{filename}"
383
+
384
+ # Hochladen der Datei
385
+ pool.submit(upload_file_to_huggingface, file.name, upload_path)
386
+ except Exception as e:
387
+ logging.error(f"Error uploading file {file.name}: {e}")
388
 
389
+ return "Dateien wurden erfolgreich hochgeladen.", display_files()
 
 
 
 
390
 
 
391
  """
 
392
  def upload_pdf(files):
393
  if not files:
394
  return "Keine Dateien hochgeladen.", display_files()
 
418
  logging.error(f"Error uploading file {file.name}: {e}")
419
 
420
  return f"Dateien erfolgreich hochgeladen.", display_files()
421
+ """
422
 
423
 
424
  #Vektorstore erst auf ButtonKlick erneuern, vorher eventuell mehrere Dateien hochladen