Fix package issues

app.py

@@ -1,5 +1,6 @@
 import json
 import os
+from pathlib import Path
 
 import google.generativeai as genai
 import gradio as gr
@@ -7,27 +8,25 @@ import pandas as pd
 from gradio_pdf import PDF
 from pdf2image import convert_from_path
 from pypdf import PdfReader
-from pathlib import Path
-dir_ = Path(__file__).parent
 
 genai.configure(api_key=os.environ.get("GOOGLE_API_KEY"))
-headers=[
-            "DUE DATE",
-            "SERVICE ADDRESS",
-            "SERVICE PERIOD",
-            "ELECTRICITY USAGE (KWH)",
-            "ELECTRICITY SPEND ($)",
-            "GAS USAGE (THERMS)",
-            "GAS SPEND ($)",
-            "WATER USAGE (CCF)",
-            "WATER SPEND ($)",
-            "SEWER ($)",
-            "REFUSE ($)",
-            "STORM DRAIN ($)",
-            "UTILITY USERS TAX ($)",
-            "TOTAL CURRENT CHARGES ($)",
-            "TOTAL AMOUNT DUE",
-        ]
+headers = [
+    "DUE DATE",
+    "SERVICE ADDRESS",
+    "SERVICE PERIOD",
+    "ELECTRICITY USAGE (KWH)",
+    "ELECTRICITY SPEND ($)",
+    "GAS USAGE (THERMS)",
+    "GAS SPEND ($)",
+    "WATER USAGE (CCF)",
+    "WATER SPEND ($)",
+    "SEWER ($)",
+    "REFUSE ($)",
+    "STORM DRAIN ($)",
+    "UTILITY USERS TAX ($)",
+    "TOTAL CURRENT CHARGES ($)",
+    "TOTAL AMOUNT DUE",
+]
 
 
 inputs = [PDF(label="Document")]
@@ -78,7 +77,6 @@ def get_content_between_curly_braces(text):
         return None
 
 
-
 def parse_utility_bill(filepath):
     print("FOUND PDF!")
     reader = PdfReader(filepath)
@@ -116,7 +114,7 @@ def parse_utility_bill(filepath):
     print(respone_dict)
     rectified_dict = {}
     for target_key in headers:
-        
+
         for key, value in respone_dict.items():
             if key == target_key:
                 rectified_dict[key] = value
@@ -125,15 +123,21 @@ def parse_utility_bill(filepath):
             rectified_dict[target_key] = None
     print(rectified_dict)
     example_data = [rectified_dict]
-    
 
     return pd.DataFrame(example_data)
 
+
 gr.Interface(
     fn=parse_utility_bill,
     inputs=inputs,
     outputs=outputs,
-    examples=["utl-bill-sample.pdf", "nem-2-utility-bill-sample.pdf", "Sample_Utility_Bill.pdf", "Water Bill Sample.pdf", "canada.pdf", "water.pdf"],
+    examples=[
+        "utl-bill-sample.pdf",
+        "nem-2-utility-bill-sample.pdf",
+        "Sample_Utility_Bill.pdf",
+        "Water Bill Sample.pdf",
+        "canada.pdf",
+        "water.pdf",
+    ],
     title="🌏⚡💧🔥PDF Utitlity Bill Parser",
 ).launch()
-

requirements.txt

@@ -1,4 +1,5 @@
 pypdf
 pytesseract
 gradio_pdf
-google-generativeai
+google-generativeai
+pdf2image