Update app.py

app.py

@@ -3,6 +3,7 @@ import numpy as np
 import pandas as pd
 from datetime import datetime
 import os
+import re
 
 from huggingface_hub import hf_hub_url, cached_download
 from gensim.models.fasttext import load_facebook_model
@@ -18,11 +19,11 @@ model = load_facebook_model(cached_download(url))
 
 def process(_input, topn):
 
-    # convert input to lower, replace whitespaces by underscores
-    _input = _input.strip().lower().replace(' ', '_')
-    _input = _input.split('\n')
+    # split by delimiting characters
+    _input = re.split('[,;\n]', _input)
 
-    _input = [s for s in _input if s]
+    # convert input to lower, replace whitespaces by underscores
+    _input = [s.strip().lower().replace(' ', '_') for s in _input if s]
 
     if _input[0] != ACCESS_KEY:
         with open('log.txt', 'a') as f:
@@ -47,7 +48,7 @@ def process(_input, topn):
             prompts = [p.strip().split('+++') for p in prompts]
         result = pd.DataFrame(prompts, columns=['Time', 'Prompt'])
     result.to_csv('result.csv')
-    return result, 'result.csv'
+    return result, 'result.csv', '\n'.join(_input)
 
 def save(df):
     df.to_csv('result.csv')
@@ -58,35 +59,28 @@ demo = gr.Blocks()
 with demo:
     gr.Markdown("# Call2Vec")
     gr.Markdown("## Earnings call transformation project")
-    with gr.Tabs():
-        with gr.TabItem(label='Block Interface'):
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("""
+                #### Project Description
+                Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet. Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet.""")
+            gr.Markdown(
+                """#### App usage:
+                Add your input prompts to the text field on the right. To use multiple input prompts at once separate
+                them by comma, semicolon or a new line
+                ##### Examples
+                - Climate change
+                - Financial risk, energy dependency, climate neutrality
+                """
+            )
+        with gr.Column():
+            text_input = gr.Textbox(lines=1)
             with gr.Row():
-                with gr.Column():
-                    similar_radio = gr.Radio(label="Single or multiple input prompts", value="Single", choices=["Single", "Multiple"])
-                    n_output = gr.Slider(minimum=5, maximum=50, step=1)
-                    gr.Markdown(
-                        """### Example prompts:
-                        - Example 1
-                        - Example 2
-                        """
-                    )
-                with gr.Column():
-                    text_input = gr.Textbox(lines=1)
-                    with gr.Row():
-                        compute_button = gr.Button("Compute")
-                    df_output = gr.Dataframe(interactive=False)
-                    file_out = gr.File(interactive=False)
-                with gr.Column():
-                    gr.Markdown("""
-                        ### Project Description
-                        Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet. Lorem ipsum dolor sit amet, consetetur sadipscing elitr, sed diam nonumy eirmod tempor invidunt ut labore et dolore magna aliquyam erat, sed diam voluptua. At vero eos et accusam et justo duo dolores et ea rebum. Stet clita kasd gubergren, no sea takimata sanctus est Lorem ipsum dolor sit amet.""")
+                n_output = gr.Slider(minimum=5, maximum=50, step=1)
+                compute_button = gr.Button("Compute")
+            df_output = gr.Dataframe(interactive=False)
+            file_out = gr.File(interactive=False)
 
-            compute_button.click(process, inputs=[text_input, n_output], outputs=[df_output, file_out])
-            similar_radio.change(lambda x: "\n\n\n\n\n\n\n" if x=='Multiple' else "", inputs=[similar_radio], outputs=[text_input])
-        with gr.TabItem('Traditional Interface'):
-            gr.Interface(process, inputs=[gr.Textbox(lines=3), gr.Slider(minimum=5, maximum=50, step=1)],
-                         outputs=[gr.Dataframe(interactive=False), gr.File(interactive=False)],
-                         examples=[["Test example", 5],
-                                   ["Multiple prompts\nexample", 7]])
+    compute_button.click(process, inputs=[text_input, n_output], outputs=[df_output, file_out, text_input])
 
 demo.launch()