Update app2.py

app2.py

@@ -1811,11 +1811,6 @@ def lawyer_finder_ui():
             else:
                 st.warning(f"No lawyers found in {city}, {state}. Try selecting a different city or state.")
 
-import streamlit as st
-import pandas as pd
-import plotly.express as px
-# ... other imports ... 
-
 def analyze_policy(policy_text: str) -> Dict[str, Any]:
     """Analyzes the given policy text for its potential impact and benefits."""
     analysis_prompt = f"""
@@ -1851,17 +1846,25 @@ def analyze_policy(policy_text: str) -> Dict[str, Any]:
 
 def policy_analysis_ui():
     st.subheader("Policy Analysis & Impact")
-    with st.expander("How to use"):
-        st.write('''Enter the US policy text you want to analyze. 
-        LexAI will provide a comprehensive analysis of the policy's potential impact, benefits, drawbacks, and more.
-        ''')
+    st.write('''
+    Enter the US policy text you want to analyze or upload a document. 
+    LexAI will provide a comprehensive analysis of the policy's potential impact, benefits, drawbacks, and more.
+    ''')
+    
+    st.warning("Please do not upload files larger than 5MB as it may cause issues and consume all available tokens.")
     
     if 'policy_history' not in st.session_state:
         st.session_state.policy_history = []
 
-    display_chat_history() 
-
-    policy_text = st.text_area("Enter the US policy text:", height=200)
+    input_method = st.radio("Choose input method:", ("Text Input", "Document Upload"))
+    
+    policy_text = ""
+    if input_method == "Text Input":
+        policy_text = st.text_area("Enter the US policy text:", height=200)
+    else:
+        uploaded_file = st.file_uploader("Upload a document containing policy text (PDF, DOCX, or TXT)", type=["pdf", "docx", "txt"])
+        if uploaded_file is not None:
+            policy_text = extract_text_from_document(uploaded_file)
 
     if st.button("Analyze Policy"):
         if policy_text:
@@ -1874,56 +1877,147 @@ def policy_analysis_ui():
                     st.write("### Policy Analysis")
                     st.write(analysis_results.get("analysis", "No analysis available."))
 
+                    # Add download button for analysis
+                    analysis_text = analysis_results.get("analysis", "No analysis available.")
+                    st.download_button(
+                        label="Download Analysis",
+                        data=analysis_text,
+                        file_name="policy_analysis.txt",
+                        mime="text/plain"
+                    )
+
                     # Perform and display Wikipedia search
                     wiki_result = search_wikipedia(policy_text)
-                    st.session_state.policy_history.append({
-                        'type': 'wikipedia',
-                        'summary': wiki_result.get("summary", "No summary available."),
-                        'url': wiki_result.get("url", "")
-                    })
+                    st.write("### Wikipedia Summary")
+                    st.write(wiki_result.get("summary", "No summary available."))
+                    if wiki_result.get("url"):
+                        st.write(f"[Read more on Wikipedia]({wiki_result.get('url')})")
 
                     # Perform and display web search 
                     web_results = search_web_duckduckgo(policy_text)
+                    st.write("### Web Search Results")
+                    for result in web_results:
+                        st.write(f"**{result.get('title', 'No title')}**")
+                        st.write(result.get('snippet', 'No snippet available.'))
+                        st.write(f"[Read more]({result.get('link', '#')})")
+                        st.write("---")
+
                     st.session_state.policy_history.append({
-                        'type': 'web_search',
-                        'results': web_results
+                        'type': 'analysis',
+                        'text': policy_text,
+                        'analysis': analysis_results.get("analysis", "No analysis available."),
+                        'wikipedia': wiki_result,
+                        'web_results': web_results
                     })
 
-                    st.rerun()
+            st.rerun()
         else:
-            st.warning("Please enter policy text to analyze.")
+            st.warning("Please enter policy text or upload a document to analyze.")
 
+    display_chat_history()
 
-def legal_consultant_ui():
-    st.subheader("Legal Consultant")
-    with st.expander("How to use"):
-        st.write('''Describe your legal situation or ask your legal question related to US law. 
-        LexAI will provide information and guidance based on its understanding of the US legal system. 
-        Please remember that this is not a substitute for real legal advice from a qualified attorney.''')
+def provide_legal_advice(user_input: str) -> Dict[str, Any]:
+    """Provides legal advice based on the user input and performs web searches."""
+    advice_prompt = f"""
+    Provide legal advice based on US law for the following situation or question:
 
-    if 'consultant_history' not in st.session_state:
-        st.session_state.consultant_history = []
+    User Input:
+    ```
+    {user_input}
+    ```
 
-    display_chat_history()
+    Please include:
+    1. A summary of the legal issue or question
+    2. Relevant US laws or regulations that apply
+    3. Possible legal implications or consequences
+    4. General advice or next steps (without constituting specific legal counsel)
+    5. Any important disclaimers or limitations of this advice
 
-    user_input = st.text_area("Describe your legal situation or ask your question (US Law):", height=150)
+    Remember to maintain a professional and objective tone throughout your response.
+    """
 
-    if st.button("Get Legal Guidance"):
-        if user_input:
-            with st.spinner("Processing your request..."):
-                ai_response = get_ai_response(user_input)
-                st.session_state.consultant_history.append((user_input, ai_response))
+    try:
+        legal_advice = get_ai_response(advice_prompt)
+        web_results = search_web_duckduckgo(user_input, num_results=3)
+        
+        return {
+            "advice": legal_advice,
+            "web_results": web_results
+        }
+    except Exception as e:
+        return {"error": f"Error providing legal advice: {str(e)}"}
 
-                # Web search for additional resources
-                web_results = search_web_duckduckgo(user_input, num_results=3)  
-                st.session_state.consultant_history.append({
-                    'type': 'web_search',
-                    'results': web_results
-                })
+def legal_consultant_ui():
+    st.subheader("Legal Consultant")
+    
+    if 'chat_history' not in st.session_state:
+        st.session_state.chat_history = []
+    
+    if 'uploaded_document' not in st.session_state:
+        st.session_state.uploaded_document = None
+    
+    st.write('''
+    Describe your legal situation or ask your legal question related to US law. 
+    LexAI will provide information and guidance based on its understanding of the US legal system. 
+    Please remember that this is not a substitute for real legal advice from a qualified attorney.
+    ''')
 
-            st.rerun()
-        else:
-            st.warning("Please enter your legal situation or question.")
+    st.warning("Please do not upload files larger than 5MB as it may cause issues and consume all available tokens.")
+
+    # Document upload
+    uploaded_file = st.file_uploader("Upload a legal document (PDF, DOCX, or TXT)", type=["pdf", "docx", "txt"])
+    
+    if uploaded_file:
+        st.session_state.uploaded_document = extract_text_from_document(uploaded_file)
+        st.success("Document uploaded successfully!")
+    
+    display_chat_history_legal_advise()
+    
+    user_input = st.text_input("Your legal question:")
+    
+    if user_input and st.button("Send"):
+        with st.spinner("Processing your question..."):
+            if st.session_state.uploaded_document:
+                # If a document is uploaded, use it as context
+                full_input = f"Document context: {st.session_state.uploaded_document}\n\nUser question: {user_input}"
+            else:
+                full_input = user_input
+            
+            result = provide_legal_advice(full_input)
+            
+            if "error" in result:
+                st.error(result["error"])
+            else:
+                st.session_state.chat_history.append(("User", user_input))
+                st.session_state.chat_history.append(("Lex AI", result["advice"]))
+                
+                st.write("### Web Search Results")
+                for web_result in result["web_results"]:
+                    st.write(f"**{web_result.get('title', 'No title')}**")
+                    st.write(web_result.get('snippet', 'No snippet available.'))
+                    st.write(f"[Read more]({web_result.get('link', '#')})")
+                    st.write("---")
+        
+        st.rerun()
+
+def display_chat_history_legal_advise():
+    for entry in st.session_state.chat_history:
+        if isinstance(entry, tuple):
+            sender, message = entry
+            if sender == "User":
+                st.write(f"**You:** {message}")
+            else:
+                st.write(f"**AI:** {message}")
+        elif isinstance(entry, dict):
+            if entry.get('type') == 'web_search':
+                st.write("### Web Search Results")
+                for result in entry.get('results', []):
+                    st.write(f"**{result.get('title', 'No title')}**")
+                    st.write(result.get('snippet', 'No snippet available.'))
+                    st.write(f"[Read more]({result.get('link', '#')})")
+                    st.write("---")
+
+# Make sure to include the extract_text_from_document function as previously defined
 
 
 def draft_contract(contract_details: str) -> Dict[str, Any]:
@@ -1957,20 +2051,29 @@ def draft_contract(contract_details: str) -> Dict[str, Any]:
 
 def contract_drafting_ui():
     st.subheader("Contract Drafting Assistant")
-    with st.expander("How to use"):
-        st.write('''
-        Provide details about the contract you need drafted, including:
+    st.write('''
+    Provide details about the contract you need drafted, including:
 
-        * **Parties:** Names and addresses of all parties.
-        * **Subject Matter:** Clearly describe the goods, services, or purpose of the contract.
-        * **Key Terms:** Specify payment amounts, deadlines, delivery terms, or other crucial details.
-        * **Governing Law:** State the US state whose laws will govern the contract.
-        * **Additional Provisions:** Include any specific clauses, conditions, or requirements.
+    * Parties: Names and addresses of all parties.
+    * Subject Matter: Clearly describe the goods, services, or purpose of the contract.
+    * Key Terms: Specify payment amounts, deadlines, delivery terms, or other crucial details.
+    * Governing Law: State the US state whose laws will govern the contract.
+    * Additional Provisions: Include any specific clauses, conditions, or requirements.
 
-        Be as clear and thorough as possible to ensure the drafted contract meets your needs. 
-        ''')
+    Be as clear and thorough as possible to ensure the drafted contract meets your needs. 
+    ''')
 
-    contract_details = st.text_area("Enter contract details:", height=200)
+    st.warning("Please do not upload files larger than 5MB as it may cause issues and consume all available tokens.")
+
+    input_method = st.radio("Choose input method:", ("Text Input", "Document Upload"))
+    
+    contract_details = ""
+    if input_method == "Text Input":
+        contract_details = st.text_area("Enter contract details:", height=200)
+    else:
+        uploaded_file = st.file_uploader("Upload a document containing contract details (PDF, DOCX, or TXT)", type=["pdf", "docx", "txt"])
+        if uploaded_file is not None:
+            contract_details = extract_text_from_document(uploaded_file)
 
     if st.button("Draft Contract"):
         if contract_details:
@@ -1991,8 +2094,7 @@ def contract_drafting_ui():
                     mime="text/plain"
                 )
         else:
-            st.warning("Please enter contract details to proceed.")
-
+            st.warning("Please enter contract details or upload a document to proceed.")
 
 def analyze_case_for_prediction(case_details: str) -> Dict[str, Any]:
     """Analyzes the case details to provide a predictive analysis."""
@@ -2025,21 +2127,22 @@ def analyze_case_for_prediction(case_details: str) -> Dict[str, Any]:
 
 def predictive_analysis_ui():
     st.subheader("Predictive Case Analysis")
-    with st.expander("How to use"):
-        st.write('''Enter the details of your case, including:
+    st.write('''
+    Enter the details of your case, including:
+
+    * Facts: Briefly describe the key events that led to the legal dispute. 
+    * Legal Issues: State the specific legal questions or claims in the case.
+    * Relevant Law: Identify any relevant US laws, statutes, or regulations.
+    * Jurisdiction: Specify the US state where the case is filed.
 
-        * Facts: Briefly describe the key events that led to the legal dispute. 
-        * Legal Issues: State the specific legal questions or claims in the case.
-        * Relevant Law: Identify any relevant US laws, statutes, or regulations.
-        * Jurisdiction: Specify the US state where the case is filed.
+    LexAI will provide a predictive analysis, outlining potential outcomes, strengths and weaknesses of the case, and relevant US case law.
+    ''')
 
-        You can also upload a document with this information.
-        LexAI will provide a predictive analysis, outlining potential outcomes, strengths and weaknesses of the case, and relevant US case law.
-        ''')
+    st.warning("Please do not upload files larger than 5MB as it may cause issues and consume all available tokens.")
 
     input_method = st.radio("Choose input method:", ("Text Input", "Document Upload"))
+    
     case_details = ""
-
     if input_method == "Text Input":
         case_details = st.text_area("Enter case details:", height=200)
     else:
@@ -2057,6 +2160,25 @@ def predictive_analysis_ui():
                 st.error(analysis_results["error"])
             else:
                 st.write(analysis_results.get("analysis", "No analysis available."))
+
+                # Add download button for analysis
+                analysis_text = analysis_results.get("analysis", "No analysis available.")
+                st.download_button(
+                    label="Download Analysis",
+                    data=analysis_text,
+                    file_name="case_analysis.txt",
+                    mime="text/plain"
+                )
+
+                # Perform and display web search
+                web_results = search_web_duckduckgo(case_details)
+                st.write("### Related Web Resources")
+                for result in web_results:
+                    st.write(f"**{result.get('title', 'No title')}**")
+                    st.write(result.get('snippet', 'No snippet available.'))
+                    st.write(f"[Read more]({result.get('link', '#')})")
+                    st.write("---")
+
         else:
             st.warning("Please enter case details or upload a document to analyze.")
 
@@ -2112,7 +2234,7 @@ with st.sidebar:
     
     feature = st.selectbox(
         "Select a feature",
-        ["Legal Chatbot", "Document Analysis", "Case Precedent Finder", "Legal Cost Estimator", "Contract Analysis", "Case Trend Visualizer", "Case Information Retrieval", "Automated Legal Brief Generation", "Find the Lawyers", "Find the Lawyers", "Policy Analysis & Impact", "Legal Consultant", "Contract Drafting Assistant", "Predictive Case Analysis"]
+        ["Legal Chatbot", "Document Analysis", "Case Precedent Finder", "Legal Cost Estimator", "Contract Analysis", "Case Trend Visualizer", "Case Information Retrieval", "Automated Legal Brief Generation", "Find the Lawyers", "Policy Analysis & Impact", "Legal Consultant", "Contract Drafting Assistant", "Predictive Case Analysis"]
     )
 if feature == "Legal Chatbot":
     st.subheader("Legal Chatbot")