Update utils.py
Browse files
utils.py
CHANGED
@@ -430,10 +430,14 @@ def query(api_llm, payload):
|
|
430 |
def extract_document_info(documents):
|
431 |
extracted_info = []
|
432 |
for doc in documents:
|
|
|
|
|
|
|
|
|
433 |
info = {
|
434 |
'content': doc.page_content,
|
435 |
'metadata': doc.metadata,
|
436 |
-
'titel':
|
437 |
'seite': doc.metadata.get("page", "Unbekannte Seite"),
|
438 |
'pfad': doc.metadata.get("path", "Kein Pfad verfügbar")
|
439 |
}
|
|
|
430 |
def extract_document_info(documents):
|
431 |
extracted_info = []
|
432 |
for doc in documents:
|
433 |
+
# Extract the filename from the path to use as the title
|
434 |
+
filename = os.path.basename(doc.metadata.get("path", ""))
|
435 |
+
title = filename if filename else "Keine Überschrift"
|
436 |
+
|
437 |
info = {
|
438 |
'content': doc.page_content,
|
439 |
'metadata': doc.metadata,
|
440 |
+
'titel': title,
|
441 |
'seite': doc.metadata.get("page", "Unbekannte Seite"),
|
442 |
'pfad': doc.metadata.get("path", "Kein Pfad verfügbar")
|
443 |
}
|