Update app.py

app.py

@@ -1,33 +1,53 @@
 import streamlit as st
 import pandas as pd
-import os
+from transformers import pipeline
 
-# Function to store feedback in a file
-def store_feedback(contract_text, feedback, filename='feedback_data.csv'):
-    feedback_data = pd.read_csv(filename) if os.path.exists(filename) else pd.DataFrame(columns=['Contract', 'Feedback'])
+# Load the EasyTerms/legalSummerizerET model from Hugging Face
+summarizer = pipeline("summarization", model="EasyTerms/legalSummerizerET")
 
-    feedback_data = feedback_data.append({'Contract': contract_text, 'Feedback': feedback}, ignore_index=True)
-    feedback_data.to_csv(filename, index=False)
+# Function to generate summary using the EasyTerms/legalSummerizerET model
+def generate_summary(contract_text):
+    summary = summarizer(contract_text, max_length=150, min_length=50, length_penalty=2.0, num_beams=4, early_stopping=True)
+    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'])
+    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
 
 # Main Streamlit app
 def main():
-    st.title("Contract Summarizer with Feedback")
+    st.title("Legal Contract Summarizer with Feedback")
+
+    # Input area for legal contract
+    contract_text = st.text_area("Enter the legal contract:")
+
+    # Button to generate summary
+    if st.button("Generate Summary"):
+        summary = generate_summary(contract_text)
+        st.subheader("Generated Summary:")
+        st.write(summary)
+
+        # Feedback section
+        st.subheader("Feedback:")
+        thumbs_up = st.button("👍 Thumbs Up")
+        thumbs_down = st.button("👎 Thumbs Down")
 
-    # Get contract input from user
-    contract_text = st.text_area("Paste your contract here:")
+        chosen = 1 if thumbs_up else 0
+        rejected = 1 if thumbs_down else 0
 
-    # Display the input contract
-    st.subheader("Input Contract:")
-    st.write(contract_text)
+        feedback_data.append((contract_text, summary, chosen, rejected))
 
-    # Get feedback from the user using emojis
-    feedback = st.radio("Feedback:", ['👍', '👎'])
+    # Handle feedback data
+    if feedback_data:
+        handle_feedback(feedback_data)
 
-    if st.button("Submit Feedback"):
-        # Store the feedback in a file
-        store_feedback(contract_text, feedback)
-        st.success("Feedback submitted successfully!")
+# 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()