Update Tags_Extractor.py

Tags_Extractor.py

@@ -7,14 +7,10 @@ class Tags:
         """
         Initialize the Extractor class.
         """
+        self.client = OpenAI()
 
-        # Set OpenAI API key
 
-        
-        # openai.api_key = ""
-
-
-    def extract_tags(extracted_summary):
+    def extract_tags(self,extracted_summary):
         """
         Extract tags from the refined summary using OpenAI API.
 
@@ -23,17 +19,20 @@ class Tags:
         """
         try:
             
-            # Use OpenAI's Completion API to analyze the text and extract tags
-            response = openai.Completion.create(
-                engine="text-davinci-003",
-                temperature=0,
-                prompt=f"analyze the given contract to extract tags for following contract in triple backticks. tags should be bullet points.contract :```{extracted_summary}```.",
-                max_tokens=1000
+            conversation = [
+                            {"role": "system", "content": "You are a helpful Tags Extracter."},
+                            {"role": "user", "content": f"""analyze the given contract to extract tags for following contract in triple backticks. tags should be bullet points.contract :```{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 and return the chatbot's reply
-            result = response['choices'][0]['text'].strip()
-            return result
+            response = chat_completion.choices[0].message.content
+            return response
 
         except Exception as e:
             print(f"Error occurred while extracting tags: {str(e)}")