CoolFace
Apppublic

productions/Data_Conversion

sourceHugging Faceupdated 2y agoView on Hugging Face
0likes
Clauses_Extractor.py50 linesDownload Raw Back to root
1from openai import AzureOpenAI2import os3 4# Define the Clauses class5class Clauses:6    7    def __init__(self):8            9           # openai.api_type = os.getenv['api_type']10           # openai.api_base = os.getenv['api_base']11           # openai.api_version = os.getenv['api_version']12           # openai.api_key = os.getenv['api_key']13           pass14 15    def get_extracted_clauses(extracted_summary):16        """17        Gets extracted clauses using GPT-3 based on the provided PDF.18 19        Args:20            max_tokens (int, optional): Maximum number of tokens for GPT-3 response.21 22        Returns:23            str: Extracted clauses from GPT-3 response.24        """25        try:26            client = AzureOpenAI(api_key=os.getenv("AZURE_OPENAI_KEY"),  27                                api_version="2023-07-01-preview",28                                azure_endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")29                                 )30            conversation = [31                            {"role": "system", "content": "You are a helpful Cluases and SubCluases Extracter From Given Content."},32                            {"role": "user", "content": f"""Extract clauses and sub-clauses from the provided contract PDF33                            {extracted_summary}"""}34                          ]35                        36            # Call OpenAI GPT-3.5-turbo37            chat_completion = client.chat.completions.create(38                model = "GPT-3",39                messages = conversation,40                max_tokens=1000,41                temperature=042            )43            response = chat_completion.choices[0].message.content44            return response45 46        except Exception as e:47            # If an error occurs during GPT-3 processing, log the error and raise an exception48            print(f"Error occurred while processing PDF with GPT-3. Error message: {str(e)}")49            raise50