precompute at start

Dockerfile

@@ -1,6 +1,6 @@
 # Use an official Miniconda image as the base
 FROM python:3.10.15-bullseye
-ENV PIP_DEFAULT_TIMEOUT=300
+ENV PIP_DEFAULT_TIMEOUT=500
 
 RUN apt-get update && \
   apt-get install -y \

main.py

@@ -1,6 +1,8 @@
 import mesop as me
 from rag_app.chat_utils import State, _make_style_chat_bubble_wrapper, _ROLE_ASSISTANT, on_chat_input, _make_chat_bubble_style, \
     on_click_submit_chat_msg, _STYLE_CHAT_BUBBLE_NAME, handle_pdf_upload
+from rag_app.rag_2 import check_if_exists, precompute_index
+
 
 _COLOR_BACKGROUND = me.theme_var("background")
 
@@ -20,6 +22,10 @@ _STYLE_APP_CONTAINER = me.Style(
     allowed_iframe_parents=["https://huggingface.co"]
 ))
 def app():
+    if not check_if_exists():
+        print("precomputing the index")
+        precompute_index()
+        
     state = me.state(State)
     with me.box(style=_STYLE_APP_CONTAINER):
         with me.box(style=me.Style(

rag_app/chat_utils.py

@@ -37,9 +37,9 @@ class State:
 
 
 def respond_to_chat(query: str, history: list[ChatMessage]):
-    if not check_if_exists():
-        print("computing the vector index and the BM 25 retriever which will later be used")
-        precompute_index()
+    # if not check_if_exists():
+    #     print("computing the vector index and the BM 25 retriever which will later be used")
+    #     precompute_index()
     
     assistant_message = ChatMessage(role=_ROLE_ASSISTANT)
     yield assistant_message