Update app.py

app.py

@@ -127,9 +127,20 @@ def ui():
     # Add a submit button
     if st.button("Submit"):
         # Generate a chatbot response
+        name_input = st.text_input("Enter your name:")
         bio = conversational_chat(prompt, name_input)
+
         # Count words in the generated bio
         bio_word_count = word_count(bio)
+        
+        # Check if the bio exceeds 250 words
+        if bio_word_count > 250:
+            st.warning("Generated Bio exceeded 250 words. Re-inferencing...")
+            bio = conversational_chat(prompt, name_input)  # Re-inferencing
+            
+            # Count words in the re-inferenced bio
+            bio_word_count = word_count(bio)
+        
         st.write(f"Generated Bio Word Count: {bio_word_count}")
         st.write(bio)