Update app.py

app.py

@@ -71,6 +71,7 @@ DOCS_DONE = get_status(DB_DONE, "processingfoundation", "right")
 TAB_1 = "Chatbot"
 TAB_2 = "Details"
 TAB_3 = "Database"
+TAB_4 = "TODO"
 
 
 
@@ -98,7 +99,7 @@ webui_title = """
 # OpenAI Chatbot Based on Vector Database
 """
 
-dup_link = f'''<a href="https://huggingface.co/spaces/ShawnAI/3GPP-ChatBot?duplicate=true"
+dup_link = f'''<a href="https://huggingface.co/spaces/ShawnAI/VectorDB-ChatBot?duplicate=true"
 style="display:grid; width: 200px;">
 <img src="{get_logo(FORK_BADGE, "addthis")}"></a>'''
 
@@ -126,15 +127,6 @@ Question:
 Answer:"""
 )
 
-'''
-Optinal:
-Expand each clause/figure name in the answer with corresponding metadata file name.
-
-Desired format:
-Clause/figure name: <dot_separated_numbers>
-File name: *.*
-'''
-
 PROMPT_BASE = PromptTemplate(
     input_variables=['question', "chat_history"],
     template="""Chat History:
@@ -152,8 +144,17 @@ Answer:"""
 
 #----------------------------------------------------------------------------------------------------------
 #----------------------------------------------------------------------------------------------------------
+def init_rwkv():
+    try:
+        import rwkv
+        return True
+    except Exception:
+        print("RWKV not found, skip local llm")
+        return False
+
 
 def init_model(api_key, emb_name, emb_loader, db_api_key, db_env, db_index):
+    init_rwkv()
     try:
         if not (api_key and api_key.startswith("sk-") and len(api_key) > 50):
             return None,MODEL_NULL+DOCS_NULL,None,None,None,None
@@ -212,7 +213,6 @@ def doc_similarity(query, db, top_k, score):
                                            k=top_k)
     #docsearch = db.as_retriever(search_kwargs={'k':top_k})
     #docs = docsearch.get_relevant_documents(query)
-    print(docs)
     udocs = remove_duplicates(docs, score)
     return udocs
 
@@ -406,6 +406,10 @@ with gr.Blocks(
                     lines=1,
                     interactive=True,
                     type='email')
+    with gr.Tab(TAB_4):
+        "TODO"
+
+        
 
     init_input  = [llm_api_textbox, emb_textbox, emb_dropdown, db_api_textbox, db_env_textbox, db_index_textbox]
     init_output = [llm_api_textbox, model_statusbox,