Update app.py

app.py

@@ -3,16 +3,14 @@ from query import VectaraQuery
 import os
 import gradio as gr
 
-
 def isTrue(x) -> bool:
     if isinstance(x, bool):
         return x
     return x.strip().lower() == 'true'
 
-corpus_ids = str(os.environ['corpus_ids']).split(',')
+corpus_keys = str(os.environ['corpus_keys']).split(',')
 cfg = OmegaConf.create({
-    'customer_id': str(os.environ['customer_id']),
-    'corpus_ids': corpus_ids,
+    'corpus_keys': corpus_keys,
     'api_key': str(os.environ['api_key']),
     'title': os.environ['title'],
     'description': os.environ['description'],
@@ -22,7 +20,7 @@ cfg = OmegaConf.create({
     'examples': os.environ.get('examples', None)
 })
 
-vq = VectaraQuery(cfg.api_key, cfg.customer_id, cfg.corpus_ids, cfg.prompt_name)
+vq = VectaraQuery(cfg.api_key, cfg.corpus_keys, cfg.prompt_name)
 
 def respond(message, history):
     if cfg.streaming:
@@ -37,8 +35,8 @@ def respond(message, history):
     else:
         # Call non-stream response and return message output
         response = vq.submit_query(message)
-        # yield response
-        return response
+        yield response
+
 
 cfg.description = f'''
                 <table>