Update app.py

app.py

@@ -31,7 +31,7 @@ def load_and_analyze_csv(file, text_field, event_model):
     flood_related = gr.CheckboxGroup(choices=floods)
     not_related = gr.CheckboxGroup(choices=nones)
     
-    return flood_related, fire_related, not_related, model_confidence
+    return flood_related, fire_related, not_related, model_confidence, len(df[text_field].to_list())
 
 def analyze_selected_texts(selections):
     selected_texts = selections
@@ -41,10 +41,11 @@ def analyze_selected_texts(selections):
     result_df = pd.DataFrame({"Selected Text": selected_texts, "Analysis": analysis_results})
     return result_df
 
-def calculate_accuracy(flood_selections, fire_selections, none_selections):
+def calculate_accuracy(flood_selections, fire_selections, none_selection, num_posts):
     incorrect = len(flood_selections) + len(fire_selections) + len(none_selections)
-
-    return incorrect
+    correct = num_posts - incorrect
+    accuracy = (correct/num_posts)*100
+    return incorrect, correct, accuracy
     
 
             
@@ -110,9 +111,9 @@ with gr.Blocks() as demo:
 
         accuracy_button = gr.Button("Calculate Accuracy")
         predict_button.click(load_and_analyze_csv, inputs=[file_input, text_field, event_model], 
-                         outputs=[flood_checkbox_output, fire_checkbox_output, none_checkbox_output, model_confidence])
-        accuracy_button.click(calculate_accuracy, inputs=[flood_checkbox_output, fire_checkbox_output, none_checkbox_output], 
-                             outputs=incorrect)
+                         outputs=[flood_checkbox_output, fire_checkbox_output, none_checkbox_output, model_confidence, number_posts])
+        accuracy_button.click(calculate_accuracy, inputs=[flood_checkbox_output, fire_checkbox_output, none_checkbox_output, num_posts], 
+                             outputs=[incorrect, correct, accuracy])
                 
     with gr.Tab("Question Answering"):
         # XXX Add some button disabling here, if the classification process is not completed first XXX