| import streamlit as st | |
| import os | |
| import json | |
| import pandas as pd | |
| from docx import Document | |
| from dotenv import load_dotenv | |
| from openai import AzureOpenAI | |
| # Load environment variables | |
| 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 | |
| # 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) | |
| # st.session_state.compliance_results = compliance_results | |
| # 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) | |
| # # 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() | |