Update app.py

app.py

@@ -41,20 +41,14 @@ def main():
     input_text = st.text_input("Enter the Word")
     if st.button("Check"):
         top_classes, top_probabilities = mat(input_text)
-        words = input_text.split()  # Split input text into words
-        for word in words:
-            if len(word) > 0:
-                st.markdown(f'<span style="color: blue">{word}</span>', unsafe_allow_html=True)
-        st.write()  # Add line break after displaying input text
         for i, (class_, probability) in enumerate(zip(top_classes, top_probabilities)):
             suggested_word = data[data.Label == class_].Word.values[0]
-            button_text = f"Did you mean '{suggested_word}'?"
-            if st.button(button_text):
-                definition = dataa[dataa['Word'] == suggested_word]['Definition'].values
+            if st.button(f"Suggested Word: {suggested_word}"):
+                definition = dataa[dataa['Word'].str.lower() == suggested_word.lower()]['Definition'].values
                 if len(definition) > 0:
-                    st.markdown(f"The dictionary meaning of '{suggested_word}' is: {definition[0]}")
+                    st.write(f"The dictionary meaning of '{suggested_word}' is: {definition[0]}")
                 else:
-                    st.markdown(f"No definition found for '{suggested_word}' in the dictionary.")
+                    st.write(f"No definition found for '{suggested_word}' in the dictionary.")
 
 if __name__ == "__main__":
     main()