Update app.py

app.py

@@ -116,17 +116,17 @@ else:
 
 
 '''
+
+
 import streamlit as st
 import json
-import base64
 import openai
 import fitz  # PyMuPDF
 import docx  # python-docx
 
 # Setup for the 3rd party OpenAI API
-base_url="https://api.aimlapi.com/v1"
-api_key='3fbfe25109b647efb7bf2f45bd667163'
-
+base_url = "https://api.aimlapi.com/v1"
+api_key = '3fbfe25109b647efb7bf2f45bd667163'
 
 # Set the API key for OpenAI
 openai.api_key = api_key
@@ -137,9 +137,8 @@ def call_ai_api(prompt, max_tokens=1000, temperature=0.7):
     Function to call the 3rd party OpenAI API.
     """
     try:
-        response = openai.ChatCompletion.create
-        #response = client.chat.completions.create(
-    model="gpt-3.5-turbo",  # or the appropriate model for your API
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",  # or the appropriate model for your API
             messages=[{"role": "user", "content": prompt}],
             max_tokens=max_tokens,
             temperature=temperature
@@ -203,9 +202,7 @@ def analyze_contract(file_content, file_type):
     
     for chunk in chunks:
         prompt = f"""Analyze the following contract section and provide a detailed breakdown of its clauses, including their titles, content, risk level, and a brief explanation for each. The document content is as follows:
-
 {chunk}
-
 Format your response as a JSON object with a 'clauses' key containing an array of clause objects. Each clause object should have 'title', 'content', 'risk_level', and 'explanation' keys.
 """
         # Call AI API to analyze each chunk
@@ -237,16 +234,17 @@ if uploaded_file is not None:
         clause_decisions = {}
         
         for i, clause in enumerate(clauses):
-            st.subheader(f"Clause {i+1}: {clause['title']}")
+            st.subheader(f"Clause {i + 1}: {clause['title']}")
             st.write(clause['content'])
             st.write(f"Risk Level: {clause['risk_level']}")
             st.write(f"Explanation: {clause['explanation']}")
             
-            decision = st.radio(f"Decision for Clause {i+1}", ["Accept", "Negotiate", "Reject"], key=f"decision_{i}")
+            decision = st.radio(f"Decision for Clause {i + 1}", ["Accept", "Negotiate", "Reject"], key=f"decision_{i}")
             clause_decisions[i] = decision
             
             if decision == "Negotiate":
-                negotiation_points = st.text_area(f"Enter negotiation points for Clause {i+1}", key=f"negotiation_{i}")
+                negotiation_points = st.text_area(f"Enter negotiation points for Clause {i + 1}", key=f"negotiation_{i}")
+                clause_decisions[f"{i}_points"] = negotiation_points  # Save negotiation points
 
         # Finalize Contract button
         if st.button("Finalize Contract"):
@@ -254,7 +252,7 @@ if uploaded_file is not None:
             
             for i, clause in enumerate(clauses):
                 decision = clause_decisions[i]
-                prompt += f"Clause {i+1} ({clause['title']}): {decision}\n"
+                prompt += f"Clause {i + 1} ({clause['title']}): {decision}\n"
                 if decision == "Negotiate":
                     prompt += f"Negotiation points: {clause_decisions.get(f'{i}_points', 'No specific points provided.')}\n"
                 prompt += "\n"
@@ -274,5 +272,3 @@ if uploaded_file is not None:
         st.write("No clauses found in the contract analysis. Please try again.")
 else:
     st.write("Please upload a contract to begin the analysis.")
-
-