Update legal_document_analysis.py

legal_document_analysis.py

@@ -408,13 +408,14 @@ def generate_pdf_analysis(document_text, summary, detected_clauses, hidden_oblig
     return pdf_buffer
 
 # Function to handle chatbot interaction
-def chatbot_query(user_input):
+def chatbot_query(user_input, document_text):
+    prompt = f"You are a legal expert. Answer the question based on the following document text:\n\n{document_text}\n\nQuestion: {user_input}"
     try:
-        response = model({"text": user_input})
-        if isinstance(response, dict) and 'text' in response:
-            return response['text']
+        response = model.invoke(prompt)
+        if hasattr(response, 'content'):
+            return response.content.strip()
         else:
-            return "Error: Unexpected response format."
+            return str(response).strip()
     except Exception as e:
         return f"Error: {str(e)}"
 
@@ -530,7 +531,7 @@ def display_legal_analysis_page():
             st.error("Unsupported file type!")
             return
 
-        tabs = st.tabs(["📄 Document Text", "🔍 Summary", "🔑 Key Clauses", "🔒 Hidden Obligations", "⚠ Risk Analysis", "💡 Suggestions & Chatbot", "🔄 Update Tracker", "📜 GDPR Updates"])
+        tabs = st.tabs(["📄 Document Text", "🔍 Summary", "🔑 Key Clauses", "🔒 Hidden Obligations", "⚠ Risk Analysis", "💡 Suggestions & Chatbot", "🔄 Update Tracker"])
 
         with tabs[0]:
             st.subheader("Document Text")
@@ -602,41 +603,12 @@ def display_legal_analysis_page():
             user_input = st.text_input("Ask the chatbot about your document:")
             if st.button("Send"):
                 if user_input:
-                    chatbot_response = chatbot_query(user_input)
+                    chatbot_response = chatbot_query(user_input, document_text)
                     st.write("*Chatbot Response:*")
                     st.write(chatbot_response)
                 else:
                     st.warning("Please enter a question.")
 
-            # Download PDF Analysis Button
-            st.subheader("Download Analysis as PDF")
-            pdf_buffer = generate_pdf_analysis(document_text, summary, detected_clauses, hidden_obligations, detected_risks, risk_assessment_matrix, risk_level_distribution, risks_by_type, stacked_bar_chart, risk_heatmap)
-            pdf_buffer.seek(0)
-
-            # Add download button for PDF
-            st.download_button(
-                label="Download PDF Analysis",
-                data=pdf_buffer,
-                file_name="legal_document_analysis.pdf",
-                mime="application/pdf"
-            )
-
-            # Input for recipient email
-            recipient_email = st.text_input("Enter your email address to receive the PDF:")
-
-            # Button to send PDF via email
-            if st.button("Send PDF Analysis"):
-                if recipient_email:
-                    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.")
-                else:
-                    st.warning("Please enter a valid email address.")
-
-            # Feedback Form Section
-            display_feedback_form()
-
         with tabs[6]:  # Update Tracker Tab
             st.subheader("Document Updates")
             updates = track_updates(document_text)
@@ -652,53 +624,6 @@ def display_legal_analysis_page():
                 else:
                     st.write("No updates detected.")
 
-        with tabs[7]:  # GDPR Updates Tab
-            st.subheader("GDPR Website Updates")
-            if st.button("Fetch Live Recitals"):
-                with st.spinner("Fetching updates..."):
-                    recitals = fetch_gdpr_recitals()
-                    if recitals:
-                        for number, details in recitals.items():
-                            st.markdown(f"*Recital {number}: {details['title']}*")
-                            st.write(details['content'])
-                    else:
-                        st.write("No recitals found.")
-
-# Function to fetch live recitals from the GDPR website
-def fetch_gdpr_recitals():
-    url = "https://gdpr-info.eu/recitals/"
-    response = requests.get(url)
-
-    # Check if the request was successful
-    if response.status_code != 200:
-        st.error("Failed to fetch data from the GDPR website.")
-        return {}
-
-    soup = BeautifulSoup(response.content, 'html.parser')
-
-    recitals = {}
-    # Locate all recital links
-    articles = soup.find_all('div', class_='artikel')
-    
-    # Extract each recital's link and title
-    for i, article in enumerate(articles):
-        if i >= 3:  # Limit to the first 3 recitals
-            break
-        link = article.find('a')['href']
-        number = article.find('span', class_='nummer').text.strip('()')
-        title = article.find('span', class_='titel').text.strip()
-        
-        # Fetch the content of each recital
-        rec_response = requests.get(link)
-        if rec_response.status_code == 200:
-            rec_soup = BeautifulSoup(rec_response.content, 'html.parser')
-            content = rec_soup.find('div', class_='entry-content').get_text(strip=True)
-            recitals[number] = {'title': title, 'content': content}
-        else:
-            print(f"Failed to fetch recital {number} from {link}")
-
-    return recitals
-
 # Run the application
 if __name__ == "__main__":
     display_legal_analysis_page()