Update app.py

app.py

@@ -107,33 +107,31 @@ def save_uploaded_file(uploaded_file, path):
 
 # Streamlit UI
 st.title('PDF Similarity Checker')
-
+confirmationEdit = Modal("Contract Comparizer", key= "popUp_edit")
 col1, col2 = st.columns(2)
-
 # Clear the templates and contracts folders before uploading new files
 templates_folder = './templates'
 contracts_folder = './contracts'
-
-clear_folder(templates_folder)
-clear_folder(contracts_folder)
+SimilarityCalculator.clear_folder(templates_folder)
+SimilarityCalculator.clear_folder(contracts_folder)
 
 with col1:
     st.header("Upload Templates")
-    uploaded_files_templates = st.file_uploader("Choose PDF files", accept_multiple_files=True, type=['pdf'])
+    uploaded_files_templates = st.file_uploader("PDF Template", accept_multiple_files=True, type=['pdf'])
     os.makedirs(templates_folder, exist_ok=True)
     for uploaded_file in uploaded_files_templates:
-        if save_uploaded_file(uploaded_file, templates_folder):
+        if SimilarityCalculator.save_uploaded_file(uploaded_file, templates_folder):
             st.write(f"Saved: {uploaded_file.name}")
 
 with col2:
     st.header("Upload Contracts")
-    uploaded_files_contracts = st.file_uploader("Choose PDF files", key="contracts", accept_multiple_files=True, type=['pdf'])
+    uploaded_files_contracts = st.file_uploader("PDF Contracts", key="contracts", accept_multiple_files=True, type=['pdf'])
     os.makedirs(contracts_folder, exist_ok=True)
     for uploaded_file in uploaded_files_contracts:
-        if save_uploaded_file(uploaded_file, contracts_folder):
+        if SimilarityCalculator.save_uploaded_file(uploaded_file, contracts_folder):
             st.write(f"Saved: {uploaded_file.name}")
 
-model_name = st.selectbox("Select Model", ['sentence-transformers/multi-qa-mpnet-base-dot-v1'], index=0)
+model_name = st.selectbox("Select Model", ['sentence-transformers/all-mpnet-base-v2','sentence-transformers/all-MiniLM-L6-v2','sentence-transformers/multi-qa-mpnet-base-dot-v1','sentence-transformers/multi-qa-MiniLM-L6-cos-v1'], index=0)
 
 if st.button("Compute Similarities"):
     pdf_processor = PDFProcessor()
@@ -150,7 +148,7 @@ if st.button("Compute Similarities"):
     contract_embeddings = embedding_processor.get_embeddings(contract_texts)
 
     # Compute similarities
-    similarities = compute_similarity(template_embeddings, contract_embeddings)
+    similarities = SimilarityCalculator.compute_similarity(template_embeddings, contract_embeddings)
 
     # Display results in a table format
     similarity_data = []
@@ -166,20 +164,23 @@ if st.button("Compute Similarities"):
     # Create a DataFrame for the table
     columns = ["SI No", "Contract"] + [os.path.basename(template_files[j]) for j in range(len(template_files))]
     similarity_df = pd.DataFrame(similarity_data, columns=columns)
-
-    # Display maximize option
-    if st.checkbox("Maximize Table View"):
-        st.write("Similarity Scores Table (Maximized):")
-        st.dataframe(similarity_df)  # Maximized view
+    if similarity_df.empty:
+        st.write("No similarities computed.")
     else:
-        st.write("Similarity Scores Table:")
-        st.table(similarity_df.style.hide(axis="index"))  # Normal view
-
-    # Download option
-    csv = similarity_df.to_csv(index=False).encode('utf-8')
-    st.download_button(
-        label="Download Similarity Table as CSV",
-        data=csv,
-        file_name='similarity_scores.csv',
-        mime='text/csv',
-    )
+        with confirmationEdit.container():
+            st.write("Similarity Scores Table:")
+            st.table(similarity_df.style.hide(axis="index"))
+
+            if st.button('Close Window'):
+                confirmationEdit.close()
+
+    submitted = st.button("Show Result")
+    if submitted:
+          confirmationEdit.open()
+    if confirmationEdit.is_open():
+        with confirmationEdit.container():
+            st.write("Similarity Scores Table:")
+            st.table(similarity_df.style.hide(axis="index"))
+
+            if st.button('Close Result'):
+                confirmationEdit.close()