Api

app.py

@@ -11,168 +11,167 @@ load_dotenv()
 
 # Azure OpenAI credentials
 key = os.getenv("AZURE_OPENAI_API_KEY")
-print(key)
-# endpoint_url = "https://interview-key.openai.azure.com/"
-# api_version = "2024-05-01-preview"
-# deployment_id = "interview"
-
-# # Initialize Azure OpenAI client
-# client = AzureOpenAI(
-#     api_version=api_version,
-#     azure_endpoint=endpoint_url,
-#     api_key=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])
-
-# def extract_terms_from_contract(contract_text):
-#     prompt = (
-#         "You are an AI tasked with analyzing a contract and extracting key terms and constraints. The contract contains "
-#         "various sections and subsections with terms related to budget constraints, types of allowable work, timelines, "
-#         "penalties, responsibilities, and other conditions for work execution. Your job is to extract these key terms and "
-#         "structure them in a clear JSON format, reflecting the hierarchy of sections and subsections. "
-#         "Ensure to capture all important constraints and conditions specified in the contract text. If a section or subsection "
-#         "contains multiple terms, list them all.\n\n"
-#         "Contract text:\n"
-#         f"{contract_text}\n\n"
-#         "Provide the extracted terms in JSON format."
-#     )
-
-#     try:
-#         response = client.chat.completions.create(
-#             model=deployment_id,
-#             messages=[
-#                 {"role": "system", "content": "You are an AI specialized in extracting structured data from text documents."},
-#                 {"role": "user", "content": prompt},
-#             ],
-#             max_tokens=1250,
-#             n=1,
-#             stop=None,
-#             temperature=0.1,
-#         )
-#         return response.choices[0].message.content
-#     except Exception as e:
-#         st.error(f"Error extracting terms from contract: {e}")
-#         return None
-
-# def analyze_task_compliance(task_description, cost_estimate, contract_terms):
-#     print("Task D: ", task_description, cost_estimate)
-#     prompt = (
-#         "You are an AI tasked with analyzing a task description and its associated cost estimate for compliance with contract conditions. "
-#         "Below are the key terms and constraints extracted from the contract, followed by a task description and its cost estimate. "
-#         "Your job is to analyze each task description and specify if it violates any conditions from the contract. "
-#         "If there are violations, list the reasons for each violation. Provide detailed answers and do not give only true or false answers.\n\n"
-#         f"Contract terms:\n{json.dumps(contract_terms, indent=4)}\n\n"
-#         f"Task description:\n{task_description}\n"
-#         f"Cost estimate:\n{cost_estimate}\n\n"
-#         "Provide the compliance analysis in a clear JSON format."
-#     )
-
-#     try:
-#         response = client.chat.completions.create(
-#             model=deployment_id,
-#             messages=[
-#                 {"role": "system", "content": "You are an AI specialized in analyzing text for compliance with specified conditions."},
-#                 {"role": "user", "content": prompt},
-#             ],
-#             max_tokens=1250,
-#             n=1,
-#             stop=None,
-#             temperature=0.1,
-#         )
-#         return json.loads(response.choices[0].message.content)
-#     except Exception as e:
-#         st.error(f"Error analyzing task compliance: {e}")
-#         return None
-
-# def main():
-#     st.markdown("<h1 class='centered-title'>Contract Compliance Analyzer</h1>", unsafe_allow_html=True)
-
-#     # Initialize session state
-#     if 'contract_terms' not in st.session_state:
-#         st.session_state.contract_terms = None
-#     if 'compliance_results' not in st.session_state:
-#         st.session_state.compliance_results = None
-
-#     # File upload buttons one after another
-#     docx_file = st.sidebar.file_uploader("Upload Contract Document (DOCX)", type="docx", key="docx_file")
-#     data_file = st.sidebar.file_uploader("Upload Task Descriptions (XLSX or CSV)", type=["xlsx", "csv"], key="data_file")
-#     submit_button = st.sidebar.button("Submit")
-
-#     if submit_button and docx_file and data_file:
-#         # 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
+endpoint_url = "https://interview-key.openai.azure.com/"
+api_version = "2024-05-01-preview"
+deployment_id = "interview"
+
+# Initialize Azure OpenAI client
+client = AzureOpenAI(
+    api_version=api_version,
+    azure_endpoint=endpoint_url,
+    api_key=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])
+
+def extract_terms_from_contract(contract_text):
+    prompt = (
+        "You are an AI tasked with analyzing a contract and extracting key terms and constraints. The contract contains "
+        "various sections and subsections with terms related to budget constraints, types of allowable work, timelines, "
+        "penalties, responsibilities, and other conditions for work execution. Your job is to extract these key terms and "
+        "structure them in a clear JSON format, reflecting the hierarchy of sections and subsections. "
+        "Ensure to capture all important constraints and conditions specified in the contract text. If a section or subsection "
+        "contains multiple terms, list them all.\n\n"
+        "Contract text:\n"
+        f"{contract_text}\n\n"
+        "Provide the extracted terms in JSON format."
+    )
+
+    try:
+        response = client.chat.completions.create(
+            model=deployment_id,
+            messages=[
+                {"role": "system", "content": "You are an AI specialized in extracting structured data from text documents."},
+                {"role": "user", "content": prompt},
+            ],
+            max_tokens=1250,
+            n=1,
+            stop=None,
+            temperature=0.1,
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        st.error(f"Error extracting terms from contract: {e}")
+        return None
+
+def analyze_task_compliance(task_description, cost_estimate, contract_terms):
+    print("Task D: ", task_description, cost_estimate)
+    prompt = (
+        "You are an AI tasked with analyzing a task description and its associated cost estimate for compliance with contract conditions. "
+        "Below are the key terms and constraints extracted from the contract, followed by a task description and its cost estimate. "
+        "Your job is to analyze each task description and specify if it violates any conditions from the contract. "
+        "If there are violations, list the reasons for each violation. Provide detailed answers and do not give only true or false answers.\n\n"
+        f"Contract terms:\n{json.dumps(contract_terms, indent=4)}\n\n"
+        f"Task description:\n{task_description}\n"
+        f"Cost estimate:\n{cost_estimate}\n\n"
+        "Provide the compliance analysis in a clear JSON format."
+    )
+
+    try:
+        response = client.chat.completions.create(
+            model=deployment_id,
+            messages=[
+                {"role": "system", "content": "You are an AI specialized in analyzing text for compliance with specified conditions."},
+                {"role": "user", "content": prompt},
+            ],
+            max_tokens=1250,
+            n=1,
+            stop=None,
+            temperature=0.1,
+        )
+        return json.loads(response.choices[0].message.content)
+    except Exception as e:
+        st.error(f"Error analyzing task compliance: {e}")
+        return None
+
+def main():
+    st.markdown("<h1 class='centered-title'>Contract Compliance Analyzer</h1>", unsafe_allow_html=True)
+
+    # Initialize session state
+    if 'contract_terms' not in st.session_state:
+        st.session_state.contract_terms = None
+    if 'compliance_results' not in st.session_state:
+        st.session_state.compliance_results = None
+
+    # File upload buttons one after another
+    docx_file = st.sidebar.file_uploader("Upload Contract Document (DOCX)", type="docx", key="docx_file")
+    data_file = st.sidebar.file_uploader("Upload Task Descriptions (XLSX or CSV)", type=["xlsx", "csv"], key="data_file")
+    submit_button = st.sidebar.button("Submit")
+
+    if submit_button and docx_file and data_file:
+        # 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:
-#             st.session_state.contract_terms = json.loads(extracted_terms_json)
-#         except json.JSONDecodeError as e:
-#             st.error(f"JSON decoding error: {e}")
-#             return
+        try:
+            st.session_state.contract_terms = json.loads(extracted_terms_json)
+        except json.JSONDecodeError as e:
+            st.error(f"JSON decoding error: {e}")
+            return
         
-#         # 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 = []
-
-#         # Process tasks sequentially
-#         for _, row in tasks_df.iterrows():
-#             result = analyze_task_compliance(row['Task Description'], row['Amount'], st.session_state.contract_terms)
-#             if result is not None:
-#                 print(result)
-#                 compliance_results.append(result)
+        # 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 = []
+
+        # Process tasks sequentially
+        for _, row in tasks_df.iterrows():
+            result = analyze_task_compliance(row['Task Description'], row['Amount'], st.session_state.contract_terms)
+            if result is not None:
+                print(result)
+                compliance_results.append(result)
         
-#         st.session_state.compliance_results = compliance_results
+        st.session_state.compliance_results = compliance_results
         
-#     col1, col2 = st.columns(2)
+    col1, col2 = st.columns(2)
 
-#     with col1:
-#         if st.session_state.contract_terms:
-#             st.write("Extracted Contract Terms:")
-#             st.json(st.session_state.contract_terms)
+    with col1:
+        if st.session_state.contract_terms:
+            st.write("Extracted Contract Terms:")
+            st.json(st.session_state.contract_terms)
             
-#             # Download button for contract terms
-#             st.download_button(
-#                 label="Download Contract Terms",
-#                 data=json.dumps(st.session_state.contract_terms, indent=4),
-#                 file_name="contract_terms.json",
-#                 mime="application/json"
-#             )
-
-#     with col2:
-#         if st.session_state.compliance_results:
-#             st.write("Compliance Results:")
-#             st.json(st.session_state.compliance_results)
-
-#             # Download button for compliance results
-#             st.download_button(
-#                 label="Download Compliance Results",
-#                 data=json.dumps(st.session_state.compliance_results, indent=4),
-#                 file_name="compliance_results.json",
-#                 mime="application/json"
-#             )
-
-# if __name__ == "__main__":
-#     main()
+            # Download button for contract terms
+            st.download_button(
+                label="Download Contract Terms",
+                data=json.dumps(st.session_state.contract_terms, indent=4),
+                file_name="contract_terms.json",
+                mime="application/json"
+            )
+
+    with col2:
+        if st.session_state.compliance_results:
+            st.write("Compliance Results:")
+            st.json(st.session_state.compliance_results)
+
+            # Download button for compliance results
+            st.download_button(
+                label="Download Compliance Results",
+                data=json.dumps(st.session_state.compliance_results, indent=4),
+                file_name="compliance_results.json",
+                mime="application/json"
+            )
+
+if __name__ == "__main__":
+    main()