wjm55 commited on
Commit
cf99179
·
1 Parent(s): 266e1d2

trying to fix download

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -301,16 +301,18 @@ with gr.Blocks(css=css) as demo:
301
  with open(json_path, 'w', encoding='utf-8') as f:
302
  json.dump(json_data, f, indent=2, ensure_ascii=False)
303
 
304
- # Create zip file
305
- zip_path = os.path.join(temp_dir, f"{fname}.zip")
 
 
 
306
  with zipfile.ZipFile(zip_path, 'w') as zipf:
307
  zipf.write(img_path, os.path.basename(img_path))
308
  zipf.write(txt_path, os.path.basename(txt_path))
309
  zipf.write(json_path, os.path.basename(json_path))
310
 
311
- # Read the zip file into memory
312
- with open(zip_path, 'rb') as f:
313
- return (f"{fname}.zip", f.read())
314
  except Exception as e:
315
  print(f"Error creating zip: {str(e)}")
316
  return None
 
301
  with open(json_path, 'w', encoding='utf-8') as f:
302
  json.dump(json_data, f, indent=2, ensure_ascii=False)
303
 
304
+ # Create zip file in a location that will persist
305
+ output_dir = "downloads"
306
+ os.makedirs(output_dir, exist_ok=True)
307
+ zip_path = os.path.join(output_dir, f"{fname}.zip")
308
+
309
  with zipfile.ZipFile(zip_path, 'w') as zipf:
310
  zipf.write(img_path, os.path.basename(img_path))
311
  zipf.write(txt_path, os.path.basename(txt_path))
312
  zipf.write(json_path, os.path.basename(json_path))
313
 
314
+ return zip_path # Return just the path to the zip file
315
+
 
316
  except Exception as e:
317
  print(f"Error creating zip: {str(e)}")
318
  return None