alexkueck commited on
Commit
20f3cf9
·
verified ·
1 Parent(s): 40923e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +37 -1
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...