Update app.py

app.py

@@ -57,6 +57,10 @@ def conversational_chat(query, replacement_word=None):
         output = re.sub(r'\bjack\b', replacement_word, output, flags=re.IGNORECASE)
     
     return extract_text_from_html(output)
+    
+def word_count(text):
+    words = re.findall(r'\w+', text)
+    return len(words)
 
 
 
@@ -126,6 +130,10 @@ def ui():
     if st.button("Submit"):
         # Generate a chatbot response
         bio = conversational_chat(prompt, name_input)
+        # Count words in the generated bio
+        bio_word_count = word_count(bio)
+        st.write(f"Generated Bio Word Count: {bio_word_count}")
+        st.write(bio)
         st.write(bio)