Update app.py

app.py

@@ -32,13 +32,6 @@ import plotly.express as px
 import zipfile
 import torch 
 
-import matplotlib.pyplot as plt
-from wordcloud import WordCloud
-from nltk.corpus import stopwords
-
-import nltk
-nltk.download('stopwords')
-
 
 
 with st.sidebar:
@@ -142,8 +135,7 @@ if st.button("Sentiment Analysis", type="secondary"):
                     st.error(f"Exception during comment extraction: {e}")
                 driver.quit()
                 df = pd.DataFrame(data, columns=["Comment"])
-                
-                
+                st.dataframe(df)
 
                 if tokenizer and model:
                     inputs = tokenizer(df['Comment'].tolist(), return_tensors="pt", padding=True, truncation=True)
@@ -164,9 +156,12 @@ if st.button("Sentiment Analysis", type="secondary"):
                         fig1.update_traces(textposition='inside', textinfo='percent+label')
                         st.plotly_chart(fig1)
 
+                    result = pd.concat([df, sentiment_df], axis=1)
+                    
+
                     with tab2:
                         text = " ".join(comment for comment in df['Comment'])
-                        stopwords_set = set(stopwords.words('english')) # Correct import and usage
+                        stopwords_set = set(stopwords.words('english')) 
                         text = re.sub('[^A-Za-z]+', ' ', text)
                         words = text.split()
                         clean_text = [word for word in words if word.lower() not in stopwords_set]
@@ -177,8 +172,6 @@ if st.button("Sentiment Analysis", type="secondary"):
                         plt.axis('off')
                         st.pyplot(fig)
 
-                    result = pd.concat([df, sentiment_df], axis=1)
-                    
 
                     csv = result.to_csv(index=False)
                     st.download_button(
@@ -193,11 +186,8 @@ if st.button("Sentiment Analysis", type="secondary"):
     else:
         st.warning(f"You have reached the maximum URL attempts ({max_attempts}).")
 
-if 'url_count' in st.session_state: 
+if 'url_count' in st.session_state: #added if statement.
     st.write(f"URL pasted {st.session_state['url_count']} times.")
 
 
-
-
-               
-                
+