Update app.py

app.py

@@ -53,6 +53,7 @@ def bot(history, api_kind):
     top_rerank = 15
     top_k_rank = 8
     query = history[-1][0]
+    print('history[-1][0]',history[-1][0])
 
     if not query:
          gr.Warning("Please submit a non-empty string as a prompt")
@@ -66,12 +67,12 @@ def bot(history, api_kind):
     print(query)
     query_vec_flat = [arr.flatten() for arr in query_vec]
     logger.warning(f'Finished query vec')
-    doc1 = table.search(query_vec_flat, vector_column_name=VECTOR_COLUMN_NAME).limit(top_k_rank)
+    documents = table.search(query_vec_flat, vector_column_name=VECTOR_COLUMN_NAME).limit(top_k_rank).to_list()
 
 
 
     logger.warning(f'Finished search')
-    documents = table.search(query_vec, vector_column_name=VECTOR_COLUMN_NAME).limit(top_rerank).to_list()
+    #documents = table.search(query_vec, vector_column_name=VECTOR_COLUMN_NAME).limit(top_rerank).to_list()
     documents = [doc[TEXT_COLUMN_NAME] for doc in documents]
     logger.warning(f'start cross encoder {len(documents)}')
     # Retrieve documents relevant to query
@@ -151,10 +152,12 @@ with gr.Blocks() as demo:
 
     # Turn it back on
     txt_msg.then(lambda: gr.Textbox(interactive=True), None, [txt], queue=False)
-
-    # Turn off interactivity while generating if you hit enter
-    txt_msg = txt.submit(add_text, [chatbot, txt], [chatbot, txt], queue=False).then(
-            bot, [chatbot, api_kind], [chatbot, prompt_html])
+    try:
+        # Turn off interactivity while generating if you hit enter
+        txt_msg = txt.submit(add_text, [chatbot, txt], [chatbot, txt], queue=False).then(
+                bot, [chatbot, api_kind], [chatbot, prompt_html])
+    except Exception as e:
+        print ('Exception  ' ,str(e))
 
     # Turn it back on
     txt_msg.then(lambda: gr.Textbox(interactive=True), None, [txt], queue=False)