cache model

app.py

@@ -8,6 +8,14 @@ def extract_text_from_docx(file_path):
     doc = Document(file_path)
     return " ".join([para.text.strip() for para in doc.paragraphs if para.text.strip()])
 
+@st.cache_resource()
+def load_chatbot():
+    with st.spinner("Loading project information..."):
+        doc1_text = extract_text_from_docx("Files/ZEGA AI Capabilities Overview for Clients (1).docx")
+        doc2_text = extract_text_from_docx("Files/ZEGA AI Document Capabilities Overview for Clients (1).docx")
+    
+    return GeminiQanA(doc1_text, doc2_text)
+
 # Streamlit App Configuration
 st.set_page_config(page_title="Zega AI Sales Agent", page_icon="🤖", layout="centered")
 
@@ -28,7 +36,7 @@ with st.spinner("Loading project information..."):
     doc2_text = extract_text_from_docx("Files/ZEGA AI Document Capabilities Overview for Clients (1).docx")
 
 # Initialize chatbot
-chatbot = GeminiQanA(doc1_text, doc2_text)
+chatbot = load_chatbot()
 
 # Initialize chat session
 if "messages" not in st.session_state: