Prashasst's picture
Update utils.py
d258e41 verified
raw
history blame
649 Bytes
import pandas as pd
import time
def measure_time(func): #already decorated the read_file()
"""Decorator for measuring time taken during the OCR"""
def wrapper(*args, **kwargs):
start = time.time()
result = func(*args, **kwargs)
end = time.time()
print(f"Execution time: {end - start:.6f} seconds")
return result
return wrapper
def safe_dataframe(data, key):
"""Safely convert extracted data to a pandas DataFrame"""
try:
return pd.DataFrame(data[key])
except Exception as e:
print(f"Error creating {key} DataFrame: {e}")
return pd.DataFrame()