Fixed 'user-review' issue in streamlit

Author: taquiansari

Web_app/streamlit_app.py

@@ -27,9 +27,16 @@ def load_data(file):
         st.write("Column Names:")
         st.write(reviews_df.columns.tolist())
 
+        # Check for 'review' or 'user-review' columns
+        review_column = None
         if 'review' in reviews_df.columns:
+            review_column = 'review'
+        elif 'user-review' in reviews_df.columns:
+            review_column = 'user-review'
+
+        if review_column:
             st.write("Sentiment Analysis:")
-            sentiment_df, analyzed_df = analyze_reviews(reviews_df)
+            sentiment_df, analyzed_df = analyze_reviews(reviews_df, review_column)
             st.write(sentiment_df)
 
             st.write("Analyzed DataFrame with Sentiments:")
@@ -39,6 +46,6 @@ def load_data(file):
             recommendations = recommend_movies(analyzed_df)
             st.write(recommendations)
         else:
-            st.error("The uploaded CSV file does not contain a 'review' column.")
+            st.error("The uploaded CSV file does not contain a 'review' or 'user-review' column.")
 else:
     st.write("Please upload a CSV file to proceed.")