Update app.py

app.py

@@ -8,6 +8,7 @@ from langchain.chains import VectorDBQA
 from huggingface_hub import snapshot_download
 from langchain import OpenAI
 from langchain import PromptTemplate
+from loguru import logger
 
 ns = SimpleNamespace(counter=0)
 
@@ -27,9 +28,6 @@ with st.sidebar:
 st.title(f"Talk2Book: {BOOK_NAME}")
 st.markdown(f"#### Have a conversation with {BOOK_NAME} by {AUTHOR_NAME} 🙊")
 
-
-
-
 ##### functionss ####
 @st.experimental_singleton(show_spinner=False)
 def load_vectorstore():
@@ -171,14 +169,15 @@ if ask:
         st.write(f"**{BOOK_NAME}:** Whoops looks like you forgot your API key buddy")
         st.stop()
     else:
-        if ns.counter:           
-            msg = "Um... excuse me but... this can take about a minute for your first question because some stuff have to be downloaded 🥺👉🏻👈🏻"
-            ns.counter = 1
-        else:
+        if ns.counter: 
             msg = "Just one sec"
+        else:
+            msg = "Um... excuse me but... this can take about a minute for your first question because some stuff have to be downloaded 🥺👉🏻👈🏻"
+            ns.counter = 1            
         with st.spinner(msg):
             try:
                 answer, pages, extract = get_answer(question=user_input)
+                logger.info(f"answer: {answer}")
             except Exception as exc:
                 st.write(f"**{BOOK_NAME}:**: {exc}")
                 st.stop()