Update app.py

app.py

@@ -64,6 +64,7 @@ def generate_story(prompt):
     return story, summary, gr.update(visible=True)
 
 def on_change_event(app_state):
+    print(f'on_change_event {app_state}')
     if app_state and app_state['running'] and app_state['img']:
         img = app_state['img']
         step = app_state['step']
@@ -71,9 +72,11 @@ def on_change_event(app_state):
         print(f'Updating the image:! {app_state}')
         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']
+        message = app_state['status_msg'] + '.' * app_state['dots']
+        print (f'message={message}')
+        return gr.update(value=app_state['img_list'], label='intermediate steps'), gr.update(value=message)
     else:
-        return gr.update(label='images list'), ''
+        return gr.update(label='images list'), gr.update(value='')
 
 with gr.Blocks() as demo:
     
@@ -92,13 +95,13 @@ with gr.Blocks() as demo:
             app_state['step'] = step
             app_state['img_list'].append(res)
             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=1)
         app_state['running'] = False
         app_state['img'] = None
         app_state['status_msg'] = ''
+        app_state['dots'] = 0
         return gr.update(value=img.images[0], label='Generated image')
     
     app_state = gr.State({'img': None, 
@@ -122,7 +125,7 @@ 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='')
+    status_msg = gr.Markdown()
     
     gallery = gr.Gallery()
     image = gr.Image(label='Illustration for your story', show_label=True)
@@ -140,11 +143,12 @@ 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, status_msg], every=10)
-    eventslider.change(fn=on_change_event, inputs=[app_state], outputs=[gallery, status_msg], every=10, cancels=[dep])
+    dep = demo.load(on_change_event, app_state, [gallery, status_msg], every=5)
+    eventslider.change(fn=on_change_event, inputs=[app_state], outputs=[gallery, status_msg], every=5, cancels=[dep])
     
 
 if READ_TOKEN:
     demo.queue().launch()
 else:
     demo.queue().launch(share=True, debug=True)
+