Update app.py

app.py

@@ -9,16 +9,13 @@ def load_data(csv_file):
 
 
 def convert_to_rating(sentiment):
-    if sentiment >= -1 and sentiment < -0.6:
-        return 1
-    elif sentiment >= -0.6 and sentiment < -0.2:
-        return 2
-    elif sentiment >= -0.2 and sentiment < 0.2:
+    if sentiment > 0:
+        return 5
+    elif sentiment == 0:
         return 3
-    elif sentiment >= 0.2 and sentiment < 0.6:
-        return 4
     else:
-        return 5
+        return 1
+
 
 def main():
     st.title("Review to Rating Converter")
@@ -36,16 +33,16 @@ def main():
 
         # Display the converted ratings
         st.subheader("Converted Ratings:")
-        st.dataframe(df['Rating'])
+        st.dataframe(df[['Reviews', 'Rating']])
 
         # Download the converted ratings as CSV
         st.download_button(
             label="Download Converted Ratings",
-            data=df['Rating'].to_csv(index=False),
+            data=df[['Reviews', 'Rating']].to_csv(index=False),
             file_name="converted_ratings.csv",
             mime="text/csv"
         )
 
 
 if __name__ == '__main__':
-    main()
+    main()