Update app.py

app.py

@@ -33,9 +33,13 @@ BUTTON_MIN_WIDTH = 180
 STATUS_NOK = "404-MODEL UNREADY-critical"
 STATUS_OK  = "200-MODEL LOADED-9cf"
 
+
+def get_logo(inputs, logo) -> str:
+    return f"""https://img.shields.io/badge/{inputs}?style=flat&logo={logo}&logoColor=white"""
+
 def get_status(inputs) -> str:
     return f"""<img
-    src   = "https://img.shields.io/badge/{inputs}?style=flat&logo=openai";
+    src   = "{get_logo(inputs, "openai")}";
     style = "margin: 0 auto;"
     >"""
     
@@ -45,21 +49,25 @@ MODEL_DONE = get_status(STATUS_OK)
 
 MODEL_WARNING = "Please paste your OpenAI API Key from openai.com and press 'Enter' to initialize this application!"
 
+KEY_INIT   = "Initialize Model"
+KEY_SUBMIT = "Submit"
+KEY_CLEAR  = "Clear"
 
 webui_title = """
 # OpenAI Chatbot Based on Vector Database
 ## Example of 3GPP
 """
 
-KEY_INIT   = "Initialize Model"
-KEY_SUBMIT = "Submit"
-KEY_CLEAR  = "Clear"
+dup_link = f'''<a href="https://huggingface.co/spaces/ShawnAI/3GPP-ChatBot?duplicate=true">
+<img src="{get_logo("Fork-HuggingFace Space-9cf", "addthis")}"></a> '''
 
 init_message = f"""Welcome to use 3GPP Chatbot, this demo toolkit is based on OpenAI with LangChain and Pinecone
     1. Insert your OpenAI API key and click  `{KEY_INIT}`
     2. Insert your Question and click  `{KEY_SUBMIT}`
 """
 
+
+
 #----------------------------------------------------------------------------------------------------------
 #----------------------------------------------------------------------------------------------------------
 
@@ -170,13 +178,14 @@ with gr.Blocks(
     llm_chain = gr.State()
     vector_db = gr.State()
     gr.Markdown(webui_title)
+    gr.HTML(dup_link)
     gr.Markdown(init_message)
     
     with gr.Row():
         with gr.Column(scale=10):
             llm_api_textbox = gr.Textbox(
                 label = "OpenAI API Key",
-                show_label = False,
+                # show_label = False,
                 value = OPENAI_API_KEY,
                 placeholder = "Paste Your OpenAI API Key (sk-...) and Hit ENTER",
                 lines=1,
@@ -191,16 +200,7 @@ with gr.Blocks(
         with gr.Row():
             with gr.Column(scale=10):
                 chatbot = gr.Chatbot(elem_classes="bigbox")
-            '''
-            with gr.Column(scale=1, min_width=BUTTON_MIN_WIDTH):
-                temp = gr.Slider(0,
-                          2,
-                          value=OPENAI_TEMP,
-                          step=0.1,
-                          label="temperature",
-                          interactive=True)
-                init = gr.Button("Init")
-            '''
+
         with gr.Row():
             with gr.Column(scale=10):
                 query = gr.Textbox(label="Question:",
@@ -272,6 +272,8 @@ with gr.Blocks(
         [chatbot, ref, detail_panel]
     )
     
+    print(dup_link)
+    
     clear.click(lambda: (None,None,None), None, [query, ref, chatbot], queue=False)
 
 #----------------------------------------------------------------------------------------------------------