Update backend.py

backend.py

@@ -6,7 +6,8 @@ from typing import List, Dict
 from langchain.prompts import PromptTemplate
 from langchain_google_genai import GoogleGenerativeAI
 
-os.environ["GOOGLE_API_KEY"] = "AIzaSyCYGj5e2eAQbUi9HtuMaW0LDSnDuxLG54U"
+
+api_key = "AIzaSyCYGj5e2eAQbUi9HtuMaW0LDSnDuxLG54U"
 
 
 class InvoicePipeline:
@@ -18,14 +19,14 @@ class InvoicePipeline:
     # This funcition will help in extracting and run the code, and will produce a dataframe for us
     def run(self) -> pd.DataFrame:
         # We have defined the way the data has to be returned
-        df = pd.DataFrame(
+        df = pd.DataFrame({
             "Invoice ID": pd.Series(dtype = "int"),
             "DESCRIPTION": pd.Series(dtype = "str"),
             "Issue Data": pd.Series(dtype = "str"),
             "UNIT PRICE": pd.Series(dtype = "str"),
             "AMOUNT": pd.Series(dtype = "int"),
             "Bill For": pd.Series(dtype = "str"),
-            "From": pd.Series(dtype =" str"),
+            "From": pd.Series(dtype ="str"),
             "Terms": pd.Series(dtype = "str")}
         )
 
@@ -57,4 +58,6 @@ class InvoicePipeline:
 
     def _extract_data_from_llm(self, raw_data:str) -> str:
         resp = self._llm(self._prompt_template.format(pages = raw_data))
-        return resp
+        return resp
+
+