Update app.py

app.py

@@ -7,26 +7,27 @@ import os
 import streamlit as st
 from PIL import Image
 
-corpus_ids = str(os.environ['corpus_ids']).split(',')
-questions = list(eval(os.environ['examples']))
-
-cfg = OmegaConf.create({
-    'customer_id': str(os.environ['customer_id']),
-    'corpus_ids': corpus_ids,
-    'api_key': str(os.environ['api_key']),
-    'title': os.environ['title'],
-    'description': os.environ['description'],
-    'examples': questions,
-    'source_data_desc': os.environ['source_data_desc']
-})
-
-vq = VectaraQuery(cfg.api_key, cfg.customer_id, cfg.corpus_ids)
 
 def launch_bot():
     def generate_response(question):
         response = vq.submit_query(question)
         return response
     
+    corpus_ids = str(os.environ['corpus_ids']).split(',')
+    questions = list(eval(os.environ['examples']))
+    cfg = OmegaConf.create({
+        'customer_id': str(os.environ['customer_id']),
+        'corpus_ids': corpus_ids,
+        'api_key': str(os.environ['api_key']),
+        'title': os.environ['title'],
+        'description': os.environ['description'],
+        'examples': questions,
+        'source_data_desc': os.environ['source_data_desc']
+    })
+
+    if vq is None:
+        vq = VectaraQuery(cfg.api_key, cfg.customer_id, cfg.corpus_ids)
+
     st.set_page_config(page_title=cfg.title, layout="wide")
 
     # left side content
@@ -72,4 +73,5 @@ def launch_bot():
         st.session_state.messages.append(message)
     
 if __name__ == "__main__":
+    vq = None
     launch_bot()