simonlee-cb commited on
Commit
abc4f31
·
1 Parent(s): f1dd29d

feat: replace current image with edited image

Browse files
Files changed (1) hide show
  1. gradio_chat.py +1 -0
gradio_chat.py CHANGED
@@ -144,6 +144,7 @@ async def stream_from_agent(chat_input, chatbot, past_messages, current_image):
144
  "content": gr.Image(call.content.edited_image_url),
145
  "files": [call.content.edited_image_url]
146
  })
 
147
  else:
148
  gr_message['content'] += (
149
  f'\nOutput: {call.content}'
 
144
  "content": gr.Image(call.content.edited_image_url),
145
  "files": [call.content.edited_image_url]
146
  })
147
+ current_image = call.content.edited_image_url
148
  else:
149
  gr_message['content'] += (
150
  f'\nOutput: {call.content}'