update

app.py

@@ -266,8 +266,6 @@ with gr.Blocks() as demo:
 										("IC-U-Net", "ICUNet"),
 										("IC-U-Net++", "UNetpp"),
 										("IC-U-Net-Attn", "AttUnet")],
-										#"(mapped data)",
-										#"(denoised data)"],
 										value="EEGART",
 										label="Model")
 					run_btn = gr.Button("Run", interactive=False)
@@ -302,7 +300,6 @@ with gr.Blocks() as demo:
 			os.mkdir(rootpath+"/session_data/")
 			#print(e)
 		os.mkdir(rootpath+"/session_data/stage1/")
-		os.mkdir(rootpath+"/session_data/stage2/")
 		
 		# initialize channel_info, app_info
 		channel_info = {}
@@ -329,11 +326,12 @@ with gr.Blocks() as demo:
 	    		]
 			},
 			"stage2" : {
-				"filePath" : rootpath+"/session_data/stage2/",
+				"filePath" : "",
 				"fileNames" : {
 					"input_data" : "",
 					"output_data" : ""
 				},
+				"state" : "",
 				"totalBatchNum" : None,
 				"sampleRate" : None
 			}
@@ -815,9 +813,10 @@ with gr.Blocks() as demo:
 	# +========================================================================================+
 	# |                                  Stage2: decode data                                   |
 	# +========================================================================================+
+	# verify that all required inputs have been provided
 	@gr.on(triggers = [in_data_file.upload, in_data_file.clear, in_samplerate.change, out_json_file.change],
 			inputs = [app_info_json, in_data_file, in_samplerate], outputs = run_btn)
-	def check_input2(app_info, in_data, samplerate):
+	def check_input(app_info, in_data, samplerate):
 		if app_info["stage1"]["state"]=="finished" and in_data!=None and samplerate!="":
 			return gr.Button(interactive=True)
 		else:
@@ -826,11 +825,11 @@ with gr.Blocks() as demo:
 	def reset_run(app_info, in_data, samplerate, modelname):
 		stage1_info = app_info["stage1"]
 		stage2_info = app_info["stage2"]
-	    
-	    # delete the previous folder of Stage2
-		filepath = stage2_info["filePath"]
-		utils.dataDelete(filepath)
-		# establish a new folder for Stage2
+		
+		# delete the previous folder
+		if stage2_info["state"] == "finished":
+			utils.dataDelete(stage2_info["filePath"])
+		# establish a new folder
 		new_filepath = app_info["rootPath"]+"stage2_"+str(random.randint(1,10000))+"/"
 		os.mkdir(new_filepath)
 		# generate the output filename
@@ -843,6 +842,7 @@ with gr.Blocks() as demo:
 				"input_data" : in_data,
 				"output_data" : new_filepath + new_filename
 			},
+			"state" : "running",
 			"sampleRate" : int(samplerate)
 		})
 		app_info["stage2"] = stage2_info
@@ -899,7 +899,9 @@ with gr.Blocks() as demo:
 			yield {run_btn : gr.Button(visible=True),
 					cancel_btn : gr.Button(visible=False)}
 		else:
-			yield {run_btn : gr.Button(visible=True),
+			app_info["stage2"]["state"] = "finished"
+			yield {app_info_json : app_info,
+					run_btn : gr.Button(visible=True),
 					cancel_btn : gr.Button(visible=False),
 					batch_md : gr.Markdown(visible=False),
 					out_data_file : gr.File(new_filename, visible=True)}
@@ -916,7 +918,7 @@ with gr.Blocks() as demo:
 	).success(
 		fn = run_model,
 		inputs = [app_info_json, in_modelname],
-		outputs = [run_btn, cancel_btn, batch_md, out_data_file]
+		outputs = [app_info_json, run_btn, cancel_btn, batch_md, out_data_file]
 	)
 	
 if __name__ == "__main__":