Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import pandas as pd
 from transformers import pipeline
+import base64
 
 # Load the EasyTerms/legalSummerizerET model from Hugging Face
 summarizer = pipeline("summarization", model="EasyTerms/legalSummerizerET")
@@ -11,30 +12,25 @@ def generate_summary(contract_text):
     return summary[0]['summary_text']
 
 # Function to handle feedback and store it in a CSV file
-def handle_feedback(feedback_data):
-    feedback_df = pd.DataFrame(feedback_data, columns=['Contract', 'Summary', 'Chosen', 'Rejected'])
-    
-    # Save the dataframe to a temporary CSV file
-    temp_csv_path = 'temp_feedback.csv'
-    feedback_df.to_csv(temp_csv_path, index=False, header=not st.session_state.feedback_csv_exists)
-    
-    # Display a download link for the user
-    st.download_button(
-        label="Download Feedback CSV",
-        data=temp_csv_path,
-        key="download_feedback_csv",
-        on_click=download_feedback_csv,
-        args=(temp_csv_path,),
-        help="Click to download the feedback CSV file."
-    )
-
-    # Append feedback to the main CSV file
-    feedback_df.to_csv('feedback.csv', mode='a', index=False, header=not st.session_state.feedback_csv_exists)
-    st.session_state.feedback_csv_exists = True
-
-# Callback function for download button
-def download_feedback_csv(file_path):
-    st.markdown(f"Download [Feedback CSV File]({file_path})", unsafe_allow_html=True)
+def handle_feedback(feedback_data, feedback_file):
+    feedback_df = pd.DataFrame(feedback_data, columns=['Contract', 'Summary', '👍', '👎'])
+
+    # Save the dataframe to the feedback CSV file
+    feedback_df.to_csv(feedback_file, mode='a', index=False, header=not st.session_state.feedback_csv_exists)
+
+    # Display a download button for the user
+    download_button = st.button("Download Feedback CSV")
+
+    # If the download button is clicked, initiate the download
+    if download_button:
+        st.markdown(get_binary_file_downloader_html(feedback_file, 'Feedback Data'), unsafe_allow_html=True)
+
+# Function to create a download link for a binary file
+def get_binary_file_downloader_html(file_path, file_label):
+    with open(file_path, 'rb') as file:
+        file_content = file.read()
+    b64 = base64.b64encode(file_content).decode()
+    return f'<a href="data:file/csv;base64,{b64}" download="{file_label}.csv">Click here to download {file_label}</a>'
 
 # Main Streamlit app
 def main():
@@ -51,23 +47,23 @@ def main():
 
         # Feedback section
         st.subheader("Feedback:")
-        thumbs_up = st.button("👍 Thumbs Up")
-        thumbs_down = st.button("👎 Thumbs Down")
+        thumbs_up = st.button("👍")
+        thumbs_down = st.button("👎")
 
-        chosen = 1 if thumbs_up else 0
-        rejected = 1 if thumbs_down else 0
+        chosen = "👍" if thumbs_up else None
+        rejected = "👎" if thumbs_down else None
 
         feedback_data.append((contract_text, summary, chosen, rejected))
 
     # Handle feedback data
     if feedback_data:
-        handle_feedback(feedback_data)
+        feedback_file = 'feedback.csv'
+        st.session_state.feedback_csv_exists = True
+        handle_feedback(feedback_data, feedback_file)
 
 # Initialize feedback data
 feedback_data = []
 
 # Run the app
 if __name__ == "__main__":
-    # Create a Streamlit session state to persist variables across reruns
-    st.session_state.feedback_csv_exists = False
     main()