Update headings_extractor.py

headings_extractor.py

@@ -16,7 +16,7 @@ class HeadingsExtractor:
         Returns:
             str: Extracted headings.
         """
-        pass
+        self.client=OpenAI()
         
     def file_output_fnn(self,file_path):
         file_path = file_path.name
@@ -35,15 +35,22 @@ class HeadingsExtractor:
         """
         try:
             #get response from openai api
-            response = openai.Completion.create(
-                engine="text-davinci-003",
-                prompt=f"""Extract Headings from given paragraph do not generate jsu extract the headings from paragraph.
-                        ```paragraph :{contract_page}```""",
-                max_tokens=100,
+            
+            conversation = [
+                            {"role": "system", "content": "You are a helpful assistant."},
+                            {"role": "user", "content": f"""Extract Headings from given paragraph do not generate jsu extract the headings from paragraph.
+                                    ```paragraph :{contract_page}```"""}
+                        ]
+            
+            # Call OpenAI GPT-3.5-turbo
+            chat_completion =self.client.chat.completions.create(
+                model = "gpt-3.5-turbo",
+                messages = conversation,
+                max_tokens=1000,
                 temperature=0
             )
-            headings = response.choices[0].text.strip()
-            return headings
+            response = chat_completion.choices[0].message.content
+            return response 
 
         except Exception as e:
             # If an error occurs during the key-value extraction process, log the error