simplify app

app.py

@@ -14,30 +14,50 @@ def initialize_and_tokenize(tokenizer):
     rec.tokenize_text()
 
 names = []
-def recommend (movies) :
+def recommend (movies, tok) :
+    initialize_and_tokenize(tok)
     pool = concurrent.futures.ThreadPoolExecutor(max_workers=10)
     futures = [pool.submit(rec.recommend_k_items, movie, 5) for movie in movies]
     idss = [f.result() for f in futures]
     ids = [id for ids in idss for id in ids]
     ids = list(set(ids))
     names = desc[desc['id'].isin(ids)]['title'].to_list()
+    return ', '.join(names)
 
-
-with gr.Blocks() as demo:
-    gr.Markdown("Start typing below and then click **Run** to see the output.")
-    with gr.Row():
-        radio = gr.Radio(["bert", "scibert", "nltk" , "none"], value="none", 
-                         label="Tokenization and text preprocess")
-    btn = gr.Button("Tokenize and Preprocess")
-    btn.click(fn=initialize_and_tokenize, inputs=radio, outputs=[])
-
-    gr.Markdown("Choose 3 movies")
-    with gr.Row():
-        dropdown = gr.Dropdown(choices = list(desc['title']), multiselect=True, max_choices=3,
-                               label="Movies")
-    btn2 = gr.Button("Recommend")
-    btn2.click(fn=recommend, inputs=dropdown,outputs=[])
-    gr.Markdown("rec{}".format(len(names)))
+demo = gr.Interface(fn=recommend, 
+             inputs=[gr.Dropdown(choices = list(desc['title']), multiselect=True, max_choices=3, label="Movies"), 
+                     gr.Radio(["bert", "scibert", "nltk" , "none"], value="none", label="Tokenization and text preprocess")], 
+             outputs=gr.Textbox())
 demo.launch()
 
+
+# ===========================
+# with gr.Blocks() as demo:
+#     gr.Markdown("Start typing below and then click **Run** to see the output.")
+#     with gr.Row():
+#         radio = gr.Radio(["bert", "scibert", "nltk" , "none"], value="none", 
+#                          label="Tokenization and text preprocess")
+#         btn = gr.Button("Tokenize and Preprocess")
+#         btn.click(fn=initialize_and_tokenize, inputs=radio)
+# # demo.launch()
+# # with gr.Blocks() as demo2:
+#     gr.Markdown("Choose 3 movies")
+#     with gr.Row():
+#         dropdown = gr.Dropdown(choices = list(desc['title']), multiselect=True, max_choices=3,
+#                                label="Movies")
+#         box = gr.Textbox(lines=3, label="recs")
+#         btn2 = gr.Button("Recommend")
+#         btn2.click(fn=recommend, inputs=dropdown,outputs=[])
+#         gr.Markdown("rec{}".format(len(names)))
+# demo.launch()
+
+# ==========================
+
+# with gr.Blocks() as demo :
+#     gr.Markdown("Start typing below and then click **Run** to see the output.")
+#     with gr.Row():
+#         radio = gr.Radio(["bert", "scibert", "nltk" , "none"], value="none", 
+#                          label="Tokenization and text preprocess")
+#         btn = gr.Button("Tokenize and Preprocess")
+#         btn.click(fn=initialize_and_tokenize, inputs=radio, outputs=[])
 # demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label")