improved UI

app.py

@@ -33,9 +33,14 @@ huggingface_embeddings= HuggingFaceEmbeddings(
 # %%
 persist_directory = 'chroma/'
 vectordb = Chroma(embedding_function=huggingface_embeddings,persist_directory=persist_directory)
-print(vectordb._collection.count())
+print(f"Vector count: {vectordb._collection.count()}")
 
 # %%
+text_input_label=["谜面","謎面","Riddle"]
+text_output_label=["谜底","謎底","Answer"]
+
+clear_label = ["清除","清除","Clear"]
+submit_label = ["提交","提交","Submit"]
 # helper functions for prompt processing for this LLM
 
 def preprocess(text):
@@ -70,9 +75,12 @@ def helper_rag(text):
     return context
 
 # helper function for prompt
-def helper_text(text_input,radio=None):
+def helper_text(text_input,lang=None):
     chinese_type = "simplified"
 
+    if lang == "繁體中文" or lang == "traditional":
+        chinese_type = "traditional"
+
     if hanzidentifier.is_traditional(text_input):
         text_input = chinese_converter.to_simplified(text_input)
         chinese_type = "traditional"
@@ -81,7 +89,7 @@ def helper_text(text_input,radio=None):
 
     if not any(c in text_input for c in ["猜", "打"]):
         warning = "请给一个提示，提示格式，例子：猜一水果，打一字。"
-        if chinese_type == "traditional" or radio == "繁體中文":
+        if chinese_type == "traditional":
             warning = chinese_converter.to_traditional(warning)
         return warning
                                          
@@ -101,8 +109,9 @@ def helper_text(text_input,radio=None):
 
 
 # Gradio function for configure the language of UI
-def change_language(radio,text_input,text_output,clear_btn,submit_btn,markdown, 
-                    markdown_msg1, markdown_msg2):
+def change_language(radio,text_input,text_output,markdown, 
+                    markdown_msg1, markdown_msg2,language):
+
     if radio == "简体中文":
         index = 0
         text_input_update=gr.Textbox.update(value = chinese_converter.to_simplified(text_input), label = text_input_label[index])
@@ -136,8 +145,10 @@ def change_language(radio,text_input,text_output,clear_btn,submit_btn,markdown,
     clear_btn_update = gr.ClearButton.update(value = clear_label[index])
     submit_btn_update = gr.Button.update(value = submit_label[index])
 
+    language = radio
+
     return [text_input_update,text_output_update,clear_btn_update,submit_btn_update,markdown_update, 
-            markdown_msg1_update ,markdown_msg2_update]
+            markdown_msg1_update, markdown_msg2_update,language]
 
 
 def clear_text():
@@ -148,11 +159,6 @@ def clear_text():
     
 
 # %%
-text_input_label=["谜面","謎面","Riddle"]
-text_output_label=["谜底","謎底","Answer"]
-
-clear_label = ["清除","清除","Clear"]
-submit_label = ["提交","提交","Submit"]
 
 # css = """
 # #markdown { background-image: url("file/data/DSC_0105.jpg");
@@ -162,6 +168,7 @@ submit_label = ["提交","提交","Submit"]
 
 with gr.Blocks() as demo:
     index = 0
+    language = gr.State()
     example_list = [
         ["小家伙穿黄袍，花丛中把房造。飞到西来飞到东，人人夸他爱劳动。（猜一动物）"],
         ["一物生来身穿三百多件衣，每天脱一件，年底剩张皮。（猜一物品）"],
@@ -176,6 +183,7 @@ with gr.Blocks() as demo:
             # Chinese Lantern Riddles Solver with LLM
             ## 用语言大模型来猜灯谜 
             """,elem_id="markdown")
+
     with gr.Row():
         with gr.Column():
             text_input = gr.Textbox(label=text_input_label[index], 
@@ -196,11 +204,11 @@ with gr.Blocks() as demo:
             )
             markdown_msg1 = gr.Markdown(
                 """
-                灯谜是中华文化特色文娱活动，自北宋盛行。每年逢正月十五元宵节，或是八月十五中秋节，将谜语贴在花灯上，让大家可一起猜谜。
+                灯谜是中华文化特色文娱活动，自北宋盛行。每年逢正月十五元宵节，将谜语贴在花灯上，让大家可一起猜谜。
 
                 Lantern riddle is a traditional Chinese cultural activity. Being popular since the Song Dynasty (960-1276), it \
-                is held in the Lantern Festival (15th day of the first lunar month) or the Mid-Autumn Festival (15th day of \
-                the eighth lunar month). When people are viewing the flower lanterns, they can guess the riddles on the lanterns together. 
+                is held in the Lantern Festival (15th day of the first lunar month). \
+                When people are viewing the flower lanterns, they can guess the riddles on the lanterns together. 
 
 
                 """
@@ -247,18 +255,19 @@ with gr.Blocks() as demo:
 
 
 
-    submit_btn.click(fn=helper_text, inputs=[text_input,radio], outputs=text_output)
+    submit_btn.click(fn=helper_text, inputs=[text_input,radio], outputs=text_output, api_name="answer-the-riddle")
 
     clear_btn.click(fn=clear_text,outputs=[text_input,text_output])
-    radio.change(fn=change_language,inputs=[radio,text_input,text_output,clear_btn,submit_btn,
+    radio.change(fn=change_language,inputs=[radio,text_input,text_output,
                                             markdown, markdown_msg1,markdown_msg2],
                  outputs=[text_input,text_output,clear_btn,submit_btn, 
-                          markdown, markdown_msg1,markdown_msg2])
+                          markdown, markdown_msg1,markdown_msg2,language])
 
     #demo = gr.Interface(fn=helper_text, inputs=text_input, outputs=text_output, 
     #                      flagging_options=["Inappropriate"],allow_flagging="never",
     #                      title="aaa",description="aaa",article="aaa")
-demo.launch()   
+demo.queue(api_open=False)
+demo.launch(show_api=False)   
 
 # %%