Update app.py

app.py

@@ -69,17 +69,20 @@ def on_change_event(app_state):
         step = app_state['step']
         label = f'Reconstructed image from the latent state at step {step}. It will get better :)'
         print(f'Updating the image:! {app_state}')
-        return gr.update(value=app_state['img_list'], label='intermediate steps')
+        app_state['dots'] += 1
+        app_state['dots'] = app_state['dots'] % 10
+        return gr.update(value=app_state['img_list'], label='intermediate steps'), app_state['staus_msg'] + '.' * app_state['dots']
     else:
-        return gr.update(label='images list')
+        return gr.update(label='images list'), ''
 
 with gr.Blocks() as demo:
     
     def generate_image(prompt, inference_steps, app_state):
         app_state['running'] = True
         app_state['img_list'] = []
+        app_state['status_msg'] = 'Starting'
         def callback(step, ts, latents):
-            print (f'In Callback on {step} {ts} !')
+            app_state['status_msg'] = f'Reconstructing an image from the latent state on step {step}'
             latents = 1 / 0.18215 * latents
             res = pipe.vae.decode(latents).sample
             res = (res / 2 + 0.5).clamp(0, 1)
@@ -88,18 +91,22 @@ with gr.Blocks() as demo:
             app_state['img'] = res
             app_state['step'] = step
             app_state['img_list'].append(res)
-            print (f'In Callback on {app_state} Done!')
+            app_state['status_msg'] = f'Generating state {step}'
+            app_state['dots'] = 0
         
         prompt = prompt + ' masterpiece charcoal pencil art lord of the rings illustration'
-        img = pipe(prompt, height=512, width=512, num_inference_steps=inference_steps, callback=callback, callback_steps=2)
+        img = pipe(prompt, height=512, width=512, num_inference_steps=inference_steps, callback=callback, callback_steps=1)
         app_state['running'] = False
         app_state['img'] = None
+        app_state['status_msg'] = ''
         return gr.update(value=img.images[0], label='Generated image')
     
     app_state = gr.State({'img': None, 
                           'step':0, 
                           'running':False,
-                           'img_list': []
+                          'status_msg': '',
+                          'img_list': [],
+                          'dots': 0
                          })
     title = gr.Markdown('## Lord of the rings app')
     description = gr.Markdown(f'#### A Lord of the rings inspired app that combines text and image generation.'
@@ -115,6 +122,8 @@ with gr.Blocks() as demo:
     img_description = gr.Markdown('Image generation takes some time'
                                   ' but here you can see what is generated from the latent state of the diffuser every few steps.'
                                   ' Usually there is a significant improvement around step 12 that yields a much better image')
+    status_msg = gr.Markdown(label='')
+    
     gallery = gr.Gallery()
     image = gr.Image(label='Illustration for your story', show_label=True)
 
@@ -131,8 +140,8 @@ with gr.Blocks() as demo:
     bt_make_image.click(fn=generate_image, inputs=[summary, inference_steps, app_state], outputs=image)
     
     eventslider = gr.Slider(visible=False)
-    dep = demo.load(on_change_event, app_state, [gallery], every=10)
-    eventslider.change(fn=on_change_event, inputs=[app_state], outputs=[gallery], every=10, cancels=[dep])
+    dep = demo.load(on_change_event, app_state, [gallery, status_msg], every=10)
+    eventslider.change(fn=on_change_event, inputs=[app_state], outputs=[gallery, status_msg], every=10, cancels=[dep])
     
 
 if READ_TOKEN: