Update utils.py

utils.py

@@ -267,11 +267,23 @@ def rag_chain(prompt, db, k=3):
     rag_template = "Nutze ausschließlich die folgenden Kontext Teile am Ende, um die Frage zu beantworten . " + template + "Frage: " + prompt + "Kontext Teile: "
     retrieved_chunks  = db.similarity_search(prompt, k)
 
-    neu_prompt = rag_template 
+    # Erstelle ein Dictionary für die Chunks
+    chunks_dict = []
     for i, chunk in enumerate(retrieved_chunks):
-        neu_prompt += f"{i+1}. {chunk}\n"
+        chunk_dict = {
+            "chunk_index": i + 1,
+            "page_content": chunk.page_content,  # assuming chunk has page_content attribute
+            "metadata": chunk.metadata  # assuming chunk has metadata attribute
+        }
+        chunks_dict.append(chunk_dict)
+
+    # Erstelle das neue Prompt
+    neu_prompt = rag_template
+    for chunk in chunks_dict:
+        neu_prompt += f"{chunk['chunk_index']}. {chunk['page_content']}\n"
+
+    return neu_prompt, chunks_dict  # returning both the new prompt and the dictionary
 
-    return neu_prompt