Update headings_extractor.py

headings_extractor.py

@@ -1,6 +1,5 @@
-from openai import OpenAI
+import openai
 from PyPDF2 import PdfReader
-# import fitz
 import os
 import gradio as gr
 
@@ -16,7 +15,10 @@ class HeadingsExtractor:
         Returns:
             str: Extracted headings.
         """
-        self.client=OpenAI()
+        openai.api_type = os.getenv['api_type']
+        openai.api_base = os.getenv['api_base']
+        openai.api_version = os.getenv['api_version']
+        openai.api_key = os.getenv['api_key']
         
     def file_output_fnn(self,file_path):
         file_path = file_path.name