contract verification

app.py

@@ -6,6 +6,7 @@ import pandas as pd
 from docx import Document
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from dotenv import load_dotenv
+import time
 
 # Load the OpenAI API key from environment variables
 load_dotenv()
@@ -29,18 +30,29 @@ def extract_terms_from_contract(contract_text):
         "Provide the extracted terms in JSON format."
     )
 
-    response = openai.ChatCompletion.create(
-        model="gpt-4",
-        messages=[
-            {"role": "system", "content": "You are an AI specialized in extracting structured data from text documents."},
-            {"role": "user", "content": prompt},
-        ],
-        max_tokens=4096,
-        n=1,
-        stop=None,
-        temperature=0.1,
-    )
-    return response.choices[0].message["content"]
+    retries = 5
+    for i in range(retries):
+        try:
+            response = openai.ChatCompletion.create(
+                model="gpt-4",
+                messages=[
+                    {"role": "system", "content": "You are an AI specialized in extracting structured data from text documents."},
+                    {"role": "user", "content": prompt},
+                ],
+                max_tokens=4096,
+                n=1,
+                stop=None,
+                temperature=0.1,
+            )
+            return response.choices[0].message["content"]
+        except openai.error.RateLimitError as e:
+            if i < retries - 1:
+                wait_time = 2 ** i
+                st.warning(f"Rate limit exceeded. Retrying in {wait_time} seconds...")
+                time.sleep(wait_time)
+            else:
+                st.error("Rate limit exceeded. Please try again later.")
+                return None
 
 def analyze_task_compliance(task_description, cost_estimate, contract_terms):
     prompt = (
@@ -54,23 +66,31 @@ def analyze_task_compliance(task_description, cost_estimate, contract_terms):
         "Provide the compliance analysis in a clear JSON format."
     )
 
-    response = openai.ChatCompletion.create(
-        model="gpt-4",
-        messages=[
-            {"role": "system", "content": "You are an AI specialized in analyzing text for compliance with specified conditions."},
-            {"role": "user", "content": prompt},
-        ],
-        max_tokens=4096,
-        n=1,
-        stop=None,
-        temperature=0.1,
-    )
-    
-    # Parse the response to extract structured explanations
-    response_content = response.choices[0].message["content"]
-    compliance_analysis = json.loads(response_content)
-    
-    return compliance_analysis
+    retries = 5
+    for i in range(retries):
+        try:
+            response = openai.ChatCompletion.create(
+                model="gpt-4",
+                messages=[
+                    {"role": "system", "content": "You are an AI specialized in analyzing text for compliance with specified conditions."},
+                    {"role": "user", "content": prompt},
+                ],
+                max_tokens=4096,
+                n=1,
+                stop=None,
+                temperature=0.1,
+            )
+            response_content = response.choices[0].message["content"]
+            compliance_analysis = json.loads(response_content)
+            return compliance_analysis
+        except openai.error.RateLimitError as e:
+            if i < retries - 1:
+                wait_time = 2 ** i
+                st.warning(f"Rate limit exceeded. Retrying in {wait_time} seconds...")
+                time.sleep(wait_time)
+            else:
+                st.error("Rate limit exceeded. Please try again later.")
+                return None
 
 def main():
     st.title("Contract Compliance Analyzer")
@@ -85,10 +105,16 @@ def main():
         xlsx_file = st.file_uploader("Upload Task Descriptions (XLSX)", type="xlsx")
 
     if docx_file and xlsx_file:
+        # Clear previous information
+        st.session_state.clear()
+
         # Extract contract text and terms
         contract_text = extract_text_from_docx(docx_file)
         extracted_terms_json = extract_terms_from_contract(contract_text)
         
+        if extracted_terms_json is None:
+            return
+        
         try:
             contract_terms = json.loads(extracted_terms_json)
         except json.JSONDecodeError as e:
@@ -111,7 +137,8 @@ def main():
             for future in as_completed(futures):
                 try:
                     result = future.result()
-                    compliance_results.append(result)
+                    if result is not None:
+                        compliance_results.append(result)
                 except Exception as e:
                     st.error(f"An error occurred: {e}")
 
@@ -144,3 +171,4 @@ def main():
 
 if __name__ == "__main__":
     main()
+