Update app.py

app.py

@@ -15,18 +15,17 @@ def launch_bot():
 
     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']
+            'source_data_desc': os.environ['source_data_desc'],
+            'prompt_name': os.environ.get('prompt_name', None)
         })
         st.session_state.cfg = cfg
-        st.session_state.vq = VectaraQuery(cfg.api_key, cfg.customer_id, cfg.corpus_ids)
+        st.session_state.vq = VectaraQuery(cfg.api_key, cfg.customer_id, cfg.corpus_ids, cfg.prompt_name)
 
     cfg = st.session_state.cfg
     vq = st.session_state.vq
@@ -36,7 +35,7 @@ def launch_bot():
     with st.sidebar:
         image = Image.open('Vectara-logo.png')
         st.markdown(f"## Welcome to {cfg.title}\n\n"
-                    f"With this demo uses Retrieval Augmented Generation to ask questions about {cfg.source_data_desc}\n\n")
+                    f"This demo uses Retrieval Augmented Generation to ask questions about {cfg.source_data_desc}\n\n")
 
         st.markdown("---")
         st.markdown(
@@ -76,3 +75,4 @@ def launch_bot():
     
 if __name__ == "__main__":
     launch_bot()
+