Update app.py

app.py

@@ -23,20 +23,21 @@ PINECONE_INDEX = "3gpp"
 
 EMBEDDING_MODEL = "sentence-transformers/all-mpnet-base-v2"
 
-# return top-k text chunk from vector store
-VECTOR_SEARCH_TOP_K = 10
+# return top-k text chunks from vector store
+TOP_K_DEFAULT = 10
+TOP_K_MAX = 25
 
-# LLM input history length
-LLM_HISTORY_LEN = 3
 
+BUTTON_MIN_WIDTH = 180
 
-BUTTON_MIN_WIDTH = 150
-
-STATUS_NOK = "404-MODEL UNREADY-red"
-STATUS_OK  = "200-MODEL LOADED-brightgreen"
+STATUS_NOK = "404-MODEL UNREADY-critical"
+STATUS_OK  = "200-MODEL LOADED-9cf"
 
 def get_status(inputs) -> str:
-    return f"""<img src="https://img.shields.io/badge/{inputs}?style=flat"></a>"""
+    return f"""<img
+    src   = "https://img.shields.io/badge/{inputs}?style=flat&logo=openai";
+    style = "margin: 0 auto;"
+    >"""
     
 
 MODEL_NULL = get_status(STATUS_NOK)
@@ -59,6 +60,9 @@ init_message = f"""Welcome to use 3GPP Chatbot, this demo toolkit is based on Op
     2. Insert your Question and click `{KEY_SUBMIT}`
 """
 
+#----------------------------------------------------------------------------------------------------------
+#----------------------------------------------------------------------------------------------------------
+
 def init_model(api_key):
     try:
         if api_key and api_key.startswith("sk-") and len(api_key) > 50:
@@ -104,7 +108,8 @@ def remove_duplicates(documents):
 def doc_similarity(query, db, top_k):
     docsearch = db.as_retriever(search_kwargs={'k':top_k})
     docs = docsearch.get_relevant_documents(query)
-    return remove_duplicates(docs)
+    udocs = remove_duplicates(docs)
+    return udocs
 
 def user(user_message, history):
     return "", history+[[user_message, None]]
@@ -152,18 +157,25 @@ def bot(box_message, ref_message, chain, db, top_k):
     box_message[-1][1] = bot_message
     return box_message, "", [[details, source]]
 
-with gr.Blocks(css=""".bigbox {
+#----------------------------------------------------------------------------------------------------------
+#----------------------------------------------------------------------------------------------------------
+
+with gr.Blocks(
+    theme = "Base",
+    css = """.bigbox {
     min-height:200px;
-}""") as demo:
+}
+""") as demo:
     llm_chain = gr.State()
     vector_db = gr.State()
     gr.Markdown(webui_title)
     gr.Markdown(init_message)
     
     with gr.Row():
-        with gr.Column(scale=9):
+        with gr.Column(scale=10):
             api_textbox = gr.Textbox(
                 label = "OpenAI API Key",
+                show_label = False,
                 value = OPENAI_API_KEY,
                 placeholder = "Paste Your OpenAI API Key (sk-...) and Hit ENTER",
                 lines=1,
@@ -171,7 +183,7 @@ with gr.Blocks(css=""".bigbox {
             
         with gr.Column(scale=1, min_width=BUTTON_MIN_WIDTH):
             
-            init = gr.Button(KEY_INIT).style(full_width=False)
+            init = gr.Button(KEY_INIT) #.style(full_width=False)
             model_statusbox = gr.HTML(MODEL_NULL)
     
     with gr.Tab("3GPP-Chatbot"):
@@ -200,8 +212,8 @@ with gr.Blocks(css=""".bigbox {
 
     with gr.Tab("Details"):
         top_k = gr.Slider(1,
-                          20,
-                          value=VECTOR_SEARCH_TOP_K,
+                          TOP_K_MAX,
+                          value=TOP_K_DEFAULT,
                           step=1,
                           label="Vector similarity top_k",
                           interactive=True)
@@ -226,6 +238,9 @@ with gr.Blocks(css=""".bigbox {
     
     clear.click(lambda: (None,None,None), None, [query, ref, chatbot], queue=False)
 
+#----------------------------------------------------------------------------------------------------------
+#----------------------------------------------------------------------------------------------------------
+    
 if __name__ == "__main__":
     demo.launch(share=False, inbrowser=True)