add expander

app.py

@@ -144,32 +144,42 @@ def retriever(query: str, selected_retriever: str):
 
     corpus = '\n\n'.join(topk_documents)
 
-    retrieved_document_titles = ""
-    for document in topk_documents:
+    for idx, document in topk_documents:
         try:
             idx = document.lower().index("abstract:")
             title = document[:idx].replace('\n', ' ')
-            retrieved_document_titles = retrieved_document_titles + title + ' ||| '
+            doc_details = document[idx:]
+            with st.expander(f"Rank {ix+1} - {title}"):
+                st.write(f"doc_details")
         except:
-            retrieved_document_titles = retrieved_document_titles + document + ' ||| '
-
-    display_retrieved_documents = f"Retrieved Documents - ||| {retrieved_document_titles}"
-
-    with st.chat_message("assistant"):
-        placeholder = st.empty()
-        text = ""
-
-        print(display_retrieved_documents)
-
-        for chunk in display_retrieved_documents.split():
-            if chunk == "|||":
-                text += "  \n" + "  \n"
-            else:
-                text += chunk + " "
-            time.sleep(0.05)
-            # Add a blinking cursor to simulate typing
-            placeholder.markdown(text + "▌")
-        placeholder.markdown(text)
+            pass
+
+    # retrieved_document_titles = ""
+    # for document in topk_documents:
+    #     try:
+    #         idx = document.lower().index("abstract:")
+    #         title = document[:idx].replace('\n', ' ')
+    #         retrieved_document_titles = retrieved_document_titles + title + ' ||| '
+    #     except:
+    #         retrieved_document_titles = retrieved_document_titles + document + ' ||| '
+
+    # display_retrieved_documents = f"Retrieved Documents - ||| {retrieved_document_titles}"
+
+    # with st.chat_message("assistant"):
+    #     placeholder = st.empty()
+    #     text = ""
+
+    #     print(display_retrieved_documents)
+
+    #     for chunk in display_retrieved_documents.split():
+    #         if chunk == "|||":
+    #             text += "  \n" + "  \n"
+    #         else:
+    #             text += chunk + " "
+    #         time.sleep(0.05)
+    #         # Add a blinking cursor to simulate typing
+    #         placeholder.markdown(text + "▌")
+    #     placeholder.markdown(text)
 
     # You could return retrieved info here.
     return corpus