Update app.py

app.py

@@ -12,10 +12,30 @@ def generate_summary(contract_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'])
+    feedback_df = pd.DataFrame(feedback_data, columns=['Contract', 'Summary', '👍', '👎'])
+    
+    # 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)
+
 # Main Streamlit app
 def main():
     st.title("Legal Contract Summarizer with Feedback")
@@ -31,11 +51,11 @@ 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))