apply systematic btn en/disable in chargen->storygen

app.py

@@ -266,7 +266,7 @@ with gr.Blocks(css=STYLE) as demo:
 			story_writing_done_btn = gr.Button("export your story →", elem_classes=["wrap", "control-button"], scale=2)
 
 	with gr.Column(visible=False) as export_phase:
-		gr.Markdown("# 📤 Story writing")
+		gr.Markdown("# 📤 Export your story")
 		gr.Markdown(desc.export_phase_description, elem_classes=["markdown-justify"])
 
 		title_txt = gr.Textbox("Your Own Story", elem_classes=["no-label"])
@@ -416,6 +416,16 @@ with gr.Blocks(css=STYLE) as demo:
 		view_change_ui.move_to_next_view,
 		inputs=None,
 		outputs=[character_setup_phase, writing_phase]
+	).then(
+		story_gen_ui.disable_btns,
+		inputs=None,
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
+		]
 	).then(
 		story_gen_ui.first_story_gen,
 		inputs=[			
@@ -443,9 +453,29 @@ with gr.Blocks(css=STYLE) as demo:
 		outputs=[
 			action_btn1, action_btn2, action_btn3, progress_comp
 		]
+	).then(
+		story_gen_ui.enable_btns,
+		inputs=[story_image, story_audio],
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
+		]
 	)
 
 	regen_actions_btn.click(
+		story_gen_ui.disable_btns,
+		inputs=None,
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
+		]
+	).then(
 		story_gen_ui.actions_gen,
 		inputs=[
 			cursors,
@@ -458,9 +488,29 @@ with gr.Blocks(css=STYLE) as demo:
 		outputs=[
 			action_btn1, action_btn2, action_btn3, progress_comp
 		]		
+	).then(
+		story_gen_ui.enable_btns,
+		inputs=[story_image, story_audio],
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
+		]
 	)
  
 	regen_story_btn.click(
+		story_gen_ui.disable_btns,
+		inputs=None,
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
+		]
+	).then(
 		story_gen_ui.update_story_gen,
 		inputs=[
 			cursors, cur_cursor,
@@ -486,6 +536,16 @@ with gr.Blocks(css=STYLE) as demo:
 		outputs=[
 			action_btn1, action_btn2, action_btn3, progress_comp
 		]
+	).then(
+		story_gen_ui.enable_btns,
+		inputs=[story_image, story_audio],
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
+		]
 	)
  
 	#### Setups
@@ -552,24 +612,74 @@ with gr.Blocks(css=STYLE) as demo:
 	)
 
 	image_gen_btn.click(
+		story_gen_ui.disable_btns,
+		inputs=None,
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
+		]
+	).then(
 		story_gen_ui.image_gen,
 		inputs=[
-			genre_dd, place_dd, mood_dd, title, story_content, cursors, cur_cursor, story_audio
+			genre_dd, place_dd, mood_dd, title, story_content, cursors, cur_cursor
 		],
 		outputs=[
-			story_image, img_audio_combine_btn, cursors, progress_comp,
+			story_image, cursors, progress_comp,
+		]
+	).then(
+		story_gen_ui.enable_btns,
+		inputs=[story_image, story_audio],
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
 		]
 	)
 
 	audio_gen_btn.click(
+		story_gen_ui.disable_btns,
+		inputs=None,
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
+		]
+	).then(
 		story_gen_ui.audio_gen,
 		inputs=[
-			genre_dd, place_dd, mood_dd, title, story_content, cursors, cur_cursor, story_image
+			genre_dd, place_dd, mood_dd, title, story_content, cursors, cur_cursor
 		],
-		outputs=[story_audio, img_audio_combine_btn, cursors, progress_comp]
+		outputs=[story_audio, cursors, progress_comp]
+	).then(
+		story_gen_ui.enable_btns,
+		inputs=[story_image, story_audio],
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
+		]
 	)
 
 	img_audio_combine_btn.click(
+		story_gen_ui.disable_btns,
+		inputs=None,
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
+		]
+	).then(
 		story_gen_ui.video_gen,
 		inputs=[
 			story_image, story_audio, story_content, cursors, cur_cursor
@@ -577,7 +687,17 @@ with gr.Blocks(css=STYLE) as demo:
 		outputs=[
 			story_image, story_audio, story_video, cursors, progress_comp
 		],
-	)	
+	).then(
+		story_gen_ui.enable_btns,
+		inputs=[story_image, story_audio],
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
+		]
+	)
 
 	story_progress.input(
 		story_gen_ui.move_story_cursor,
@@ -594,10 +714,14 @@ with gr.Blocks(css=STYLE) as demo:
 	)
 
 	action_btn1.click(
-		lambda: (gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False)),
+		story_gen_ui.disable_btns,
 		inputs=None,
 		outputs=[
-			image_gen_btn, audio_gen_btn, img_audio_combine_btn
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
 		]
 	).then(
 		story_gen_ui.next_story_gen,
@@ -629,13 +753,27 @@ with gr.Blocks(css=STYLE) as demo:
 		outputs=[
 			action_btn1, action_btn2, action_btn3, progress_comp
 		]
+	).then(
+		story_gen_ui.enable_btns,
+		inputs=[story_image, story_audio],
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
+		]
 	)
 
 	action_btn2.click(
-		lambda: (gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False)),
+		story_gen_ui.disable_btns,
 		inputs=None,
 		outputs=[
-			image_gen_btn, audio_gen_btn, img_audio_combine_btn
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
 		]
 	).then(
 		story_gen_ui.next_story_gen,
@@ -667,13 +805,27 @@ with gr.Blocks(css=STYLE) as demo:
 		outputs=[
 			action_btn1, action_btn2, action_btn3, progress_comp
 		]
+	).then(
+		story_gen_ui.enable_btns,
+		inputs=[story_image, story_audio],
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
+		]
 	)
 
 	action_btn3.click(
-		lambda: (gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False)),
+		story_gen_ui.disable_btns,
 		inputs=None,
 		outputs=[
-			image_gen_btn, audio_gen_btn, img_audio_combine_btn
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
 		]
 	).then(
 		story_gen_ui.next_story_gen,
@@ -705,13 +857,27 @@ with gr.Blocks(css=STYLE) as demo:
 		outputs=[
 			action_btn1, action_btn2, action_btn3, progress_comp
 		]
+	).then(
+		story_gen_ui.enable_btns,
+		inputs=[story_image, story_audio],
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
+		]
 	)
 
 	custom_action_txt.submit(
-		lambda: (gr.update(interactive=False), gr.update(interactive=False), gr.update(interactive=False)),
+		story_gen_ui.disable_btns,
 		inputs=None,
 		outputs=[
-			image_gen_btn, audio_gen_btn, img_audio_combine_btn
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
 		]
 	).then(
 		story_gen_ui.next_story_gen,
@@ -743,6 +909,16 @@ with gr.Blocks(css=STYLE) as demo:
 		outputs=[
 			action_btn1, action_btn2, action_btn3, progress_comp
 		]
+	).then(
+		story_gen_ui.enable_btns,
+		inputs=[story_image, story_audio],
+		outputs=[
+			image_gen_btn, audio_gen_btn, img_audio_combine_btn,
+			regen_actions_btn, regen_story_btn, custom_prompt_txt,
+			action_btn1, action_btn2, action_btn3,
+			custom_action_txt,
+			restart_from_story_generation_btn, story_writing_done_btn
+		]
 	)
 
 	### Chatbot
@@ -801,4 +977,4 @@ with gr.Blocks(css=STYLE) as demo:
 		outputs=[selected_side_char_image3]
 	)
 
-demo.queue().launch()
+demo.queue().launch(share=True)

interfaces/story_gen_ui.py

@@ -357,7 +357,7 @@ def video_gen(
 
 
 def image_gen(
-	genre, place, mood, title, story_content, cursors, cur_cursor, story_audio
+	genre, place, mood, title, story_content, cursors, cur_cursor
 ):
 	# generate prompts for background image with PaLM
 	for _ in range(3):
@@ -382,20 +382,15 @@ def image_gen(
 	
 	cursors[cur_cursor]["img"] = img_filename
 
-	video_gen_btn_state = gr.update(interactive=False)
-	if story_audio is not None:
-		video_gen_btn_state = gr.update(interactive=True)
-
 	return  (
 		gr.update(visible=True, value=img_filename),
-		video_gen_btn_state,
 		cursors,
 		"  "
 	)
 
 
 def audio_gen(
-	genre, place, mood, title, story_content, cursors, cur_cursor, story_image
+	genre, place, mood, title, story_content, cursors, cur_cursor
 ):
 	# generate prompt for background music with PaLM
 	for _ in range(3):
@@ -413,13 +408,8 @@ def audio_gen(
 	bgm_filename = bgm_maker.text2music(prompt, length=60)
 	cursors[cur_cursor]["audio"] = bgm_filename
 
-	video_gen_btn_state = gr.update(interactive=False)
-	if story_image is not None:
-		video_gen_btn_state = gr.update(interactive=True)
-
 	return (
 		gr.update(visible=True, value=bgm_filename),
-		video_gen_btn_state,
 		cursors,
 		" "
 	)
@@ -474,4 +464,50 @@ def move_story_cursor(moved_cursor, cursors):
 
 def update_story_content(story_content, cursors, cur_cursor):
 	cursors[cur_cursor]["story"] = story_content
-	return cursors
+	return cursors
+
+def disable_btns():
+	return (
+		gr.update(interactive=False), # image_gen_btn
+		gr.update(interactive=False), # audio_gen_btn
+		gr.update(interactive=False), # img_audio_combine_btn
+
+		gr.update(interactive=False), # regen_actions_btn
+		gr.update(interactive=False), # regen_story_btn
+		gr.update(interactive=False), # custom_prompt_txt
+
+		gr.update(interactive=False), # action_btn1
+		gr.update(interactive=False), # action_btn2
+		gr.update(interactive=False), # action_btn3
+
+		gr.update(interactive=False), # custom_action_txt
+
+		gr.update(interactive=False), # restart_from_story_generation_btn
+		gr.update(interactive=False), # story_writing_done_btn
+	)
+ 
+def enable_btns(story_image, story_audio):
+	video_gen_btn_state = gr.update(interactive=False)
+    
+	if story_image is not None and \
+		story_audio is not None:
+		video_gen_btn_state = gr.update(interactive=True)
+    
+	return (
+		gr.update(interactive=True), # image_gen_btn
+		gr.update(interactive=True), # audio_gen_btn
+		video_gen_btn_state, # img_audio_combine_btn
+
+		gr.update(interactive=True), # regen_actions_btn
+		gr.update(interactive=True), # regen_story_btn
+		gr.update(interactive=True), # custom_prompt_txt
+
+		gr.update(interactive=True), # action_btn1
+		gr.update(interactive=True), # action_btn2
+		gr.update(interactive=True), # action_btn3
+
+		gr.update(interactive=True), # custom_action_txt
+
+		gr.update(interactive=True), # restart_from_story_generation_btn
+		gr.update(interactive=True), # story_writing_done_btn
+	)