Update app.py

app.py

@@ -12,22 +12,24 @@ 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)
+    if 'cfg' not in st.session_state:
+        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']
+        })
+        st.session_state.cfg = cfg
+        st.session_state.vq = VectaraQuery(cfg.api_key, cfg.customer_id, cfg.corpus_ids)
 
+    cfg = st.session_state.cfg
+    vq = st.session_state.vq
     st.set_page_config(page_title=cfg.title, layout="wide")
 
     # left side content
@@ -73,5 +75,4 @@ def launch_bot():
         st.session_state.messages.append(message)
     
 if __name__ == "__main__":
-    vq = None
     launch_bot()