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...
|