Update app.py
Browse files
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 |
-
|
57 |
-
|
58 |
-
|
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"
|
385 |
elif file_extension == ".docx":
|
386 |
-
upload_path = f"
|
387 |
else:
|
388 |
-
upload_path = f"
|
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):
|