Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -33,6 +33,8 @@ BUTTON_MIN_WIDTH = 180
|
|
| 33 |
STATUS_NOK = "404-MODEL UNREADY-critical"
|
| 34 |
STATUS_OK = "200-MODEL LOADED-9cf"
|
| 35 |
|
|
|
|
|
|
|
| 36 |
|
| 37 |
def get_logo(inputs, logo) -> str:
|
| 38 |
return f"""https://img.shields.io/badge/{inputs}?style=flat&logo={logo}&logoColor=white"""
|
|
@@ -44,14 +46,20 @@ def get_status(inputs) -> str:
|
|
| 44 |
>"""
|
| 45 |
|
| 46 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 47 |
MODEL_NULL = get_status(STATUS_NOK)
|
| 48 |
MODEL_DONE = get_status(STATUS_OK)
|
| 49 |
|
| 50 |
-
MODEL_WARNING = "Please paste your OpenAI API Key from
|
|
|
|
| 51 |
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
|
|
|
|
| 55 |
|
| 56 |
webui_title = """
|
| 57 |
# OpenAI Chatbot Based on Vector Database
|
|
@@ -59,7 +67,7 @@ webui_title = """
|
|
| 59 |
"""
|
| 60 |
|
| 61 |
dup_link = f'''<a href="https://huggingface.co/spaces/ShawnAI/3GPP-ChatBot?duplicate=true">
|
| 62 |
-
<img src="{get_logo(
|
| 63 |
|
| 64 |
init_message = f"""Welcome to use 3GPP Chatbot, this demo toolkit is based on OpenAI with LangChain and Pinecone
|
| 65 |
1. Insert your OpenAI API key and click `{KEY_INIT}`
|
|
@@ -147,7 +155,6 @@ def bot(box_message, ref_message, chain, db, top_k):
|
|
| 147 |
|
| 148 |
if delta_top_k > 0:
|
| 149 |
docs = doc_similarity(ref_message, db, top_k+delta_top_k)
|
| 150 |
-
print(docs)
|
| 151 |
|
| 152 |
all_output = chain({"input_documents": docs,
|
| 153 |
"question": question,
|
|
@@ -170,6 +177,7 @@ def bot(box_message, ref_message, chain, db, top_k):
|
|
| 170 |
#----------------------------------------------------------------------------------------------------------
|
| 171 |
|
| 172 |
with gr.Blocks(
|
|
|
|
| 173 |
theme = "Base",
|
| 174 |
css = """.bigbox {
|
| 175 |
min-height:200px;
|
|
@@ -196,7 +204,7 @@ with gr.Blocks(
|
|
| 196 |
init = gr.Button(KEY_INIT) #.style(full_width=False)
|
| 197 |
model_statusbox = gr.HTML(MODEL_NULL)
|
| 198 |
|
| 199 |
-
with gr.Tab(
|
| 200 |
with gr.Row():
|
| 201 |
with gr.Column(scale=10):
|
| 202 |
chatbot = gr.Chatbot(elem_classes="bigbox")
|
|
@@ -272,13 +280,13 @@ with gr.Blocks(
|
|
| 272 |
[chatbot, ref, detail_panel]
|
| 273 |
)
|
| 274 |
|
| 275 |
-
print(dup_link)
|
| 276 |
-
|
| 277 |
clear.click(lambda: (None,None,None), None, [query, ref, chatbot], queue=False)
|
| 278 |
|
| 279 |
#----------------------------------------------------------------------------------------------------------
|
| 280 |
#----------------------------------------------------------------------------------------------------------
|
| 281 |
|
| 282 |
if __name__ == "__main__":
|
| 283 |
-
demo.launch(share=False,
|
|
|
|
|
|
|
| 284 |
|
|
|
|
| 33 |
STATUS_NOK = "404-MODEL UNREADY-critical"
|
| 34 |
STATUS_OK = "200-MODEL LOADED-9cf"
|
| 35 |
|
| 36 |
+
FORK_BADGE = "Fork-HuggingFace Space-9cf"
|
| 37 |
+
|
| 38 |
|
| 39 |
def get_logo(inputs, logo) -> str:
|
| 40 |
return f"""https://img.shields.io/badge/{inputs}?style=flat&logo={logo}&logoColor=white"""
|
|
|
|
| 46 |
>"""
|
| 47 |
|
| 48 |
|
| 49 |
+
KEY_INIT = "Initialize Model"
|
| 50 |
+
KEY_SUBMIT = "Submit"
|
| 51 |
+
KEY_CLEAR = "Clear"
|
| 52 |
+
|
| 53 |
MODEL_NULL = get_status(STATUS_NOK)
|
| 54 |
MODEL_DONE = get_status(STATUS_OK)
|
| 55 |
|
| 56 |
+
MODEL_WARNING = f"Please paste your OpenAI API Key from \
|
| 57 |
+
[openai.com](https://platform.openai.com/account/api-keys) and then **{KEY_INIT}**"
|
| 58 |
|
| 59 |
+
|
| 60 |
+
TAB_1 = "3GPP Chatbot"
|
| 61 |
+
|
| 62 |
+
FAVICON = './icon.svg'
|
| 63 |
|
| 64 |
webui_title = """
|
| 65 |
# OpenAI Chatbot Based on Vector Database
|
|
|
|
| 67 |
"""
|
| 68 |
|
| 69 |
dup_link = f'''<a href="https://huggingface.co/spaces/ShawnAI/3GPP-ChatBot?duplicate=true">
|
| 70 |
+
<img src="{get_logo(FORK_BADGE, "addthis")}"></a> '''
|
| 71 |
|
| 72 |
init_message = f"""Welcome to use 3GPP Chatbot, this demo toolkit is based on OpenAI with LangChain and Pinecone
|
| 73 |
1. Insert your OpenAI API key and click `{KEY_INIT}`
|
|
|
|
| 155 |
|
| 156 |
if delta_top_k > 0:
|
| 157 |
docs = doc_similarity(ref_message, db, top_k+delta_top_k)
|
|
|
|
| 158 |
|
| 159 |
all_output = chain({"input_documents": docs,
|
| 160 |
"question": question,
|
|
|
|
| 177 |
#----------------------------------------------------------------------------------------------------------
|
| 178 |
|
| 179 |
with gr.Blocks(
|
| 180 |
+
title = TAB_1,
|
| 181 |
theme = "Base",
|
| 182 |
css = """.bigbox {
|
| 183 |
min-height:200px;
|
|
|
|
| 204 |
init = gr.Button(KEY_INIT) #.style(full_width=False)
|
| 205 |
model_statusbox = gr.HTML(MODEL_NULL)
|
| 206 |
|
| 207 |
+
with gr.Tab(TAB_1):
|
| 208 |
with gr.Row():
|
| 209 |
with gr.Column(scale=10):
|
| 210 |
chatbot = gr.Chatbot(elem_classes="bigbox")
|
|
|
|
| 280 |
[chatbot, ref, detail_panel]
|
| 281 |
)
|
| 282 |
|
|
|
|
|
|
|
| 283 |
clear.click(lambda: (None,None,None), None, [query, ref, chatbot], queue=False)
|
| 284 |
|
| 285 |
#----------------------------------------------------------------------------------------------------------
|
| 286 |
#----------------------------------------------------------------------------------------------------------
|
| 287 |
|
| 288 |
if __name__ == "__main__":
|
| 289 |
+
demo.launch(share = False,
|
| 290 |
+
inbrowser = True,
|
| 291 |
+
favicon_path = FAVICON)
|
| 292 |
|