Update app2.py

app2.py

@@ -637,21 +637,31 @@ def legal_cost_estimator_ui():
         else:
             st.write("No web search results found for the selected criteria.")
 
+def split_text(text, max_chunk_size=4000):
+    return [text[i:i+max_chunk_size] for i in range(0, len(text), max_chunk_size)]
+
 def analyze_contract(contract_text: str) -> Dict[str, Any]:
     """Analyzes the contract text for clauses, benefits, and potential exploits."""
+    chunks = split_text(contract_text)
+    full_analysis = ""
 
-    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.
+    for i, chunk in enumerate(chunks):
+        analysis_prompt = f"""
+        Analyze the following part of the contract ({i+1}/{len(chunks)}), 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 this part of the contract to support your analysis.
 
-    **Contract Text:**
-    {contract_text}
-    """
+        **Contract Text (Part {i+1}/{len(chunks)}):**
+        {chunk}
+        """
 
-    analysis = get_ai_response(analysis_prompt)
-    return {"analysis": analysis}
+        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 contract: {str(e)}"}
 
+    return {"analysis": full_analysis}
 
 def contract_analysis_ui():
     st.subheader("Contract Analyzer")
@@ -662,15 +672,17 @@ def contract_analysis_ui():
     )
 
     if uploaded_file:
-        contract_text = analyze_document(uploaded_file)
+        contract_text = analyze_uploaded_document(uploaded_file)
 
         if st.button("Analyze Contract"):
             with st.spinner("Analyzing contract..."):
                 analysis_results = analyze_contract(contract_text)
 
             st.write("### Contract Analysis")
-            st.write(analysis_results.get("analysis", "No analysis available."))
-
+            if "error" in analysis_results:
+                st.error(analysis_results["error"])
+            else:
+                st.write(analysis_results.get("analysis", "No analysis available."))
 
 CASE_TYPES = [
     "Civil Rights", "Contract", "Real Property", "Tort", "Labor", "Intellectual Property",
@@ -1287,9 +1299,6 @@ def extract_text_from_document(uploaded_file):
         text = uploaded_file.getvalue().decode("utf-8")
     return text
 
-def split_text(text, max_chunk_size=4000):
-    return [text[i:i+max_chunk_size] for i in range(0, len(text), max_chunk_size)]
-
 def generate_legal_brief(case_info):
     chunks = split_text(case_info)
     full_brief = ""