Update app2.py

app2.py

@@ -557,211 +557,85 @@ def extract_cost_estimates(text: str) -> List[str]:
     
     return estimates
 
-def generate_legal_form(form_type: str, user_details: Dict[str, str], nation: str, state: str = None) -> Dict[str, Any]:
-    """
-    Generates a legal form based on user input, nation, and state (if applicable).
-    Creates downloadable .txt and .docx files.
-    """
-    current_date = datetime.now().strftime("%B %d, %Y")
+def legal_cost_estimator_ui():
+    st.subheader("Legal Cost Estimator")
     
-    # Helper function to get jurisdiction-specific clauses
-    def get_jurisdiction_clauses(form_type, nation, state):
-        # This would ideally be a comprehensive database of clauses for different jurisdictions
-        # For demonstration, we'll use a simplified version
-        clauses = {
-            "USA": {
-                "Power of Attorney": "This Power of Attorney is governed by the laws of the State of {state}.",
-                "Non-Disclosure Agreement": "This Agreement shall be governed by and construed in accordance with the laws of the State of {state}.",
-                "Simple Will": "This Will shall be construed in accordance with the laws of the State of {state}.",
-                "Lease Agreement": "This Lease Agreement is subject to the landlord-tenant laws of the State of {state}.",
-                "Employment Contract": "This Employment Contract is governed by the labor laws of the State of {state}."
-            },
-            "UK": {
-                "Power of Attorney": "This Power of Attorney is governed by the laws of England and Wales.",
-                "Non-Disclosure Agreement": "This Agreement shall be governed by and construed in accordance with the laws of England and Wales.",
-                "Simple Will": "This Will shall be construed in accordance with the laws of England and Wales.",
-                "Lease Agreement": "This Lease Agreement is subject to the landlord and tenant laws of England and Wales.",
-                "Employment Contract": "This Employment Contract is governed by the employment laws of England and Wales."
-            },
-            # Add more countries as needed
-        }
-        return clauses.get(nation, {}).get(form_type, "").format(state=state)
-
-    jurisdiction_clause = get_jurisdiction_clauses(form_type, nation, state)
+    case_type = st.selectbox("Select case type", ["Civil Litigation", "Criminal Defense", "Family Law", "Corporate Law"])
+    complexity = st.selectbox("Select case complexity", ["Simple", "Moderate", "Complex"])
+    country = st.selectbox("Select country", ["USA", "UK", "Canada"])
     
-    if form_type == "Power of Attorney":
-        form_content = f"""
-POWER OF ATTORNEY
-
-This Power of Attorney is made on {current_date}.
-
-I, {user_details['principal_name']}, hereby appoint {user_details['agent_name']} as my Attorney-in-Fact ("Agent").
-
-My Agent shall have full power and authority to act on my behalf. This power and authority shall authorize my Agent to manage and conduct all of my affairs and to exercise all of my legal rights and powers, including all rights and powers that I may acquire in the future. My Agent's powers shall include, but not be limited to:
-
-1. {', '.join(user_details['powers'])}
-
-This Power of Attorney shall become effective immediately and shall continue until it is revoked by me.
-
-{jurisdiction_clause}
-
-Signed this {current_date}.
-
-______________________
-{user_details['principal_name']} (Principal)
-
-______________________
-{user_details['agent_name']} (Agent)
-
-______________________
-Witness
-
-______________________
-Witness
-"""
-
-    elif form_type == "Non-Disclosure Agreement":
-        form_content = f"""
-NON-DISCLOSURE AGREEMENT
-
-This Non-Disclosure Agreement (the "Agreement") is entered into on {current_date} by and between:
-
-{user_details['party_a']} ("Party A")
-and
-{user_details['party_b']} ("Party B")
-
-1. Purpose: This Agreement is entered into for the purpose of {user_details['purpose']}.
-
-2. Confidential Information: Both parties may disclose certain confidential and proprietary information to each other in connection with the Purpose.
-
-3. Non-Disclosure: Both parties agree to keep all Confidential Information strictly confidential and not to disclose such information to any third parties for a period of {user_details['duration']} years from the date of this Agreement.
-
-{jurisdiction_clause}
-
-IN WITNESS WHEREOF, the parties hereto have executed this Non-Disclosure Agreement as of the date first above written.
-
-______________________
-{user_details['party_a']}
-
-______________________
-{user_details['party_b']}
-"""
-
-    elif form_type == "Simple Will":
-        beneficiaries = user_details['beneficiaries'].split('\n')
-        beneficiary_clauses = "\n".join([f"{i+1}. I give, devise, and bequeath to {b.strip()} [insert specific bequest or share of estate]." for i, b in enumerate(beneficiaries)])
+    if country == "USA":
+        state = st.selectbox("Select state", ["California", "New York", "Texas", "Florida"])
+    else:
+        state = None
+    
+    if st.button("Estimate Costs"):
+        with st.spinner("Estimating costs and performing web search..."):
+            cost_estimate = estimate_legal_costs(case_type, complexity, country, state)
         
-        form_content = f"""
-LAST WILL AND TESTAMENT
-
-I, {user_details['testator_name']}, being of sound mind, do hereby make, publish, and declare this to be my Last Will and Testament, hereby revoking all previous wills and codicils made by me.
-
-1. EXECUTOR: I appoint {user_details['executor_name']} to be the Executor of this, my Last Will and Testament.
-
-2. BEQUESTS:
-{beneficiary_clauses}
-
-3. RESIDUARY ESTATE: I give, devise, and bequeath all the rest, residue, and remainder of my estate to [insert beneficiary or beneficiaries].
-
-4. POWERS OF EXECUTOR: I grant to my Executor full power and authority to sell, lease, mortgage, or otherwise dispose of the whole or any part of my estate.
-
-{jurisdiction_clause}
-
-IN WITNESS WHEREOF, I have hereunto set my hand to this my Last Will and Testament on {current_date}.
-
-______________________
-{user_details['testator_name']} (Testator)
-
-WITNESSES:
-On the date last above written, {user_details['testator_name']}, known to us to be the Testator, signed this Will in our presence and declared it to be their Last Will and Testament. At the Testator's request, in the Testator's presence, and in the presence of each other, we have signed our names as witnesses:
-
-______________________
-Witness 1
-
-______________________
-Witness 2
-"""
-
-    elif form_type == "Lease Agreement":
-        form_content = f"""
-LEASE AGREEMENT
-
-This Lease Agreement (the "Lease") is made on {current_date} by and between:
-
-{user_details['landlord_name']} ("Landlord")
-and
-{user_details['tenant_name']} ("Tenant")
-
-1. PREMISES: The Landlord hereby leases to the Tenant the property located at {user_details['property_address']}.
-
-2. TERM: The term of this Lease shall be for {user_details['lease_term']} months, beginning on {user_details['start_date']} and ending on {user_details['end_date']}.
-
-3. RENT: The Tenant shall pay rent in the amount of {user_details['rent_amount']} per month, due on the {user_details['rent_due_day']} day of each month.
-
-4. SECURITY DEPOSIT: The Tenant shall pay a security deposit of {user_details['security_deposit']} upon signing this Lease.
-
-{jurisdiction_clause}
-
-IN WITNESS WHEREOF, the parties hereto have executed this Lease Agreement as of the date first above written.
-
-______________________
-{user_details['landlord_name']} (Landlord)
-
-______________________
-{user_details['tenant_name']} (Tenant)
-"""
-
-    elif form_type == "Employment Contract":
-        form_content = f"""
-EMPLOYMENT CONTRACT
-
-This Employment Contract (the "Contract") is made on {current_date} by and between:
-
-{user_details['employer_name']} ("Employer")
-and
-{user_details['employee_name']} ("Employee")
-
-1. POSITION: The Employee is hired for the position of {user_details['job_title']}.
-
-2. DUTIES: The Employee's duties shall include, but are not limited to: {user_details['job_duties']}.
-
-3. COMPENSATION: The Employee shall be paid a {user_details['pay_frequency']} salary of {user_details['salary_amount']}.
+        st.write("### Estimated Legal Costs")
+        for key, value in cost_estimate["cost_breakdown"].items():
+            if key != "Web search estimates":
+                st.write(f"**{key}:** {value}")
+        
+        st.write("### Web Search Estimates")
+        if cost_estimate["cost_breakdown"]["Web search estimates"]:
+            for result in cost_estimate["cost_breakdown"]["Web search estimates"]:
+                st.write(f"**Source:** [{result['source']}]({result['link']})")
+                st.write("**Estimated Costs:**")
+                for estimate in result['estimates']:
+                    st.write(f"- {estimate}")
+                st.write("---")
+        else:
+            st.write("No specific cost estimates found from web search.")
+        
+        st.write("### Potential High-Cost Areas")
+        for area in cost_estimate["high_cost_areas"]:
+            st.write(f"- {area}")
+        
+        st.write("### Cost-Saving Tips")
+        for tip in cost_estimate["cost_saving_tips"]:
+            st.write(f"- {tip}")
+        
+        st.write("### Recommended Lawyers/Law Firms")
+        for lawyer in cost_estimate["lawyer_recommendations"][:5]:  # Display top 5 recommendations
+            st.write(f"**[{lawyer['title']}]({lawyer['link']})**")
+            st.write(lawyer["snippet"])
+            st.write("---")
 
-4. TERM: This Contract shall commence on {user_details['start_date']} and continue until terminated by either party.
+def analyze_contract(contract_text: str) -> Dict[str, Any]:
+    """Analyzes the contract text for clauses, benefits, and potential exploits."""
 
-5. BENEFITS: The Employee shall be entitled to the following benefits: {user_details['benefits']}.
+    analysis_prompt = f"""
+    Analyze the following contract, identifying clauses that are favorable and unfavorable to each party involved. 
+    Highlight potential areas of concern or clauses that could be exploited. 
+    Provide specific examples within the contract to support your analysis.
 
-{jurisdiction_clause}
+    **Contract Text:**
+    {contract_text}
+    """
 
-IN WITNESS WHEREOF, the parties hereto have executed this Employment Contract as of the date first above written.
+    analysis = get_ai_response(analysis_prompt)
+    return {"analysis": analysis}
 
-______________________
-{user_details['employer_name']} (Employer)
 
-______________________
-{user_details['employee_name']} (Employee)
-"""
+def contract_analysis_ui():
+    st.subheader("Contract Analyzer")
 
-    else:
-        return {"error": "Unsupported form type"}
+    uploaded_file = st.file_uploader(
+        "Upload a contract document (PDF, DOCX, or TXT)",
+        type=["pdf", "docx", "txt"],
+    )
 
-    # Generate .txt file
-    txt_file = io.StringIO()
-    txt_file.write(form_content)
-    txt_file.seek(0)
+    if uploaded_file:
+        contract_text = analyze_document(uploaded_file)
 
-    # Generate .docx file
-    docx_file = io.BytesIO()
-    doc = Document()
-    doc.add_paragraph(form_content)
-    doc.save(docx_file)
-    docx_file.seek(0)
+        if st.button("Analyze Contract"):
+            with st.spinner("Analyzing contract..."):
+                analysis_results = analyze_contract(contract_text)
 
-    return {
-        "form_content": form_content,
-        "txt_file": txt_file,
-        "docx_file": docx_file
-    }
+            st.write("### Contract Analysis")
+            st.write(analysis_results.get("analysis", "No analysis available."))
 
 CASE_TYPES = [
     "Civil Rights", "Contract", "Real Property", "Tort", "Labor", "Intellectual Property", 
@@ -883,7 +757,7 @@ with st.sidebar:
     
     feature = st.selectbox(
         "Select a feature",
-        ["Legal Chatbot", "Document Analysis", "Case Precedent Finder", "Legal Cost Estimator", "Legal Form Generator", "Case Trend Visualizer"]
+        ["Legal Chatbot", "Document Analysis", "Case Precedent Finder", "Legal Cost Estimator", "Contract Analysis", "Case Trend Visualizer"]
     )
 if feature == "Legal Chatbot":
     st.subheader("Legal Chatbot")
@@ -1088,79 +962,8 @@ elif feature == "Legal Cost Estimator":
     else:
         st.write("Click 'Estimate Costs' to see the results.")
 
-elif feature == "Legal Form Generator":
-    st.subheader("Legal Form Generator")
-    
-    form_type = st.selectbox("Select form type", ["Power of Attorney", "Non-Disclosure Agreement", "Simple Will", "Lease Agreement", "Employment Contract"], key="form_generator_type")
-    
-    nation = st.selectbox("Select nation", ["USA", "UK"], key="form_generator_nation")
-    if nation == "USA":
-        state = st.selectbox("Select state", ["California", "New York", "Texas", "Florida"], key="form_generator_state")
-    else:
-        state = None
-    
-    user_details = {}
-    if form_type == "Power of Attorney":
-        user_details["principal_name"] = st.text_input("Principal's Full Name:")
-        user_details["agent_name"] = st.text_input("Agent's Full Name:")
-        user_details["powers"] = st.multiselect("Select powers to grant", ["Financial Decisions", "Healthcare Decisions", "Real Estate Transactions"])
-    elif form_type == "Non-Disclosure Agreement":
-        user_details["party_a"] = st.text_input("First Party's Name:")
-        user_details["party_b"] = st.text_input("Second Party's Name:")
-        user_details["purpose"] = st.text_input("Purpose of Disclosure:")
-        user_details["duration"] = st.number_input("Duration of Agreement (in years):", min_value=1, max_value=10)
-    elif form_type == "Simple Will":
-        user_details["testator_name"] = st.text_input("Testator's Full Name:")
-        user_details["beneficiaries"] = st.text_area("List Beneficiaries (one per line):")
-        user_details["executor_name"] = st.text_input("Executor's Full Name:")
-    elif form_type == "Lease Agreement":
-        user_details["landlord_name"] = st.text_input("Landlord's Full Name:")
-        user_details["tenant_name"] = st.text_input("Tenant's Full Name:")
-        user_details["property_address"] = st.text_input("Property Address:")
-        user_details["lease_term"] = st.number_input("Lease Term (in months):", min_value=1, max_value=60)
-        user_details["start_date"] = st.date_input("Lease Start Date:")
-        user_details["end_date"] = st.date_input("Lease End Date:")
-        user_details["rent_amount"] = st.number_input("Monthly Rent Amount:", min_value=0)
-        user_details["rent_due_day"] = st.number_input("Rent Due Day of Month:", min_value=1, max_value=31)
-        user_details["security_deposit"] = st.number_input("Security Deposit Amount:", min_value=0)
-    elif form_type == "Employment Contract":
-        user_details["employer_name"] = st.text_input("Employer's Full Name:")
-        user_details["employee_name"] = st.text_input("Employee's Full Name:")
-        user_details["job_title"] = st.text_input("Job Title:")
-        user_details["job_duties"] = st.text_area("Job Duties:")
-        user_details["pay_frequency"] = st.selectbox("Pay Frequency:", ["Weekly", "Bi-weekly", "Monthly"])
-        user_details["salary_amount"] = st.number_input("Salary Amount:", min_value=0)
-        user_details["start_date"] = st.date_input("Employment Start Date:")
-        user_details["benefits"] = st.text_area("Employee Benefits:")
-    
-    if st.button("Generate Form"):
-        generated_form = generate_legal_form(form_type, user_details, nation, state)
-        
-        if "error" in generated_form:
-            st.error(generated_form["error"])
-        else:
-            st.write("### Generated Legal Form:")
-            st.text(generated_form["form_content"])
-            
-            # Provide download buttons for .txt and .docx files
-            txt_download = generated_form["txt_file"].getvalue()
-            docx_download = generated_form["docx_file"].getvalue()
-            
-            st.download_button(
-                label="Download as .txt",
-                data=txt_download,
-                file_name=f"{form_type.lower().replace(' ', '_')}_{nation}{'_' + state if state else ''}.txt",
-                mime="text/plain"
-            )
-            
-            st.download_button(
-                label="Download as .docx",
-                data=docx_download,
-                file_name=f"{form_type.lower().replace(' ', '_')}_{nation}{'_' + state if state else ''}.docx",
-                mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-            )
-            
-            st.warning("Please note: This generated form is a template based on general principles of the selected jurisdiction. It should be reviewed by a legal professional licensed in the relevant jurisdiction before use.")
+elif feature == "Contract Analysis":
+    contract_analysis_ui()
 
 elif feature == "Case Trend Visualizer":
     st.subheader("Case Trend Visualizer")