Update app.py

app.py

@@ -1044,8 +1044,13 @@ class ChatBotUI(object):
             img_str = f'<img src="data:image/png;base64,{img_b64}" style="pointer-events: none;">'
             history = [(prompt,
                         f'{pre_info} The generated image is:\n {img_str}')]
+
+            img_id = get_md5(img_b64)[:12]
+            save_path = os.path.join(self.cache_dir, f'{img_id}.png')
+            img.convert('RGB').save(save_path)
+            
             return self.get_history(history), gr.update(value=''), gr.update(
-                visible=False), gr.Image(value=img), gr.update(value=-1)
+                visible=False), gr.Image(value=save_path), gr.update(value=-1)
 
         with self.eg:
             self.example_task = gr.Text(label='Task Name',