Update backend.py

backend.py

@@ -60,7 +60,16 @@ class InvoicePipeline:
     def _extract_data_from_llm(self, raw_data:str) -> str:
         resp = self._llm(self._prompt_template.format(pages = raw_data))
         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.")