fix bug in predict

app.py

@@ -174,17 +174,22 @@ def predict(message, history, request: gr.Request):
 
     messages.append(AIMessage(content=gpt_response.content))
 
-    raw_source_lines = [
-    f"{i+1} : {doc.metadata["source"].split("/")[-1]} (page {doc.metadata['page_label']}/{doc.metadata['total_pages']})\n---"
-    for i, doc in enumerate(relevant_docs)]
-
-    seen = set()
-    unique_source_lines = []
-    for line in raw_source_lines:
-        if line not in seen:          
-            seen.add(line)
-            unique_source_lines.append(line)
-    source_context = "\nSources:\n" + "\n".join(unique_source_lines)
+    try :
+        raw_source_lines = [
+        f"{i+1} : {doc.metadata["source"].split("/")[-1]} (page {doc.metadata['page_label']}/{doc.metadata['total_pages']})\n---"
+        for i, doc in enumerate(relevant_docs)]
+
+        seen = set()
+        unique_source_lines = []
+        for line in raw_source_lines:
+            if line not in seen:          
+                seen.add(line)
+                unique_source_lines.append(line)
+        source_context = "\nSources:\n" + "\n".join(unique_source_lines)
+
+    except :
+        source_context = "\n".join([
+        f"{doc.metadata["source"]}" for i, doc in enumerate(relevant_docs)])
 
     messages.append(AIMessage(content=source_context))
     print(gpt_response.content )