Update app.py

app.py

@@ -326,13 +326,19 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
             status = "Antwort der Vektordatenbank"
             results, status = generate_text(prompt, chatbot, history, 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']:
-                split_id = result.get("metadata", {}).get("split_id")
-                if split_id:
-                    original_split = SPLIT_TO_ORIGINAL_MAPPING[split_id]
-                    relevant_docs_org.append(original_split)
+            # Überprüfen, ob relevante Dokumente gefunden wurden
+            if results['relevant_docs']:
+                # in results sind die preprocessed Splits enthalten, dargestellt werden sollen die originalen:
+                relevant_docs_org = []
+                for result in results['relevant_docs']:
+                    split_id = result.get("metadata", {}).get("split_id")
+                    if split_id:
+                        original_split = SPLIT_TO_ORIGINAL_MAPPING[split_id]
+                        relevant_docs_org.append(original_split)
+            else:
+                # Keine relevanten Dokumente gefunden
+                status = "Keine relevanten Dokumente gefunden."
+                relevant_docs_org = []
     
             relevant_docs = extract_document_info(relevant_docs_org)