enable_queue moved to gr.interface rather than in launch

radiobee/__main__.py

@@ -17,6 +17,7 @@ from sklearn.cluster import DBSCAN  # noqa
 
 import joblib
 from varname import nameof
+from icecream import install as ic_install, ic
 import logzero
 from logzero import logger
 
@@ -39,6 +40,11 @@ import gradio as gr
 from radiobee.process_upload import process_upload
 from radiobee.gradiobee import gradiobee
 
+ic_install()
+ic.configureOutput(includeContext=True)
+ic.enable()
+# ic.disenable()  # to turn off
+
 sns.set()
 sns.set_style("darkgrid")
 fastlid.set_languages = ["en", "zh"]
@@ -237,7 +243,6 @@ if __name__ == "__main__":
         [
             "data/test-dual.txt",
             "data/empty.txt",
-            # None,  # does not work
             "linear",
             "None",
             "None",
@@ -248,7 +253,6 @@ if __name__ == "__main__":
         [
             "data/test-zh-ja.txt",
             "data/empty.txt",
-            # None,  # does not work
             "linear",
             "None",
             "None",
@@ -423,6 +427,7 @@ if __name__ == "__main__":
             "excelsior",
         ],  # "paragon"],
         css=f"{css_image} {css_input_file} {css_output_file}",
+        enable_queue=True,
     )
 
     iface.launch(
@@ -433,7 +438,7 @@ if __name__ == "__main__":
         # server_name="127.0.0.1",
         server_port=server_port,
         # show_tips=True,
-        enable_queue=True,
+        # enable_queue=True,
         # height=150,  # 500
         width=900,  # 900
     )

radiobee/gradiobee.py

@@ -7,9 +7,10 @@ from itertools import zip_longest
 
 # import tempfile
 
-from logzero import logger
 from sklearn.cluster import DBSCAN
 from fastlid import fastlid
+from logzero import logger
+from icecream import ic
 
 import numpy as np
 import pandas as pd
@@ -304,6 +305,7 @@ def gradiobee(
     ylim, xlim = len1, len2
 
     # does not seem to show up
+    ic(f" len1 (ylim): {len1}, len2 (xlim): {len2}")
     logger.debug(" len1 (ylim): %s, len2 (xlim): %s", len1, len2)
     if debug:
         print(f" len1 (ylim): {len1}, len2 (xlim): {len2}")
@@ -360,6 +362,7 @@ def gradiobee(
     )
 
     plt_loc = "img/plt.png"
+    ic(f" plotting to {plt_loc}")
     plt.savefig(plt_loc)
 
     # clustered
@@ -381,6 +384,8 @@ def gradiobee(
     df_aligned = df_aligned[["text2", "text1", "likelihood"]]
     df_aligned.columns = ["text1", "text2", "likelihood"]
 
+    ic(df_aligned.head())
+
     # round the last column to 2
     # df_aligned.likelihood = df_aligned.likelihood.round(2)
     # df_aligned = df_aligned.round({"likelihood": 2})
@@ -430,6 +435,9 @@ def gradiobee(
     # output_plot: gr.outputs.Image(type="auto", label="...")
     # return df_trimmed, output_plot, file_dl, file_dl_xlsx, df_aligned
     # return df_trimmed, output_plot, file_dl, file_dl_xlsx, styled, df_html  # gradio cant handle style
+
+    ic("returning outputs")
+
     return df_trimmed, output_plot, file_dl, file_dl_xlsx, df_aligned, df_html
 
     # modi outputs

requirements.txt

@@ -23,4 +23,5 @@ pyicu
 pycld2
 tqdm
 polyglot
-sentence_splitter
+sentence_splitter
+icecream