update

app.py

@@ -339,10 +339,7 @@ with gr.Blocks() as demo:
 		raw_fig.set_size_inches(5.6, 5.6)
 		raw_fig.savefig(filename, pad_inches=0)
 		
-		
-		#################################
-		#### determine the next step ####
-		#################################
+		# ------------------determine the next step-----------------------
 		
 		in_num = len(channel_info["inputByIndex"])
 		matched_num = 30 - len(app_state["missingTemplates"])
@@ -407,19 +404,24 @@ with gr.Blocks() as demo:
 			
 			name = app_state["missingTemplates"][0]
 			label = name+' (1/'+str(app_state["totalFillingNum"])+')'
+			
 			if len(app_state["stage1UnassignedInputs"])==1 or app_state["totalFillingNum"]==1:
-				btn_label = "Next step"
+				return {app_state_json : app_state,
+					    channel_info_json : channel_info,
+					    desc_md : gr.Markdown("### step2"),
+					    tpl_montage : gr.Image(visible=False),
+					    map_montage : gr.Image(visible=False),
+					    radio : gr.Radio(choices=app_state["stage1UnassignedInputs"], value=[], label=label, visible=True),
+					    next_btn : gr.Button("Next step")}
 			else:
-				btn_label = "Next"
-			
-			return {app_state_json : app_state,
-					channel_info_json : channel_info,
-					desc_md : gr.Markdown("### step2"),
-					tpl_montage : gr.Image(visible=False),
-					map_montage : gr.Image(visible=False),
-					radio : gr.Radio(choices=app_state["stage1UnassignedInputs"], value=[], label=label, visible=True),
-					step2_btn : gr.Button(btn_label, visible=True),
-					next_btn : gr.Button(visible=False)}
+			    return {app_state_json : app_state,
+					    channel_info_json : channel_info,
+					    desc_md : gr.Markdown("### step2"),
+					    tpl_montage : gr.Image(visible=False),
+					    map_montage : gr.Image(visible=False),
+					    radio : gr.Radio(choices=app_state["stage1UnassignedInputs"], value=[], label=label, visible=True),
+					    step2_btn : gr.Button(visible=True),
+					    next_btn : gr.Button(visible=False)}
 		
 		# step1 -> step3
 		elif app_state["state"] == "step3-initializing":
@@ -602,17 +604,25 @@ with gr.Blocks() as demo:
 			value = [channel_info["inputByIndex"][i] for i in value]
 			label = name+' (1/'+str(app_state["totalFillingNum"])+')'
 			
-			return {app_state_json : app_state,
-					in_fill_mode : gr.Dropdown(visible=False),
-		    		fillmode_btn : gr.Button(visible=False),
-		    		chkbox_group : gr.CheckboxGroup(choices=channel_info["inputByIndex"],
-													value=value, label=label, visible=True),
-		            step3_btn : gr.Button(visible=True)}
+			if app_state["totalFillingNum"] == 1:
+			    return {app_state_json : app_state,
+					    in_fill_mode : gr.Dropdown(visible=False),
+		        		fillmode_btn : gr.Button(visible=False),
+		        		chkbox_group : gr.CheckboxGroup(choices=channel_info["inputByIndex"],
+													    value=value, label=label, visible=True),
+		                next_btn : gr.Button(visible=True)}
+			else:
+			    return {app_state_json : app_state,
+					    in_fill_mode : gr.Dropdown(visible=False),
+		        		fillmode_btn : gr.Button(visible=False),
+		        		chkbox_group : gr.CheckboxGroup(choices=channel_info["inputByIndex"],
+													    value=value, label=label, visible=True),
+		                step3_btn : gr.Button(visible=True)}
 	
 	fillmode_btn.click(
 	    fn = fill_value,
 	    inputs = [app_state_json, channel_info_json, in_fill_mode],
-		outputs = [app_state_json, in_fill_mode, fillmode_btn, chkbox_group, step3_btn, run_btn]
+		outputs = [app_state_json, in_fill_mode, fillmode_btn, chkbox_group, step3_btn, next_btn, run_btn]
 	).success(
 	    fn = None,
 	    js = init_js,
@@ -635,6 +645,7 @@ with gr.Blocks() as demo:
 		# update the current round
 		app_state["fillingCount"] += 1
 		
+		print('cntttttt:', app_state["fillingCount"])
 		target_name = app_state["missingTemplates"][app_state["fillingCount"]-1]
 		target_idx = channel_info["templateByName"][target_name]["index"]