Update app.py

app.py

@@ -296,10 +296,10 @@ 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']:
+        for result in extract_document_info(results['relevant_docs']):
             original_split = split_to_original_mapping[result['id']]
             relevant_docs_org.append(original_split)