anujithc commited on
Commit
9b55e5c
·
1 Parent(s): 0911836
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -230,20 +230,23 @@ def run_lora(
230
  print("fatal error", e)
231
  image = None
232
 
233
- result = {"status": "failed", "message": error_message} if image is None else {"status": "success", "message": "Image generated but not uploaded"}
 
234
  if image is not None and upload_to_r2:
235
  try:
236
  url = upload_image_to_r2(image, account_id, access_key, secret_key, bucket)
237
- result = {"status": "success", "message": "upload image success", "url": url}
238
  except Exception as e:
239
  err = f"Upload failed: {e}"
240
  gr.Warning(err)
241
  print(err)
242
- result = {"status": "success", "message": "generated but upload failed"}
243
 
244
  gr.Info("Completed!")
245
  progress(100, "Completed!")
246
- return json.dumps(result)
 
 
247
 
248
  # Gradio interface
249
 
 
230
  print("fatal error", e)
231
  image = None
232
 
233
+ result = {"status": "failed", "message": error_message} if image is None else {"status": "success", "message": "Image generated but not uploaded", "seed": seed}
234
+
235
  if image is not None and upload_to_r2:
236
  try:
237
  url = upload_image_to_r2(image, account_id, access_key, secret_key, bucket)
238
+ result = {"status": "success", "message": "upload image success", "url": url, "seed": seed}
239
  except Exception as e:
240
  err = f"Upload failed: {e}"
241
  gr.Warning(err)
242
  print(err)
243
+ result = {"status": "success", "message": "generated but upload failed", "seed": seed}
244
 
245
  gr.Info("Completed!")
246
  progress(100, "Completed!")
247
+
248
+ # CHANGED: Return both image AND json
249
+ return image, json.dumps(result) # <--- THIS IS THE KEY CHANGE
250
 
251
  # Gradio interface
252