update

app.py

@@ -127,8 +127,7 @@ init_js = """
 			styleSheet.insertRule(rule, styleSheet.cssRules.length);
 		}
 	}
-	if(exist == 0) 
-	styleSheet.insertRule(rule, styleSheet.cssRules.length);
+	if(exist == 0) styleSheet.insertRule(rule, styleSheet.cssRules.length);
 }
 """
 
@@ -340,13 +339,26 @@ with gr.Blocks() as demo:
 	# stage1-1
 	def mapping_result(app_state, channel_info, raw_loc):
 		filepath = app_state["filepath"]
-		filename = filepath+"raw_montage_"+str(random.randint(1,10000))+".png"
-		app_state["filenames"]["raw_montage"] = filename
+		filename1 = filepath+"raw_montage_"+str(random.randint(1,10000))+".png"
+		filename2 = filepath+"mapped_montage_"+str(random.randint(1,10000))+".png"
+		
+		#app_state["filenames"]["raw_montage"] = filename1
+		app_state["filenames"].update({
+		    "raw_montage" : filename1,
+		    "mapped_montage" : filename2
+		})
 		
 		raw_montage = read_custom_montage(raw_loc)
 		raw_fig = raw_montage.plot()
 		raw_fig.set_size_inches(5.6, 5.6)
-		raw_fig.savefig(filename, pad_inches=0)
+		raw_fig.savefig(filename1, pad_inches=0)
+		
+		# plot red dots on unmatched input channels
+		ax = raw_fig.axes[0]
+		coords = ax.collections[0].get_offsets().data
+		idx = [channel_info["inputByName"][channel]["index"] for channel in app_state["stage1UnassignedInputs"]]
+		ax.scatter(coords[idx,0]-0.0001, coords[idx,1]+0.0001, color='red')
+		raw_fig.savefig(filename2, pad_inches=0)
 		
 		# ------------------determine the next step-----------------------
 		
@@ -362,7 +374,7 @@ with gr.Blocks() as demo:
 			return {app_state_json : app_state,
 					desc_md : gr.Markdown("### Step1: Mapping result", visible=True),
 					tpl_montage : gr.Image(visible=True),
-					in_montage : gr.Image(value=filename, visible=True),
+					in_montage : gr.Image(value=filename2, visible=True),
 					run_btn : gr.Button(interactive=True)}
 		
 		# if matched channels < 30, and there're still some unmatched input channels
@@ -378,7 +390,7 @@ with gr.Blocks() as demo:
 		return {app_state_json : app_state,
 				desc_md : gr.Markdown("### Step1: Mapping result", visible=True),
 				tpl_montage : gr.Image(visible=True),
-				in_montage : gr.Image(value=filename, visible=True),
+				in_montage : gr.Image(value=filename2, visible=True),
 				next_btn : gr.Button("Next step", visible=True)}
 	
 	map_btn.click(