Update app.py

app.py

@@ -296,13 +296,13 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
         #kein Bild hochgeladen -> auf Text antworten...
         status = "Antwort der Vektordatenbank"
         results, status = generate_text(prompt, chatbot, history,vektordatenbank, retriever, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3, top_k=3)
-        
+
         #in results sind die preprocessed Splits enthalten, dargestellt werden sollen die orginalen:
         relevant_docs_org=[]
         for result in results['relevant_docs']:
-            print("result.................."+str(result))
-            if result['id'] :
-                original_split = split_to_original_mapping[result['id']]
+            split_id = result.metadata.get("split_id")
+            if split_id:
+                original_split = split_to_original_mapping[split_id]
                 relevant_docs_org.append(original_split)