Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -86,14 +86,9 @@ def load_environment():
|
|
86 |
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
|
87 |
|
88 |
def preload_modtran_document():
|
89 |
-
|
90 |
-
|
91 |
-
|
92 |
-
uploaded_files = [file_obj]
|
93 |
-
|
94 |
-
raw_text = get_uploaded_text(uploaded_files)
|
95 |
-
text_chunks = get_text_chunks(raw_text)
|
96 |
-
vectorstore = get_vectorstore(text_chunks)
|
97 |
set_global_vectorstore(vectorstore)
|
98 |
st.session_state.chat_ready = True
|
99 |
|
@@ -268,6 +263,10 @@ def main():
|
|
268 |
if not st.session_state.chat_ready:
|
269 |
with st.spinner("Loading MODTRAN document..."):
|
270 |
preload_modtran_document()
|
|
|
|
|
|
|
|
|
271 |
st.session_state.chat_ready = True
|
272 |
st.success("MODTRAN User Manual loaded successfully!")
|
273 |
|
|
|
86 |
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
|
87 |
|
88 |
def preload_modtran_document():
|
89 |
+
global vectorstore_global
|
90 |
+
embeddings = GeminiEmbeddings()
|
91 |
+
vectorstore = FAISS.load_local("modtran_vectorstore", embeddings)
|
|
|
|
|
|
|
|
|
|
|
92 |
set_global_vectorstore(vectorstore)
|
93 |
st.session_state.chat_ready = True
|
94 |
|
|
|
263 |
if not st.session_state.chat_ready:
|
264 |
with st.spinner("Loading MODTRAN document..."):
|
265 |
preload_modtran_document()
|
266 |
+
raw_text = get_uploaded_text(uploaded_files)
|
267 |
+
text_chunks = get_text_chunks(raw_text)
|
268 |
+
st.session_state.vectorstore = get_vectorstore(text_chunks)
|
269 |
+
set_global_vectorstore(st.session_state.vectorstore)
|
270 |
st.session_state.chat_ready = True
|
271 |
st.success("MODTRAN User Manual loaded successfully!")
|
272 |
|