Update app/main_IO.py

app/main_IO.py

@@ -62,6 +62,26 @@ def reset_session_state_on_upload():
             st.session_state[key] = default_val
 
 
+# Define model path - check multiple possible locations
+def get_model_path():
+    # Try local model first (relative to current script)
+    local_model_path = os.path.join(os.path.dirname(__file__), "chromadb_model")
+    
+    # Try app root model folder
+    app_model_path = os.path.join(root_path, "models", EMBEDDING_MODEL)
+    
+    # Try current working directory
+    cwd_model_path = os.path.join(os.getcwd(), "chromadb_model")
+    
+    # Check which path exists
+    for path in [local_model_path, app_model_path, cwd_model_path]:
+        if os.path.exists(path):
+            return path
+    
+    # If no local model found, return None to download
+    return None
+    
+
 def upload_pdf():
     uploaded_file = st.file_uploader("Upload your PDF", type=["pdf"], label_visibility="collapsed")