Update ContractGenerator.py

ContractGenerator.py

@@ -12,7 +12,7 @@ class ContractGenerator:
         Args:
             api_key (str): Your OpenAI API key.
         """
-        pass
+        self.client = OpenAI()
 
     def generate_contract(self, instructions: str) -> None:
         """
@@ -25,19 +25,21 @@ class ContractGenerator:
             openai.error.OpenAIError: If there is an error with the OpenAI API request.
         """
         # Define a prompt
-        prompt = f"Your task is to generate a contract form based on user instructions. ***Instructions:{instructions}***"
-
-        try:
-            # Generate text using the GPT-3.5 model
-            response = openai.Completion.create(
-                engine="text-davinci-003",
-                prompt=prompt,
-                max_tokens=500  # You can adjust the length of the generated text
-            )
-
-            # Print the generated text
-            return response.choices[0].text
-
-        except openai.error.OpenAIError as e:
-            print(f"Error generating the contract: {str(e)}")
+        client = OpenAI()
+        os.environ["OPENAI_API_KEY"]="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"
+
+        conversation = [
+                        {"role": "system", "content": "You are a helpful Contract Generator."},
+                        {"role": "user", "content": f"""Your task is to generate a contract form based on user instructions. ***Instructions:{instructions}***"""}
+                      ]
+                    
+        # Call OpenAI GPT-3.5-turbo
+        chat_completion = client.chat.completions.create(
+            model = "gpt-3.5-turbo",
+            messages = conversation,
+            max_tokens=500,
+            temperature=0
+        )
+        response = chat_completion.choices[0].message.content
+        return response