Added more features.

app.py

@@ -28,12 +28,14 @@ df_mean = pd.DataFrame({
     'topic_verification' : ['']
 })
 
-df_news_final = pd.concat([df_news, df_mean]).drop(columns=['body'])
+df_news_final = pd.concat([df_news, df_mean])
 
 df_news_final.index = np.arange(1, len(df_news_final) + 1)
 
 df_news_final
 
+st.markdown('---')
+
 try:
     st.plotly_chart(plot_3dgraph(df_news_final), use_container_width=True)
 except:

modules/data_preparation.py

@@ -27,6 +27,12 @@ def prepare_df(df, categories):
 
         filter = news_data['headline'].str.contains(pattern_del)
         news_data = news_data[~filter]
+
+        # shift column 'C' to first position 
+        first_column = news_data.pop('headline') 
+        
+        # insert column using insert(position,column_name,first_column) function 
+        news_data.insert(0, 'headline', first_column) 
     
     except Exception as E:
         print(E)