Update Clauses_Extractor.py

Clauses_Extractor.py

@@ -1,18 +1,14 @@
 import openai
 import os
-
+from openai import OpenAI
 
 # Define the Clauses class
 class Clauses:
     def __init__(self):
-        """
-        Initialize the Extractor class.
-        """
 
-        # Set OpenAI API key
-        # os.environ["OPENAI_API_KEY"] = ""
+           self.client = OpenAI()
 
-    def get_extracted_clauses(extracted_summary):
+    def get_extracted_clauses(self,extracted_summary):
         """
         Gets extracted clauses using GPT-3 based on the provided PDF.
 
@@ -23,24 +19,21 @@ class Clauses:
             str: Extracted clauses from GPT-3 response.
         """
         try:
-            # Prepare a prompt for GPT-3 that includes the extracted PDF text and instructions
-            prompt = f"""
-            Extract clauses and sub-clauses from the provided contract PDF:
-
-            {extracted_summary}
-
-            Instructions: Organize the extracted clauses and sub clauses in a readable format.
-            """
-            # Use GPT-3 to process the prompt and generate clauses
-            response = openai.Completion.create(
-                engine="text-davinci-003",
-                prompt=prompt,
-                max_tokens=1000
+            conversation = [
+                            {"role": "system", "content": "You are a helpful Cluases and SubCluases Extracter From Given Content."},
+                            {"role": "user", "content": f"""Extract clauses and sub-clauses from the provided contract PDF
+                            {extracted_summary}"""}
+                          ]
+                        
+            # Call OpenAI GPT-3.5-turbo
+            chat_completion = self.client.chat.completions.create(
+                model = "gpt-3.5-turbo",
+                messages = conversation,
+                max_tokens=500,
+                temperature=0
             )
-
-            # Extract the generated text from the GPT-3 response
-            result = response['choices'][0]['text'].strip()
-            return result
+            response = chat_completion.choices[0].message.content
+            return response
 
         except Exception as e:
             # If an error occurs during GPT-3 processing, log the error and raise an exception