Update app.py

app.py

@@ -37,7 +37,8 @@ def process(_input, topn, similar):
         frequencies = [model.wv.get_vecattr(nn[0], 'count') for nn in nearest_neighbors]
 
     result = pd.DataFrame([(a[0],a[1],b) for a,b in zip(nearest_neighbors, frequencies)], columns=['Token', 'Cosine Similarity', 'Frequency'])
-    return result
+    result.to_csv('result.csv')
+    return result, 'result.csv'
 
 def save(df):
     df.to_csv('result.csv')
@@ -66,25 +67,22 @@ with demo:
                         df_output = gr.Dataframe(interactive=False)
                     with gr.Row():
                         compute_button_s = gr.Button("Compute")
-                        export_button_s = gr.Button("Export as CSV")
+                        # export_button_s = gr.Button("Export as CSV")
                     file_out_s = gr.File(interactive=False)
                 with gr.TabItem("Multiple"):
                     with gr.Column():
-                        text_input_multiple = gr.Textbox(lines=3)
+                        text_input_multiple = gr.Textbox(lines=4)
                         df_output_multiple = gr.Dataframe(interactive=False)
                     with gr.Row():
                         compute_button_m = gr.Button("Compute")
-                        export_button_m = gr.Button("Export as CSV")
+                        # export_button_m = gr.Button("Export as CSV")
                     file_out_m = 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.""")
 
-    compute_button_s.click(process, inputs=[text_input, n_output, similar_radio], outputs=df_output)
-    compute_button_m.click(process, inputs=[text_input_multiple, n_output, similar_radio], outputs=df_output_multiple)
-
-    export_button_s.click(save, inputs=[df_output], outputs=file_out_s)
-    export_button_s.click(save, inputs=[df_output_multiple], outputs=file_out_s)
+    compute_button_s.click(process, inputs=[text_input, n_output, similar_radio], outputs=[df_output, file_out_s])
+    compute_button_m.click(process, inputs=[text_input_multiple, n_output, similar_radio], outputs=[df_output_multiple, file_out_m])
 
 demo.launch()