I use this “Edit” feature most often when accidentally submitting or hastily submitting my entries. Love that you can Edit prompts and press “Save & Submit” to regenerate results but the “Upload Image” option is not available when editing.
I have 2 related suggestions:
(1) Similar to Gmail, detect when someone has typed “attached” or “attachment” yet no image/file has been uploaded and suggest that they upload one if that was intended.
(2) Add the “upload” feature to the submission box when Editing (not just when submitting for the first time).