Update app.py

app.py

@@ -8,7 +8,6 @@ import difflib
 from sklearn.feature_extraction.text import TfidfVectorizer
 from sklearn.metrics.pairwise import cosine_similarity
 from sentence_transformers import SentenceTransformer, util
-from fpdf import FPDF
 
 # ========== CONFIGURATION ==========
 st.set_page_config(
@@ -105,32 +104,6 @@ def calculate_similarity(text1, text2):
         st.error(f"Similarity calculation error: {e}")
         return 0.0
 
-def generate_pdf_report(similarity_score, doc1, doc2):
-    pdf = FPDF()
-    pdf.add_page()
-    pdf.set_auto_page_break(auto=True, margin=15)
-
-    pdf.set_font("Arial", 'B', 16)
-    pdf.cell(0, 10, "Contract Comparison Report", ln=True, align="C")
-
-    pdf.set_font("Arial", '', 12)
-    pdf.ln(10)
-    pdf.multi_cell(0, 10, f"Document Similarity Score: {similarity_score:.2f}%")
-
-    pdf.ln(5)
-    pdf.set_font("Arial", 'B', 12)
-    pdf.cell(0, 10, "Document 1 Excerpt:", ln=True)
-    pdf.set_font("Arial", '', 10)
-    pdf.multi_cell(0, 10, doc1[:1000])
-
-    pdf.ln(5)
-    pdf.set_font("Arial", 'B', 12)
-    pdf.cell(0, 10, "Document 2 Excerpt:", ln=True)
-    pdf.set_font("Arial", '', 10)
-    pdf.multi_cell(0, 10, doc2[:1000])
-
-    return pdf.output(dest='S').encode('latin1')
-
 def load_contract(file):
     if file is None:
         return ""
@@ -229,20 +202,6 @@ def main():
                 st.markdown("### Modified Document")
                 st.markdown(f'<div style="border:1px solid #ccc; padding:10px; white-space: pre-wrap; font-family: monospace; font-size: 0.9em; max-height: 500px; overflow-y: auto;">{st.session_state.comparison_results["highlighted_diff2"]}</div>', unsafe_allow_html=True)
 
-            if st.button("Download PDF Report"):
-                with st.spinner("Generating report..."):
-                    pdf_bytes = generate_pdf_report(
-                        st.session_state.comparison_results['similarity_score'],
-                        contract_text1,
-                        contract_text2
-                    )
-                    st.download_button(
-                        label="Click to download PDF",
-                        data=pdf_bytes,
-                        file_name="contract_comparison_report.pdf",
-                        mime="application/pdf"
-                    )
-
     st.header("3. Clause Analysis")
 
     try:
@@ -293,4 +252,4 @@ def main():
             st.success(st.session_state.analysis_results.get('doc2', 'No analysis performed yet'))
 
 if __name__ == "__main__":
-    main()
+    main()