Update app.py

app.py

@@ -29,6 +29,15 @@ LLM_HISTORY_LEN = 3
 
 BUTTON_MIN_WIDTH = 150
 
+STATUS_NOK = "404-MODEL UNREADY-red"
+STATUS_OK  = "200-MODEL LOADED-brightgreen"
+
+def get_status(inputs) -> str:
+    return f"""<img src="https://img.shields.io/badge/{inputs}?style=flat"></a>"""
+    
+
+MODEL_NULL = get_status(STATUS_NOK)
+MODEL_DONE = get_status(STATUS_OK)
 
 MODEL_WARNING = "Please paste your OpenAI API Key from openai.com and press 'Enter' to initialize this application!"
 
@@ -63,12 +72,13 @@ def init_model(api_key):
             db = Pinecone.from_existing_index(index_name = PINECONE_INDEX,
                                               embedding  = embeddings)
 
-            return api_key, chain, db, None
+            return api_key, MODEL_DONE, chain, db, None
         else:
-            return None,None,None,None
+            return None,MODEL_NULL,None,None,None
     except Exception as e:
         print(e)
-        return None,None,None,None
+        return None,MODEL_NULL,None,None,None
+
 
 def get_chat_history(inputs) -> str:
     res = []
@@ -127,11 +137,18 @@ with gr.Blocks(css=""".bigbox {
     gr.Markdown(init_message)
     
     with gr.Row():
-        api_textbox = gr.Textbox(
-            value = OPENAI_API_KEY,
-            placeholder = "Paste Your OpenAI API Key (sk-...) and Hit ENTER",
-            show_label=False, lines=1, type='password')
-        init = gr.Button("Initialize Model").style(full_width=False)
+        with gr.Column(scale=9):
+            api_textbox = gr.Textbox(
+                label = "OpenAI API Key",
+                value = OPENAI_API_KEY,
+                placeholder = "Paste Your OpenAI API Key (sk-...) and Hit ENTER",
+                lines=1,
+                type='password')
+            
+        with gr.Column(scale=1, min_width=BUTTON_MIN_WIDTH):
+            
+            init = gr.Button("Initialize Model").style(full_width=False)
+            model_statusbox = gr.HTML(MODEL_NULL)
     
     with gr.Tab("3GPP-Chatbot"):
         with gr.Row():
@@ -169,10 +186,10 @@ with gr.Blocks(css=""".bigbox {
                 
     api_textbox.submit(init_model,
                        api_textbox,
-                       [api_textbox, llm_chain, vector_db, chatbot])
+                       [api_textbox, model_statusbox, llm_chain, vector_db, chatbot])
     init.click(init_model,
                        api_textbox,
-                       [api_textbox, llm_chain, vector_db, chatbot])
+                       [api_textbox, model_statusbox, llm_chain, vector_db, chatbot])
     
     submit.click(user,
                  [query, chatbot],