Update app.py

app.py

@@ -4,10 +4,9 @@ import numpy as np
 from sepa import parser
 import re
 
-
-##################################################################
-##################################################################
-##################################################################
+#####################################################################################################################################
+#####################################################################################################################################
+#####################################################################################################################################
 
 def full_function(xml_file):
     
@@ -96,11 +95,22 @@ def full_function(xml_file):
     
     return df_entries, df_entries_example
 
+#####################################################################################################################################
+#####################################################################################################################################
+#####################################################################################################################################
 
+def function_code_count(df_entries):
+    
+    #count number of values
+    df_proprietary_code_count = df_entries['proprietary_code'].value_counts()#.to_frame()
+    df_proprietary_code_count = pd.DataFrame(df_proprietary_code_count).reset_index(names="code")
+    df_proprietary_code_count.rename(columns={"proprietary_code": "count"}, inplace=True)
+    
+    return df_proprietary_code_count
 
-##################################################################
-##################################################################
-##################################################################
+#####################################################################################################################################
+#####################################################################################################################################
+#####################################################################################################################################
 
 
 def export_csv(xml_file):
@@ -109,13 +119,18 @@ def export_csv(xml_file):
     
     df_entries.to_csv("./output.csv")
     
-    out = gr.File.update(value="output.csv", visible=True)
+    out = gr.File(value="output.csv", visible=True)
+    
+    #count codes 
+    df_proprietary_code_count = function_code_count(df_entries)
     
-    return out, df_entries_example
+    return out, df_entries_example, df_proprietary_code_count
+
+#####################################################################################################################################
+#####################################################################################################################################
+#####################################################################################################################################
+
 
-##################################################################
-##################################################################
-##################################################################
 
 desc = "Upload XML file, convert to .csv file, and analyze transactions"
 
@@ -129,10 +144,14 @@ with gr.Blocks() as demo:
     with gr.Row():
         #export_button = gr.Button("Export")
         out = gr.File(label = "Output file", interactive=False, visible=False)
+        
+    with gr.Row():
+        
+        df_proprietary_code_count = gr.DataFrame(label="Number of transactions per code")
 
 
     #submit_btn = gr.Button("Run analysis on XML file")
     #export_button.click(export_csv, df_entries, csv)
 
 
-gr.Interface(fn=export_csv, inputs=xml_file, outputs=[out, df_entries_example], title=desc).launch(share=True, debug =True)
+gr.Interface(fn=export_csv, inputs=xml_file, outputs=[out, df_entries_example, df_proprietary_code_count], title=desc).launch(share=True, debug =True)