alexkueck commited on
Commit
28aefb0
1 Parent(s): 2a11e91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -53,9 +53,9 @@ hf_token = os.getenv("HF_READ")
53
  #max Anzahl der zurückgelieferten Dokumente
54
  ANZAHL_DOCS = 5
55
  PATH_WORK = "."
56
- #CHROMA_DIR = "/chroma/demo"
57
- #CHROMA_PDF = './chroma/demo/pdf'
58
- #CHROMA_WORD = './chroma/demo/word'
59
  #CHROMA_EXCEL = './chroma/demo/excel'
60
  #DOCS_DIR = "chroma/demo"
61
 
@@ -381,11 +381,11 @@ def upload_pdf(files):
381
  file_extension = os.path.splitext(filename)[1]
382
  # Bestimmen des Upload-Pfads basierend auf der Dateiendung
383
  if file_extension == ".pdf":
384
- upload_path = f"chroma/kkg/pdf/{filename}"
385
  elif file_extension == ".docx":
386
- upload_path = f"chroma/kkg/word/{filename}"
387
  else:
388
- upload_path = f"chroma/kkg/{filename}"
389
 
390
  # Entfernen der vorhandenen Datei, falls sie existiert
391
  if os.path.exists(upload_path):
 
53
  #max Anzahl der zurückgelieferten Dokumente
54
  ANZAHL_DOCS = 5
55
  PATH_WORK = "."
56
+ CHROMA_DIR = "chroma/demo"
57
+ CHROMA_DIR_PDF = 'chroma/demo/pdf'
58
+ CHROMA_DIR_WORD = 'chroma/demo/word'
59
  #CHROMA_EXCEL = './chroma/demo/excel'
60
  #DOCS_DIR = "chroma/demo"
61
 
 
381
  file_extension = os.path.splitext(filename)[1]
382
  # Bestimmen des Upload-Pfads basierend auf der Dateiendung
383
  if file_extension == ".pdf":
384
+ upload_path = f"{CHROMA_DIR_PDF}/{filename}"
385
  elif file_extension == ".docx":
386
+ upload_path = f"{CHROMA_DIR_WORD}/{filename}"
387
  else:
388
+ upload_path = f"{CHROMA_DIR}/{filename}"
389
 
390
  # Entfernen der vorhandenen Datei, falls sie existiert
391
  if os.path.exists(upload_path):