Fix gr.HTML outputs

app.py

@@ -4,4 +4,3 @@ from ubee import __main__
 
 if __name__ == "__main__":
         sys.exit(__main__.main())
-

ubee/__main__.py

@@ -32,15 +32,14 @@ from icecream import ic
 from icecream import install as ic_install
 from logzero import logger
 from set_loglevel import set_loglevel
-logzero.loglevel(set_loglevel())
-
-# for embeddable python
-# if "." not in sys.path: sys.path.insert(0, ".")
 
 from ubee import __version__
 from ubee.ubee import ubee
 
-# logzero.loglevel(10)
+# for embeddable python
+# if "." not in sys.path: sys.path.insert(0, ".")
+
+logzero.loglevel(set_loglevel())
 logger.debug(" debug on ")
 
 ic_install()
@@ -81,11 +80,7 @@ except Exception as _:
 
 
 # segment: str
-def ifn(
-    text1,
-    text2,
-    thresh
-):
+def ifn(text1, text2, thresh):
     """Take inputs, return outputs.
 
     Args:
@@ -118,7 +113,17 @@ def ifn(
 
     # return out_df, pd.DataFrame(res1_, columns=["text1", "text2", "likelihood"]), _
 
-    return pd.DataFrame(res1_, columns=["text1", "text2", "likelihood"]).to_html(), _.to_html()
+    df = pd.DataFrame(res1_, columns=["text1", "text2", "likelihood"])
+    if df:
+        html1 = (df.to_html(),)
+    else:
+        html1 = ""
+    if _:
+        html2 = _.to_html()
+    else:
+        html2 = ""
+
+    return html1, html2
 
 
 def main():
@@ -142,7 +147,8 @@ def main():
     theme = "dark-grass"
     theme = "grass"
     description = """WIP showcasing a novel aligner"""
-    article = dedent("""
+    article = dedent(
+        """
         ## NB
 
         *   The ultimatumbee aligner (``ubee`` for short) is intended for aligning text blocks (be it paragraphs, sentences or words). Since it is rather slow (30 para pairs (Wuthering Height ch1. for example) can take 10 to 20 mniutes), anything more than 50 blocks should probably be avaoided. Nevertheless, you are welcome to try. No big brother is watching.
@@ -150,23 +156,26 @@ def main():
         *   ``thresh``: longer text blocks justify a larger value; `.5` appears to be just right for paragraphs for Wuthering Height ch1.
 
         Stay tuned for more details coming soon...
-        """).strip()
+        """
+    ).strip()
 
     ex1_zh = [
-        '雪开始下大了。',
-        '我握住门柄又试一回。',
-        '这时一个没穿外衣的年轻人，扛着一根草耙，在后面院子里出现了。', '他招呼我跟着他走，穿过了一个洗衣房和一片铺平的地，那儿有煤棚、抽水机和鸽笼，我们终于到了我上次被接待过的那间温暖的、热闹的大屋子。',
-        '煤、炭和木材混合在一起燃起的熊熊炉火，使这屋子放着光彩。', '在准备摆上丰盛晚餐的桌旁，我很高兴地看到了那位“太太”，以前我从未料想到会有这么一个人存在的。',
-        '我鞠躬等候，以为她会叫我坐下。',
-        '她望望我，往她的椅背一靠，不动，也不出声。'
+        "雪开始下大了。",
+        "我握住门柄又试一回。",
+        "这时一个没穿外衣的年轻人，扛着一根草耙，在后面院子里出现了。",
+        "他招呼我跟着他走，穿过了一个洗衣房和一片铺平的地，那儿有煤棚、抽水机和鸽笼，我们终于到了我上次被接待过的那间温暖的、热闹的大屋子。",
+        "煤、炭和木材混合在一起燃起的熊熊炉火，使这屋子放着光彩。",
+        "在准备摆上丰盛晚餐的桌旁，我很高兴地看到了那位“太太”，以前我从未料想到会有这么一个人存在的。",
+        "我鞠躬等候，以为她会叫我坐下。",
+        "她望望我，往她的椅背一靠，不动，也不出声。",
     ]
     ex1_en = [
-        'The snow began to drive thickly.',
-        'I seized the handle to essay another trial; when a young man without coat, and shouldering a pitchfork, appeared in the yard behind.',
-        'He hailed me to follow him, and, after marching through a wash-house, and a paved area containing a coal shed, pump, and pigeon cot, we at length arrived in the huge, warm, cheerful apartment, where I was formerly received.',
+        "The snow began to drive thickly.",
+        "I seized the handle to essay another trial; when a young man without coat, and shouldering a pitchfork, appeared in the yard behind.",
+        "He hailed me to follow him, and, after marching through a wash-house, and a paved area containing a coal shed, pump, and pigeon cot, we at length arrived in the huge, warm, cheerful apartment, where I was formerly received.",
         "It glowed delightfully in the radiance of an immense fire, compounded of coal, peat, and wood; and near the table, laid for a plentiful evening meal, I was pleased to observe the `missis', an individual whose existence I had never previously suspected.",
-        'I bowed and waited, thinking she would bid me take a seat.',
-        'She looked at me, leaning back in her chair, and remained motionless and mute.'
+        "I bowed and waited, thinking she would bid me take a seat.",
+        "She looked at me, leaning back in her chair, and remained motionless and mute.",
     ]
     shuffle(ex1_en)
     ex1_zh = "\n".join(ex1_zh)
@@ -178,8 +187,8 @@ def main():
     ex2_en = "\n".join(ex2_en)
 
     examples = [
-        [ex2_zh, ex2_en, .3],
-        [text_zh, text_en, .5],
+        [ex2_zh, ex2_en, 0.3],
+        [text_zh, text_en, 0.5],
     ]
     lines = 15
     placeholder = "Type or paste text here"
@@ -188,14 +197,16 @@ def main():
 
     with gr.Blocks() as blocks:
         gr.Markdown(
-            dedent(f"""
+            dedent(
+                f"""
             ## Ultimatumbee-hf-dev {__version__}
 
             Align non-sequential dualtexts.
 
             可对词、句、段，每个词（或句或段）一行。可对任意语言对（英中、英德、德法、中日……等等）。建议 threshold 门槛值 -- 词： 0.3，句：0.5， 段： 0.7。如果太多 leftover，可适当调小 threshold。 如果太多误对则可以适当调大 threshold。
 
-            """).strip()
+            """
+            ).strip()
         )
         with gr.Column():
             with gr.Row():
@@ -204,7 +215,6 @@ def main():
                 )
                 text2 = gr.inputs.Textbox(
                     lines=lines, placeholder=placeholder, default=ex1_en, label="text2"
-
                 )
             with gr.Row():
                 thresh = gr.Slider(
@@ -215,7 +225,7 @@ def main():
                     label="threshold",
                 )
                 btn = gr.Button("Run")
-                
+
             _ = """
             out_df = gr.outputs.Dataframe(
                 headers=None,
@@ -226,7 +236,7 @@ def main():
                 label="To be aligned",
             )
             # """
-            
+
             with gr.Row():
                 _ = """
                 aligned = gr.Dataframe(
@@ -246,7 +256,7 @@ def main():
                     label="Leftover",
                 )
                 # """
-                
+
             aligned = gr.HTML(label="Aligned")
             leftover = gr.HTML(label="Leftover")
 
@@ -261,7 +271,7 @@ def main():
                     # out_df,
                     aligned,
                     leftover,
-                ]
+                ],
             )
 
     # blocks.launch()