add an expandable sources section

app.py

@@ -82,18 +82,32 @@ def get_answer(question):
     chain = load_chain()
     result = chain({"query": question})
 
+    answer = result["result"]
+
     # format sources
-    unique_sources = set()
 
+    # pages
+    unique_sources = set()
     for item in result['source_documents']:
         unique_sources.add(item.metadata['page'])
 
-    sources_string = ""
-
+    unique_pages = ""
     for item in unique_sources:
-        sources_string += str(item) + ", "
+        unique_pages += str(item) + ", "
+
+    pages = unique_pages # will look like 1, 2, 3,
+
+    # source text
+    full_source = ""
+    for item in result['source_documents']:
+        full_source += f"- **Page: {item.metadata['page']}**" + "\n" + item.page_content + "\n\n"
 
-    return result["result"] + "\n\n" + " - From pages: " + sources_string
+    # will look like:
+    # - Page: {number}
+    #  {extracted text from book}
+    extract = full_source
+
+    return answer, pages, extract
 
 
 # From here down is all the StreamLit UI.
@@ -103,12 +117,13 @@ st.markdown("#### Have a conversaion with 1984 by George Orwell 🙊")
 
 with st.sidebar:
     api_key = st.text_input(label = "Paste your OpenAI API key here to get started", type = "password")
-
-    # if api_key:
     os.environ["OPENAI_API_KEY"] = api_key
-
     st.info("This isn't saved 🙈")
 
+    st.markdown("---")
+
+    st.write("Based on [Talk2Book](https://github.com/batmanscode/Talk2Book)")
+
 # streamlit-chat not working
 # i get this error: https://discuss.streamlit.io/t/your-app-is-having-trouble-loading-the-xxx-component/25046
 # if "generated" not in st.session_state:
@@ -135,15 +150,23 @@ ask = col2.button("Ask")
 if ask:
 
     if api_key is "":
-        output = "Whoops looks like you forgot your API key buddy"
+        # output = "Whoops looks like you forgot your API key buddy"
+        st.write("**1984:** Whoops looks like you forgot your API key buddy")
+        break
     else:
         with st.spinner("Um... excuse me but... this can take about a minute for your first question because some stuff have to be downloaded 🥺👉🏻👈🏻"):
             try:
-                output = get_answer(question=user_input)
+                answer, pages, extract = get_answer(question=user_input)
             except:
-                output = "What's going on? That's not the right API key"
+                # output = "What's going on? That's not the right API key"
+                st.write("**1984:** What\'s going on? That's not the right API key")
+                break
+
+    st.write(f"**1984:** {answer}")
 
-    st.write(f"**1984:** {output}")
+    # sources
+    with st.expander(label = f"From pages: {pages}", expanded = False):
+        st.markdown(extract)
 
 # streamlit-chat
 #     st.session_state.past.append(user_input)