Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import streamlit as st
+from types import SimpleNamespace
 
 from langchain.embeddings import HuggingFaceInstructEmbeddings
 from langchain.vectorstores.faiss import FAISS
@@ -8,6 +9,7 @@ from huggingface_hub import snapshot_download
 from langchain import OpenAI
 from langchain import PromptTemplate
 
+ns = SimpleNamespace(counter=0)
 
 st.set_page_config(page_title="Talk2Book", page_icon="📖")
 
@@ -169,11 +171,16 @@ if ask:
         st.write(f"**{BOOK_NAME}:** Whoops looks like you forgot your API key buddy")
         st.stop()
     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 🥺👉🏻👈🏻"):
+        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:
+            msg = "Just one sec"
+        with st.spinner(msg):
             try:
                 answer, pages, extract = get_answer(question=user_input)
-            except:
-                st.write(f"**{BOOK_NAME}:** What\'s going on? That's not the right API key")
+            except Exception as exc:
+                st.write(f"**{BOOK_NAME}:**: {exc}")
                 st.stop()
 
     st.write(f"**{BOOK_NAME}:** {answer}")