Update incorrect_sentence_finder.py

incorrect_sentence_finder.py

@@ -1,4 +1,4 @@
-import fitz  # PyMuPDF
+from PyPDF2 import PdfReader
 import openai
 import gradio as gr
 
@@ -15,7 +15,7 @@ class IncorrectSentenceFinder:
         Initialize the IncorrectSentenceFinder with the OpenAI API key.
         """
         # openai.api_key = openai_api_key
-        pass
+        self.client=OpenAI()
 
     def _find_incorrect_sentence(self, text: str) -> str:
         """
@@ -28,16 +28,23 @@ class IncorrectSentenceFinder:
             str: Grammatically incorrect sentences identified by GPT-3.
         """
         # Create a request to OpenAI's GPT-3 engine to identify grammatically incorrect sentences.
-        response = openai.Completion.create(
-            engine="text-davinci-003",
-            prompt=f"list out the grammatical error sentence in the given text:\n{text}",
-            temperature=0,
-            max_tokens=1000,
+                conversation = [
+                        {"role": "system", "content": "You are a helpful Error sentence finder."},
+                        {"role": "user", "content": f"""list out the grammatical error sentence in the given text:\n{text}
+                         format:
+                          error sentence:finded error sentence
+                          """}
+                    ]
+                    
+                    # Call OpenAI GPT-3.5-turbo
+        chat_completion =client.chat.completions.create(
+            model = "gpt-3.5-turbo",
+            messages = conversation,
+            max_tokens=500,
+            temperature=0
         )
-
-        # Extract and strip the text of identified grammatically incorrect sentences from the GPT-3 response.
-        incorrect_sentences = response.choices[0].text.strip()
-        return incorrect_sentences
+        response = chat_completion.choices[0].message.content
+        return response 
 
     def get_incorrect_sentence(self, pdf_file: str) -> str:
         """
@@ -48,12 +55,14 @@ class IncorrectSentenceFinder:
         """
         try:
             # Open the PDF file using PyMuPDF's fitz library
-            doc = fitz.open(pdf_file.name)
+            doc =PdfReader(pdf_file.name)
             incorrect_sentences = ''
             # Iterate through each page in the PDF document and extract the text
-            for page in doc:
-                text = page.get_text()
-                incorrect_sentences += self._find_incorrect_sentence(text)
+            for page_number in range(len(doc.pages)):
+                            # Extract text from the page
+                            page = doc.pages[page_number]
+                            text = page.extract_text()
+                            incorrect_sentences += self._find_incorrect_sentence(text)
             return incorrect_sentences
 
         except Exception as e: