NiamaLynn commited on
Commit
545165f
·
1 Parent(s): 2244eb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -89,12 +89,13 @@ def app_outputs(uploaded_pdf):
89
  images = images[:max_imgboxes]
90
  df = dict(itertools.islice(df.items(), max_imgboxes))
91
 
92
- # save
93
  csv_files = list()
94
  for i in range(max_imgboxes):
95
  csv_file = f"csv_{i}_" + filename.replace(".pdf", ".csv")
96
- csv_files.append(gr.File.update(value=csv_file, visible=True))
97
- df[i].to_csv(csv_file, encoding="utf-8", index=False)
 
 
98
 
99
  else:
100
  img_files, images, csv_files = [""]*max_imgboxes, [""]*max_imgboxes, [""]*max_imgboxes
 
89
  images = images[:max_imgboxes]
90
  df = dict(itertools.islice(df.items(), max_imgboxes))
91
 
 
92
  csv_files = list()
93
  for i in range(max_imgboxes):
94
  csv_file = f"csv_{i}_" + filename.replace(".pdf", ".csv")
95
+ if not os.path.exists(csv_file):
96
+ os.makedirs(os.path.dirname(csv_file))
97
+ csv_files.append(gr.File(value=csv_file, visible=True)) # Create new File instances
98
+ df[i].to_csv(csv_file, encoding="utf-8", index=False)
99
 
100
  else:
101
  img_files, images, csv_files = [""]*max_imgboxes, [""]*max_imgboxes, [""]*max_imgboxes