Update app2.py

app2.py

@@ -1812,37 +1812,40 @@ def lawyer_finder_ui():
                 st.warning(f"No lawyers found in {city}, {state}. Try selecting a different city or state.")
 
 def analyze_policy(policy_text: str) -> Dict[str, Any]:
-    """Analyzes the given policy text for its potential impact and benefits."""
-    analysis_prompt = f"""
-    Analyze the following policy text, taking into account US legal and societal contexts:
+    chunks = split_text(policy_text)
+    full_analysis = ""
 
-    Policy Text:
-    ```
-    {policy_text}
-    ```
+    for i, chunk in enumerate(chunks):
+        analysis_prompt = f"""
+        Analyze the following part of the policy text ({i+1}/{len(chunks)}), taking into account US legal and societal contexts:
 
-    Provide a comprehensive analysis that includes:
-
-    * **Summary:** Briefly summarize the key points of the policy.
-    * **Large-Scale Impact:** Discuss the potential impact of this policy on a national or state level. Consider economic, social, and legal implications.
-    * **Small-Scale Impact:** Analyze how this policy might affect individuals, families, or specific communities within the US.
-    * **Long-Term Benefits:** What are the potential advantages of this policy in the long run (5-10 years or more)? 
-    * **Short-Term Benefits:** What benefits might be observed within the first few years of implementing this policy?
-    * **Potential Drawbacks:** Are there any possible negative consequences or challenges in implementing this policy?
-    * **Legal Considerations:**  Are there any existing US laws or regulations that this policy might conflict with or be affected by?
-    * **Historical Context:** Are there any historical parallels or past US policies that might inform the potential outcomes of this policy?
-    * **Suggestions for Improvement:**  Offer specific recommendations on how the policy could be modified to enhance its effectiveness or mitigate potential drawbacks.
-    * **Stakeholder Perspectives:** Identify different groups or entities in the US that might have an interest in this policy (e.g., businesses, consumers, government agencies), and how they might view it.
-
-    Support your analysis with relevant examples, statistics, or legal precedents from the US, where applicable. 
-    Maintain a neutral and objective tone, presenting both potential advantages and disadvantages of the policy.
-    """
+        Policy Text (Part {i+1}/{len(chunks)}):
+        ```
+        {chunk}
+        ```
+
+        Provide a comprehensive analysis that includes:
+        * **Summary:** Briefly summarize the key points of the policy.
+        * **Large-Scale Impact:** Discuss the potential impact of this policy on a national or state level. Consider economic, social, and legal implications.
+        * **Small-Scale Impact:** Analyze how this policy might affect individuals, families, or specific communities within the US.
+        * **Long-Term Benefits:** What are the potential advantages of this policy in the long run (5-10 years or more)? 
+        * **Short-Term Benefits:** What benefits might be observed within the first few years of implementing this policy?
+        * **Potential Drawbacks:** Are there any possible negative consequences or challenges in implementing this policy?
+        * **Legal Considerations:**  Are there any existing US laws or regulations that this policy might conflict with or be affected by?
+        * **Historical Context:** Are there any historical parallels or past US policies that might inform the potential outcomes of this policy?
+        * **Suggestions for Improvement:**  Offer specific recommendations on how the policy could be modified to enhance its effectiveness or mitigate potential drawbacks.
+        * **Stakeholder Perspectives:** Identify different groups or entities in the US that might have an interest in this policy (e.g., businesses, consumers, government agencies), and how they might view it.
+        Support your analysis with relevant examples, statistics, or legal precedents from the US, where applicable. 
+        Maintain a neutral and objective tone, presenting both potential advantages and disadvantages of the policy.
+        """
 
-    try:
-        analysis = get_ai_response(analysis_prompt)
-        return {"analysis": analysis}
-    except Exception as e:
-        return {"error": f"Error analyzing policy: {str(e)}"}
+        try:
+            chunk_analysis = get_ai_response(analysis_prompt)
+            full_analysis += chunk_analysis + "\n\n"
+        except Exception as e:
+            return {"error": f"Error analyzing part {i+1} of the policy text: {str(e)}"}
+
+    return {"analysis": full_analysis}
 
 def policy_analysis_ui():
     st.subheader("Policy Analysis & Impact")
@@ -1852,9 +1855,6 @@ def policy_analysis_ui():
     ''')
     
     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 = []
 
     input_method = st.radio("Choose input method:", ("Text Input", "Document Upload"))
     
@@ -1874,84 +1874,81 @@ def policy_analysis_ui():
                 if "error" in analysis_results:
                     st.error(analysis_results["error"])
                 else:
+                    analysis = analysis_results.get("analysis", "No analysis available.")
                     st.write("### Policy Analysis")
-                    st.write(analysis_results.get("analysis", "No analysis available."))
+                    st.write(analysis)
 
-                    # Add download button for analysis
-                    analysis_text = analysis_results.get("analysis", "No analysis available.")
+                    # Download button
                     st.download_button(
-                        label="Download Analysis",
-                        data=analysis_text,
+                        label="Download Complete Analysis",
+                        data=analysis,
                         file_name="policy_analysis.txt",
                         mime="text/plain"
                     )
-
-                    # Perform and display Wikipedia search
-                    wiki_result = search_wikipedia(policy_text)
-                    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': 'analysis',
-                        'text': policy_text,
-                        'analysis': analysis_results.get("analysis", "No analysis available."),
-                        'wikipedia': wiki_result,
-                        'web_results': web_results
-                    })
-
-            st.rerun()
         else:
             st.warning("Please enter policy text or upload a document to analyze.")
 
-    display_chat_history()
-
-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:
+def generate_web_search_query(user_input: str) -> str:
+    """Generates a web search query based on user input using Falcon."""
+    prompt = f"""
+    Generate a concise and effective web search query that can be used 
+    to find additional information about the following legal situation 
+    related to US law:
 
-    User Input:
     ```
     {user_input}
-    ```
-
-    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
-
-    Remember to maintain a professional and objective tone throughout your 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
-        }
+        response = ai71.chat.completions.create(
+            model="tiiuae/falcon-180b-chat",
+            messages=[{"role": "user", "content": prompt}],
+            stream=False,
+        )
+        return response.choices[0].message.content.strip()
     except Exception as e:
-        return {"error": f"Error providing legal advice: {str(e)}"}
+        print(f"Error generating web search query: {str(e)}")
+        return user_input
+
+def provide_legal_advice(user_input: str) -> Dict[str, Any]:
+    """Provides legal advice, suggestions, and generates a web search query."""
+    chunks = split_text(user_input)
+    full_advice = ""
+
+    for i, chunk in enumerate(chunks):
+        advice_prompt = f"""
+        Provide legal advice based on US law for the following situation or question (part {i+1}/{len(chunks)}):
+
+        User Input (Part {i+1}/{len(chunks)}):
+        ```
+        {chunk}
+        ```
+
+        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
+        Remember to maintain a professional and objective tone throughout your response.
+
+        Suggestions: Provide 2-3 actionable suggestions that the user could consider based on the legal situation.
+        """
+
+        try:
+            chunk_advice = get_ai_response(advice_prompt)
+            full_advice += chunk_advice + "\n\n"
+        except Exception as e:
+            return {"error": f"Error providing legal advice (part {i+1}): {str(e)}"}
+
+    return {"advice": full_advice} 
 
 def legal_consultant_ui():
     st.subheader("Legal Consultant")
     
-    if 'chat_history' not in st.session_state:
-        st.session_state.chat_history = []
+    if 'consultant_chat_history' not in st.session_state:
+        st.session_state.consultant_chat_history = []
     
     if 'uploaded_document' not in st.session_state:
         st.session_state.uploaded_document = None
@@ -1964,90 +1961,97 @@ def legal_consultant_ui():
 
     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()
-    
+    for message in st.session_state.consultant_chat_history:
+        if isinstance(message, tuple):
+            user_msg, bot_msg = message
+            st.info(f"**You:** {user_msg}")
+            st.success(f"**Lex AI:** {bot_msg}")
+        elif isinstance(message, dict):
+            if message.get('type') == 'web_search':
+                web_results_msg = "Web Search Results:\n"
+                for result in message.get('results', []):
+                    web_results_msg += f"[{result.get('title', 'No title')}]({result.get('link', '#')})\n{result.get('snippet', 'No snippet available.')}\n\n"
+                st.success(f"**Lex AI:** {web_results_msg}") 
+
     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.session_state.consultant_chat_history.append((user_input, result["advice"]))
+                
+                # Generate web search query and perform search
+                search_query = generate_web_search_query(user_input)
+                web_results = search_web_duckduckgo(search_query)
                 
-                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.session_state.consultant_chat_history.append({
+                    'type': 'web_search',
+                    'results': web_results
+                }) 
         
         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("---")
+    for message in st.session_state.chat_history:
+        if isinstance(message, tuple):
+            user_msg, bot_msg = message
+            st.info(f"**You:** {user_msg}")
+            st.success(f"**Lex AI:** {bot_msg}")
+        elif isinstance(message, dict):
+            if message.get('type') == 'web_search':
+                web_results_msg = "Web Search Results:\n"
+                for result in message.get('results', []):
+                    web_results_msg += f"[{result.get('title', 'No title')}]({result.get('link', '#')})\n{result.get('snippet', 'No snippet available.')}\n\n"
+                st.success(f"**Lex AI:** {web_results_msg}") 
 
-# Make sure to include the extract_text_from_document function as previously defined
+def draft_contract(contract_details: str) -> Dict[str, Any]:
+    chunks = split_text(contract_details)
+    full_draft = ""
 
+    for i, chunk in enumerate(chunks):
+        drafting_prompt = f"""
+        Draft a legally sound and comprehensive contract based on the following details (part {i+1}/{len(chunks)}), ensuring compliance with US law.
 
-def draft_contract(contract_details: str) -> Dict[str, Any]:
-    """Drafts a contract based on the provided details."""
-    drafting_prompt = f"""
-    Draft a legally sound and comprehensive contract based on the following details, ensuring compliance with US law.
+        Contract Details (Part {i+1}/{len(chunks)}):
+        ```
+        {chunk}
+        ```
+
+        Output Format:
+        Present the drafted contract in a clear and organized manner, using sections and headings. 
+        Include the following essential clauses (and any others relevant to the provided details):
+        * Parties: Clearly identify the names and addresses of all parties entering into the contract.
+        * Term and Termination: Specify the duration of the contract and conditions for renewal or termination.
+        * Payment Terms: Outline payment details, including amounts, schedule, and methods. 
+        * Governing Law: State that the contract shall be governed by the laws of the state specified in the details.
+        * Dispute Resolution: Include a clause outlining how disputes will be handled (e.g., mediation, arbitration).
+        * Entire Agreement: State that the written contract represents the entire agreement between the parties.
+        * Signatures: Leave space for the dated signatures of all parties involved.
+        """
 
-    Contract Details:
-    ```
-    {contract_details}
-    ```
+        try:
+            chunk_draft = get_ai_response(drafting_prompt)
+            full_draft += chunk_draft + "\n\n"
 
-    Output Format:
-    Present the drafted contract in a clear and organized manner, using sections and headings. 
-    Include the following essential clauses (and any others relevant to the provided details):
-
-    * Parties: Clearly identify the names and addresses of all parties entering into the contract.
-    * Term and Termination: Specify the duration of the contract and conditions for renewal or termination.
-    * Payment Terms: Outline payment details, including amounts, schedule, and methods. 
-    * Governing Law: State that the contract shall be governed by the laws of the state specified in the details.
-    * Dispute Resolution: Include a clause outlining how disputes will be handled (e.g., mediation, arbitration).
-    * Entire Agreement: State that the written contract represents the entire agreement between the parties.
-    * Signatures: Leave space for the dated signatures of all parties involved.
-    """
+        except Exception as e:
+            return {"error": f"Error drafting contract (part {i+1}): {str(e)}"}
 
-    try:
-        contract_draft = get_ai_response(drafting_prompt)
-        return {"draft": contract_draft}
-    except Exception as e:
-        return {"error": f"Error drafting contract: {str(e)}"}
+    return {"draft": full_draft}
 
 def contract_drafting_ui():
     st.subheader("Contract Drafting Assistant")
@@ -2085,7 +2089,7 @@ def contract_drafting_ui():
             else:
                 st.write("### Drafted Contract")
                 contract_draft = draft_results.get("draft", "No draft available.")
-                st.text_area("Contract Draft", contract_draft, height=300)
+                st.write(contract_draft) 
 
                 st.download_button(
                     label="Download Contract",
@@ -2093,37 +2097,43 @@ def contract_drafting_ui():
                     file_name="drafted_contract.txt",
                     mime="text/plain"
                 )
+
         else:
             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."""
-    analysis_prompt = f"""
-    Analyze the following case details in the context of the US legal system and provide a predictive analysis.
+    chunks = split_text(case_details)
+    full_analysis = ""
 
-    Case Details:
-    ```
-    {case_details}
-    ```
+    for i, chunk in enumerate(chunks):
+        analysis_prompt = f"""
+        Analyze the following case details (part {i+1}/{len(chunks)}) in the context of the US legal system and provide a predictive analysis.
 
-    Your analysis should address the following:
+        Case Details (Part {i+1}/{len(chunks)}):
+        ```
+        {chunk}
+        ```
+
+        Your analysis should address the following:
+        * **Case Summary:** Briefly summarize the key facts, legal claims, and parties involved in the case.
+        * **Predicted Outcome:** What is the most likely outcome of this case based on the provided information, US legal precedents, and similar cases? Explain your reasoning.
+        * **Strengths of the Case:**  Identify the most compelling arguments and evidence that support a favorable outcome. 
+        * **Weaknesses of the Case:**  What are potential weaknesses in the case, or areas where the opposing party might have strong arguments? 
+        * **Areas of Caution:**  What potential pitfalls or challenges should be considered? What strategies could the opposing party use? 
+        * **Relevant US Case Law:** Cite specific US legal precedents and similar cases that support your analysis and predicted outcome. 
+        * **Recommended Strategies:** Offer specific, actionable recommendations on how to strengthen the case and increase the likelihood of a positive result. 
+
+        Please maintain a neutral and objective tone throughout your analysis. The goal is to provide a realistic assessment of the case, not to advocate for a particular side. 
+        """
 
-    * **Case Summary:** Briefly summarize the key facts, legal claims, and parties involved in the case.
-    * **Predicted Outcome:** What is the most likely outcome of this case based on the provided information, US legal precedents, and similar cases? Explain your reasoning.
-    * **Strengths of the Case:**  Identify the most compelling arguments and evidence that support a favorable outcome. 
-    * **Weaknesses of the Case:**  What are potential weaknesses in the case, or areas where the opposing party might have strong arguments? 
-    * **Areas of Caution:**  What potential pitfalls or challenges should be considered? What strategies could the opposing party use? 
-    * **Relevant US Case Law:** Cite specific US legal precedents and similar cases that support your analysis and predicted outcome. 
-    * **Recommended Strategies:** Offer specific, actionable recommendations on how to strengthen the case and increase the likelihood of a positive result. 
+        try:
+            chunk_analysis = get_ai_response(analysis_prompt)
+            full_analysis += chunk_analysis + "\n\n"
 
-    Please maintain a neutral and objective tone throughout your analysis. The goal is to provide a realistic assessment of the case, not to advocate for a particular side. 
-    """
+        except Exception as e:
+            return {"error": f"Error analyzing case (part {i+1}): {str(e)}"}
 
-    try:
-        analysis = get_ai_response(analysis_prompt)
-        return {"analysis": analysis}
-    except Exception as e:
-        return {"error": f"Error analyzing case: {str(e)}"}
+    return {"analysis": full_analysis}
 
 def predictive_analysis_ui():
     st.subheader("Predictive Case Analysis")
@@ -2159,26 +2169,16 @@ def predictive_analysis_ui():
             if "error" in analysis_results:
                 st.error(analysis_results["error"])
             else:
-                st.write(analysis_results.get("analysis", "No analysis available."))
+                analysis = analysis_results.get("analysis", "No analysis available.")
+                st.write(analysis)
 
-                # Add download button for analysis
-                analysis_text = analysis_results.get("analysis", "No analysis available.")
+                # Download button for analysis
                 st.download_button(
                     label="Download Analysis",
-                    data=analysis_text,
+                    data=analysis,
                     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.")
 
@@ -2419,7 +2419,7 @@ st.markdown("---")
 st.markdown(
     """
     <div style="text-align: center;">
-        <p>© 2023 Lex AI. All rights reserved.</p>
+        <p>© 2024 Lex AI. All rights reserved.</p>
         <p><small>Disclaimer: This tool provides general legal information and assistance. It is not a substitute for professional legal advice. Please consult with a qualified attorney for specific legal matters.</small></p>
     </div>
     """,