add word counter

app.py

@@ -140,6 +140,10 @@ def word_limit_validation(human_input):
 def on_textbox_change(session_index, session_manager, service, SHEET_ID):
     return save_data(session_index, session_manager, service, SHEET_ID)
 
+def update_word_count(text):
+    words = text.split()
+    return f"Word count: {len(words)}"
+
 if __name__ == "__main__":
     api_key = get_api_key(local=False)
     service, SHEET_IDs = get_sheet_service(local=False)
@@ -187,6 +191,8 @@ if __name__ == "__main__":
         with group_a_content:
             with gr.Row():
                 human_input = gr.Textbox(label="Human Input (At least 50 words)")
+                word_count_display_a = gr.Label(value="Word count: 0")
+                human_input.change(fn=update_word_count, inputs=human_input, outputs=word_count_display_a)
             with gr.Row():
                 submit_btn = gr.Button("Create")
             with gr.Row():
@@ -238,6 +244,8 @@ if __name__ == "__main__":
             with gr.Row():
                 initial_answer = gr.Textbox(label="AI Initial Answer")
                 modification_suggestions = gr.Textbox(label="Refine AI output (At least 50 words)")
+                word_count_display_b = gr.Label(value="Word count: 0")
+                modification_suggestions.change(fn=update_word_count, inputs=modification_suggestions, outputs=word_count_display_b)
             with gr.Row():
                 create_final_btn = gr.Button("Modify")
             with gr.Row():
@@ -296,7 +304,13 @@ if __name__ == "__main__":
 
         with group_c_content:
             with gr.Row():
-                human_input = gr.Textbox(label="Human Input (At least 50 words)")
+                human_input = gr.Textbox(label="Human Input (At least 50 words)", elem_id="human_input")
+                word_count = gr.Label(value="Word count: 0", elem_id="word_count")
+            human_input.change(
+                fn=update_word_count,
+                inputs=human_input,
+                outputs=word_count
+            )
             with gr.Row():
                 create_btn = gr.Button("Create")
             with gr.Row():