Update app.py

app.py

@@ -418,19 +418,6 @@ df.drop(columns={col_to_delete}, inplace=True)
             st.pyplot(fig)
             new_line()
         
-        # Feature Importance (Only if a model has been trained)
-        if 'trained_model' in st.session_state and st.session_state.trained_model is not None:
-            feature_importance = st.checkbox("Show Feature Importance (Only click after training or it will throw error)", value=False)
-            new_line()
-            if feature_importance:
-                model = st.session_state.trained_model
-                importances = pd.Series(model.feature_importances_, index=X_train.columns)
-                fig, ax = plt.subplots()
-                importances.sort_values().plot(kind='barh', ax=ax)
-                ax.set_title('Feature Importance')
-                st.pyplot(fig)
-                new_line()
-        
         new_line()
         if st.checkbox("Identify Outliers", value=False):
             numeric_cols = df.select_dtypes(include=np.number).columns.tolist()
@@ -479,20 +466,26 @@ df.drop(columns={col_to_delete}, inplace=True)
 
         new_line()
         if st.checkbox("Show Word Cloud", value=False):
-            text_col = st.selectbox("Select Text Column for Word Cloud", options=df.select_dtypes(include=[np.object]).columns.tolist())
-            text_data = ' '.join(df[text_col].dropna()).strip()  # Collect and strip the text data
+            # Get the list of object-type columns for user to choose from
+            text_col_options = df.select_dtypes(include=[np.object, 'string']).columns.tolist()
             
-            if not text_col and text_data:
-                st.error("No words available to create a word cloud.")
-            elif text_data:  # Check if there is any text data
-                wordcloud = WordCloud(width=800, height=400).generate(text_data)
-                fig, ax = plt.subplots()
-                ax.imshow(wordcloud, interpolation='bilinear')
-                ax.axis('off')
-                st.pyplot(fig)
+            if text_col_options:
+                # Let the user select a text column
+                text_col = st.selectbox("Select Text Column for Word Cloud", options=text_col_options)
+                
+                # Collect text data, dropping NA values and joining them into a single string
+                text_data = ' '.join(df[text_col].dropna()).strip()
+                
+                if text_data:  # Check if there is any text data to use
+                    wordcloud = WordCloud(width=800, height=400).generate(text_data)
+                    fig, ax = plt.subplots()
+                    ax.imshow(wordcloud, interpolation='bilinear')
+                    ax.axis('off')
+                    st.pyplot(fig)
+                else:
+                    st.error("No words available to create a word cloud. Please check the selected text data.")
             else:
-                st.write("No words available to create a word cloud.")
-
+                st.error("No suitable text columns found for creating a word cloud.")
 
 
         new_line()