Update backend.py

backend.py

@@ -52,10 +52,20 @@ class InvoicePipeline:
     def _get_raw_text_from_pdf(self, path:str) -> str:
         text = ""
         pdf_reader = PdfReader(path)
-        for page in pdf_reader:
+        for page in pdf_reader.pages:
             text += page.extract_text()
         return text
 
     def _extract_data_from_llm(self, raw_data:str) -> str:
         resp = self._llm(self._prompt_template.format(pages = raw_data))
-        return resp
+        return resp
+        
+    def _parse_response(self, response: str) -> Dict[str, str]:
+        pattern = r'{(.+)}'
+        re_match = re.search(pattern, response, re.DOTALL)
+        if re_match:
+            extracted_text = re_match.group(1)
+            data = eval('{' + extracted_text + '}')
+            return data
+        else:
+            raise Exception("No match found.")