SetuG commited on
Commit
34820ab
·
verified ·
1 Parent(s): 6c04feb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -152,7 +152,6 @@ class InvoiceDuplicateDetector:
152
  results.sort(key=lambda x: x[1], reverse=True)
153
  return True, results
154
 
155
-
156
  detector = InvoiceDuplicateDetector()
157
 
158
  def upload_files(files):
@@ -161,8 +160,9 @@ def upload_files(files):
161
  results = []
162
  for file in files:
163
  try:
164
- file_bytes = file if isinstance(file, bytes) else file.value
165
- filename = getattr(file, "name", "uploaded")
 
166
  success, message = detector.store_invoice(file_bytes, filename)
167
  results.append(f"{filename}: {message}")
168
  except Exception as e:
@@ -171,8 +171,9 @@ def upload_files(files):
171
 
172
  def check_duplicates(file):
173
  try:
174
- file_bytes = file if isinstance(file, bytes) else file.value
175
- filename = getattr(file, "name", "uploaded")
 
176
  ok, result = detector.find_duplicates(file_bytes, filename)
177
  if not ok:
178
  return result
@@ -203,3 +204,4 @@ demo.launch()
203
 
204
 
205
 
 
 
152
  results.sort(key=lambda x: x[1], reverse=True)
153
  return True, results
154
 
 
155
  detector = InvoiceDuplicateDetector()
156
 
157
  def upload_files(files):
 
160
  results = []
161
  for file in files:
162
  try:
163
+ with open(file.name, "rb") as f:
164
+ file_bytes = f.read()
165
+ filename = os.path.basename(file.name)
166
  success, message = detector.store_invoice(file_bytes, filename)
167
  results.append(f"{filename}: {message}")
168
  except Exception as e:
 
171
 
172
  def check_duplicates(file):
173
  try:
174
+ with open(file.name, "rb") as f:
175
+ file_bytes = f.read()
176
+ filename = os.path.basename(file.name)
177
  ok, result = detector.find_duplicates(file_bytes, filename)
178
  if not ok:
179
  return result
 
204
 
205
 
206
 
207
+