added expander

app.py

@@ -152,20 +152,20 @@ def retriever(query: str, selected_retriever: str):
         st.session_state.last_expander_id += 1
         return f"expander_{st.session_state.last_expander_id}"
 
+    print(f"Topk documents: {topk_documents}")
+
     for idx, document in enumerate(topk_documents):
         try:
             a_idx = document.lower().index("abstract:")
             title = document[:a_idx].replace('\n', ' ')
             doc_details = document[a_idx:]
-        except:
-            print(f"Document parsing error for title and other contents-\n{document}")
+        except Exception as e:
+            print(f"Document parsing error for title and other contents-\n{document}, \n\nError: {e}")
             title = "Document title"
             doc_details = document
-        try:
-            expander = st.expander(f"Rank {ix+1} - {title}")
-            expander.write(doc_details)
-        except:
-            print(f"Error in expander {idx} logic")
+
+        with st.expander(f"Rank {idx+1} - {title.replace("Title:", "")}"):
+            st.write(doc_details)
 
     # retrieved_document_titles = ""
     # for document in topk_documents: