add another book: almanac of naval

app.py

@@ -9,10 +9,12 @@ from langchain import OpenAI
 from langchain import PromptTemplate
 
 
-BOOK_NAME = "1984"
-AUTHOR_NAME = "George Orwell"
+st.set_page_config(page_title="Talk2Book", page_icon="📖")
+
+# set later in the sidebar section
+BOOK_NAME = st.empty()
+AUTHOR_NAME = st.empty()
 
-st.set_page_config(page_title="Talk2Book: 1984", page_icon="📖")
 st.title(f"Talk2Book: {BOOK_NAME}")
 st.markdown(f"#### Have a conversation with {BOOK_NAME} by {AUTHOR_NAME} 🙊")
 
@@ -23,18 +25,18 @@ st.markdown(f"#### Have a conversation with {BOOK_NAME} by {AUTHOR_NAME} 🙊")
 @st.experimental_singleton(show_spinner=False)
 def load_vectorstore():
     # download from hugging face
-    snapshot_download(repo_id="calmgoose/orwell-1984_faiss-instructembeddings",
+    cache_dir=f"{BOOK_NAME}_cache"
+    snapshot_download(repo_id="calmgoose/book-embeddings",
                                     repo_type="dataset",
                                     revision="main",
-                                    allow_patterns="vectorstore/*",
-                                    cache_dir="orwell_faiss",
+                                    allow_patterns=f"books/{BOOK_NAME}/*",
+                                    cache_dir=cache_dir,
                                     )
 
-    dir = "orwell_faiss"
-    target_dir = "vectorstore"
+    target_dir = f"books/{BOOK_NAME}/*"
 
     # Walk through the directory tree recursively
-    for root, dirs, files in os.walk(dir):
+    for root, dirs, files in os.walk(cache_dir):
         # Check if the target directory is in the list of directories
         if target_dir in dirs:
             # Get the full path of the target directory
@@ -124,6 +126,13 @@ def get_answer(question):
 
 ##### sidebar ####
 with st.sidebar:
+    book = st.radio("Choose book: ",
+                   ["1984 - George Orwell", "The Almanac of Naval Ravikant - Eric Jorgenson"]
+                   )
+    # update st.empty() variables from earlier
+    BOOK_NAME = book.replace.split("-")[0][:-1] # "1984 - George Orwell" -> "1984"
+    AUTHOR_NAME = book.replace.split("-")[1][1:] # "1984 - George Orwell" -> "George Orwell"
+    
     api_key = st.text_input(label = "Paste your OpenAI API key here to get started", 
                             type = "password",
                             help = "This isn't saved 🙈"