Update app.py
Browse files
app.py
CHANGED
|
@@ -334,7 +334,37 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
|
|
| 334 |
return chatbot, history, None, file_history, "Erst validieren oder einen Prompt eingeben!"
|
| 335 |
|
| 336 |
########################################
|
| 337 |
-
# Hochladen von Dateien
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 338 |
def upload_pdf(file):
|
| 339 |
if file is None:
|
| 340 |
return None, "Keine Datei hochgeladen."
|
|
@@ -361,6 +391,12 @@ def upload_pdf(file):
|
|
| 361 |
_ = create_vektorstore()
|
| 362 |
|
| 363 |
return f"PDF '{filename}' erfolgreich hochgeladen."
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 364 |
|
| 365 |
########################################
|
| 366 |
# Bot- test gegen schädliche Bots die die Anwendung testen...
|
|
|
|
| 334 |
return chatbot, history, None, file_history, "Erst validieren oder einen Prompt eingeben!"
|
| 335 |
|
| 336 |
########################################
|
| 337 |
+
# Hochladen von Dateien und Vektorstore neu erstellen
|
| 338 |
+
# Beispiel-Upload-PDF-Funktion
|
| 339 |
+
async def upload_pdf(file):
|
| 340 |
+
if file is None:
|
| 341 |
+
return None, "Keine Datei hochgeladen."
|
| 342 |
+
|
| 343 |
+
# Extrahieren des Dateinamens aus dem vollen Pfad
|
| 344 |
+
filename = os.path.basename(file.name)
|
| 345 |
+
|
| 346 |
+
# Extrahieren der Dateiendung
|
| 347 |
+
file_extension = os.path.splitext(filename)[1]
|
| 348 |
+
# Bestimmen des Upload-Pfads basierend auf der Dateiendung
|
| 349 |
+
if file_extension == ".pdf":
|
| 350 |
+
upload_path = f"chroma/kkg/pdf/{filename}"
|
| 351 |
+
elif file_extension == ".docx":
|
| 352 |
+
upload_path = f"chroma/kkg/word/{filename}"
|
| 353 |
+
else:
|
| 354 |
+
upload_path = f"chroma/kkg/{filename}"
|
| 355 |
+
|
| 356 |
+
# Asynchrones Hochladen der Datei
|
| 357 |
+
loop = asyncio.get_event_loop()
|
| 358 |
+
with ThreadPoolExecutor() as pool:
|
| 359 |
+
await loop.run_in_executor(pool, upload_file_to_huggingface, file.name, upload_path)
|
| 360 |
+
|
| 361 |
+
# Asynchrones Erstellen des Vektorstores
|
| 362 |
+
await loop.run_in_executor(pool, create_vectorstore)
|
| 363 |
+
|
| 364 |
+
return f"PDF '{filename}' erfolgreich hochgeladen."
|
| 365 |
+
|
| 366 |
+
|
| 367 |
+
"""
|
| 368 |
def upload_pdf(file):
|
| 369 |
if file is None:
|
| 370 |
return None, "Keine Datei hochgeladen."
|
|
|
|
| 391 |
_ = create_vektorstore()
|
| 392 |
|
| 393 |
return f"PDF '{filename}' erfolgreich hochgeladen."
|
| 394 |
+
"""
|
| 395 |
+
|
| 396 |
+
|
| 397 |
+
|
| 398 |
+
|
| 399 |
+
|
| 400 |
|
| 401 |
########################################
|
| 402 |
# Bot- test gegen schädliche Bots die die Anwendung testen...
|