@@ -325,7 +325,7 @@ def build(self):
325325
326326 def enable_button_if_ready (self , e ):
327327 """Enable the confirm button if both files have been selected."""
328- if self .selected_image .value and self .selected_recovery .value :
328+ if ( ".zip" in self .selected_image .value ) and ( ".img" in self .selected_recovery .value ) :
329329 self .confirm_button .disabled = False
330330 self .right_view .update ()
331331 else :
@@ -457,16 +457,22 @@ def pick_image_result(self, e: FilePickerResultEvent):
457457 self .selected_image .value += (
458458 ", " .join (map (lambda f : f .name , e .files )) if e .files else "Cancelled!"
459459 )
460- self .image_path = e .files [0 ].path
461- logger .info (f"Selected image from { self .image_path } " )
460+ if e .files :
461+ self .image_path = e .files [0 ].path
462+ logger .info (f"Selected image from { self .image_path } " )
463+ else :
464+ logger .info ("No image selected." )
462465 self .selected_image .update ()
463466
464467 def pick_recovery_result (self , e : FilePickerResultEvent ):
465468 self .selected_recovery .value += (
466469 ", " .join (map (lambda f : f .name , e .files )) if e .files else "Cancelled!"
467470 )
468- self .recovery_path = e .files [0 ].path
469- logger .info (f"Selected recovery from { self .recovery_path } " )
471+ if e .files :
472+ self .recovery_path = e .files [0 ].path
473+ logger .info (f"Selected recovery from { self .recovery_path } " )
474+ else :
475+ logger .info ("No image selected." )
470476 self .selected_recovery .update ()
471477
472478
0 commit comments