Update legal_document_analysis.py

legal_document_analysis.py

@@ -15,7 +15,6 @@ from fpdf import FPDF
 import getpass
 import pandas as pd
 import seaborn as sns
-import asyncio
 
 # Load environment variables from .env file
 load_dotenv()
@@ -375,8 +374,56 @@ def generate_pdf_analysis(document_text, summary, detected_clauses, hidden_oblig
 
     return pdf_file_path  # Return the path to the saved PDF
 
-# Asynchronous function to send PDF via email
-async def send_pdf_via_email(pdf_buffer, recipient_email):
+# 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):
     sender_email = os.getenv("SENDER_EMAIL")
     password = os.getenv("EMAIL_PASS")
 
@@ -418,21 +465,6 @@ def get_update_suggestion(update):
     suggestion = generate_summary(prompt)
     return suggestion if suggestion else "No suggestion available."
 
-# 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 display feedback form
 def display_feedback_form():
     st.subheader("Feedback Form")
@@ -566,9 +598,7 @@ def display_legal_analysis_page():
             # Button to send PDF via email
             if st.button("Send PDF Analysis"):
                 if recipient_email:
-                    # Send PDF asynchronously
-                    success = asyncio.run(send_pdf_via_email(pdf_buffer, recipient_email))
-                    if success:
+                    if send_pdf_via_email(pdf_buffer, recipient_email):
                         st.success("PDF has been sent successfully!")
                     else:
                         st.error("Failed to send PDF. Please try again.")