Update app.py
Browse files
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 |
-
|
365 |
-
|
366 |
-
return
|
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
385 |
|
386 |
-
|
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
|