contract verification

app.py

@@ -13,6 +13,20 @@ load_dotenv()
 api_key = os.getenv("OPENAI_API_KEY")
 openai.api_key = api_key
 
+# Streamlit app layout
+st.set_page_config(layout="wide")
+
+# Add custom CSS for center alignment
+st.markdown("""
+    <style>
+    .centered-title {
+        text-align: center;
+        font-size: 2.5em;
+        margin-top: 0;
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
 def extract_text_from_docx(docx_path):
     doc = Document(docx_path)
     return "\n".join([para.text for para in doc.paragraphs])
@@ -30,7 +44,8 @@ def extract_terms_from_contract(contract_text):
         "Provide the extracted terms in JSON format."
     )
 
-    retries = 5
+    retries = 2
+    wait_time = 1
     for i in range(retries):
         try:
             response = openai.ChatCompletion.create(
@@ -45,11 +60,11 @@ def extract_terms_from_contract(contract_text):
                 temperature=0.1,
             )
             return response.choices[0].message["content"]
-        except openai.error.RateLimitError as e:
+        except openai.error.RateLimitError:
             if i < retries - 1:
-                wait_time = 2 ** i
                 st.warning(f"Rate limit exceeded. Retrying in {wait_time} seconds...")
                 time.sleep(wait_time)
+                wait_time *= 2  # Exponential backoff
             else:
                 st.error("Rate limit exceeded. Please try again later.")
                 return None
@@ -67,6 +82,7 @@ def analyze_task_compliance(task_description, cost_estimate, contract_terms):
     )
 
     retries = 5
+    wait_time = 1
     for i in range(retries):
         try:
             response = openai.ChatCompletion.create(
@@ -79,32 +95,38 @@ def analyze_task_compliance(task_description, cost_estimate, contract_terms):
                 n=1,
                 stop=None,
                 temperature=0.1,
+                stream=True,
             )
-            response_content = response.choices[0].message["content"]
-            compliance_analysis = json.loads(response_content)
-            return compliance_analysis
-        except openai.error.RateLimitError as e:
+
+            compliance_analysis = ""
+            for chunk in response:
+                chunk_text = chunk['choices'][0]['delta'].get('content', '')
+                compliance_analysis += chunk_text
+                st.write(chunk_text)
+                st.json(chunk_text)
+
+            return json.loads(compliance_analysis)
+        except openai.error.RateLimitError:
             if i < retries - 1:
-                wait_time = 2 ** i
                 st.warning(f"Rate limit exceeded. Retrying in {wait_time} seconds...")
                 time.sleep(wait_time)
+                wait_time *= 2  # Exponential backoff
             else:
                 st.error("Rate limit exceeded. Please try again later.")
                 return None
 
 def main():
-    st.title("Contract Compliance Analyzer")
-
-    # File upload buttons in the same row
-    col1, col2 = st.columns(2)
-    
-    with col1:
-        docx_file = st.file_uploader("Upload Contract Document (DOCX)", type="docx")
-    
-    with col2:
-        xlsx_file = st.file_uploader("Upload Task Descriptions (XLSX)", type="xlsx")
-
-    if docx_file and xlsx_file:
+    st.markdown("<h1 class='centered-title'>Contract Compliance Analyzer</h1>", unsafe_allow_html=True)
+
+    # File upload buttons one after another
+    st.sidebar.file_uploader("Upload Contract Document (DOCX)", type="docx", key="docx_file")
+    st.sidebar.file_uploader("Upload Task Descriptions (XLSX or CSV)", type=["xlsx", "csv"], key="data_file")
+    submit_button = st.sidebar.button("Submit")
+
+    docx_file = st.session_state.get("docx_file")
+    data_file = st.session_state.get("data_file")
+
+    if submit_button and docx_file and data_file:
         # Clear previous information
         st.session_state.clear()
 
@@ -121,8 +143,11 @@ def main():
             st.error(f"JSON decoding error: {e}")
             return
 
-        # Read task descriptions and cost estimates from XLSX
-        tasks_df = pd.read_excel(xlsx_file)
+        # Read task descriptions and cost estimates from XLSX or CSV
+        if data_file.type == "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet":
+            tasks_df = pd.read_excel(data_file)
+        else:
+            tasks_df = pd.read_csv(data_file)
 
         compliance_results = []
         futures = []