ZarinT commited on
Commit
e12828c
·
verified ·
1 Parent(s): 4e8ef13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -8
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
- with open("MODTRAN 6 User's Manual.pdf", "rb") as f:
90
- file_obj = io.BytesIO(f.read())
91
- file_obj.name = "MODTRAN 6 User's Manual.pdf"
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