Update legal_document_analysis.py

legal_document_analysis.py

@@ -15,6 +15,7 @@ from fpdf import FPDF
 import getpass
 import pandas as pd
 import seaborn as sns
+import asyncio
 
 # Load environment variables from .env file
 load_dotenv()
@@ -374,56 +375,8 @@ def generate_pdf_analysis(document_text, summary, detected_clauses, hidden_oblig
 
     return pdf_file_path  # Return the path to the saved PDF
 
-# Function to handle chatbot interaction
-def chatbot_query(user_input):
-    try:
-        response = model({"text": user_input})
-        if isinstance(response, dict) and 'text' in response:
-            return response['text']
-        else:
-            return "Error: Unexpected response format."
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-# Function to generate suggestions for improvement
-def generate_suggestions(text):
-    suggestions = []
-    
-    if "shall" in text.lower():
-        suggestions.append("Consider replacing 'shall' with 'must' for clarity.")
-    if "may" in text.lower():
-        suggestions.append("Clarify the conditions under which actions 'may' be taken.")
-    if "if" in text.lower() and "then" not in text.lower():
-        suggestions.append("Ensure conditional statements are clear and complete.")
-    if "not" in text.lower():
-        suggestions.append("Review negative clauses to ensure they are not overly restrictive.")
-    
-    return suggestions
-
-# Function to send feedback via email
-def send_feedback(feedback_content):
-    sender_email = os.getenv("SENDER_EMAIL")
-    receiver_email = os.getenv("FEEDBACK_EMAIL")
-    password = os.getenv("EMAIL_PASS")
-
-    msg = MIMEMultipart()
-    msg['From'] = sender_email
-    msg['To'] = receiver_email
-    msg['Subject'] = "User Feedback on Legal Document Analysis"
-
-    msg.attach(MIMEText(feedback_content, 'plain'))
-
-    try:
-        with smtplib.SMTP('smtp.gmail.com', 587) as server:
-            server.starttls()
-            server.login(sender_email, password)
-            server.send_message(msg)
-        return True
-    except Exception as e:
-        return False
-
-# Function to send PDF via email
-def send_pdf_via_email(pdf_buffer, recipient_email):
+# Asynchronous function to send PDF via email
+async def send_pdf_via_email(pdf_buffer, recipient_email):
     sender_email = os.getenv("SENDER_EMAIL")
     password = os.getenv("EMAIL_PASS")
 
@@ -598,7 +551,9 @@ def display_legal_analysis_page():
             # Button to send PDF via email
             if st.button("Send PDF Analysis"):
                 if recipient_email:
-                    if send_pdf_via_email(pdf_buffer, recipient_email):
+                    # Send PDF asynchronously
+                    success = asyncio.run(send_pdf_via_email(pdf_buffer, recipient_email))
+                    if success:
                         st.success("PDF has been sent successfully!")
                     else:
                         st.error("Failed to send PDF. Please try again.")