validation on file selection

pages/01_dcascade.py

@@ -152,6 +152,8 @@ def Page():
         filename, set_filename = solara.use_state("")
         river_filename, set_river_filename = solara.use_state("")
         q_filename, set_q_filename = solara.use_state("")
+        error_select_csv, set_error_select_csv = solara.use_state("")
+        error_select_shp, set_error_select_shp = solara.use_state("")
         # removed_data, set_removed = solara.use_state("")
         size, set_size = solara.use_state(0)
         file_browser, set_file_browser = solara.use_state(cast(Optional[Path], None))
@@ -175,18 +177,22 @@ def Page():
         
         def open_river_file():
             if file_browser:
-
-                file_path = os.path.join("public",file_browser)
-                set_river_filename(file_path)
-                # with open(file_path,"wb") as file:
-                #     file.write(content)
-                    # m.add_shp(str(file_browser), "Countries")
+                file_path = os.path.join(os.getcwd(),file_browser)
+                if file_path.endswith(".shp"):
+                    set_river_filename(file_path)
+                    set_error_select_shp(False)
+                else:
+                    set_error_select_shp(True)
+                    
 
         def open_q_file():
             if file_browser:
-
-                file_path = os.path.join("public",file_browser)
-                set_q_filename(file_path)
+                file_path = os.path.join(os.getcwd(),file_browser)
+                if file_path.endswith(".csv"):
+                    set_q_filename(file_path)
+                    set_error_select_csv(False)
+                else:
+                    set_error_select_csv(True)
 
         def remove_data():
             for file in os.listdir(os.getcwd()+"/public"):
@@ -252,12 +258,23 @@ def Page():
                     solara.Button("Select Q file",
                                 color="primary",
                                 on_click=open_q_file)
-
-
+                if error_select_csv:
+                    solara.Warning(f"Q file must be csv",
+                                text=True,
+                                dense=True,
+                                icon=True,
+                    )
+                if error_select_shp:
+                    solara.Warning(f"River file must be shp",
+                                text=True,
+                                dense=True,
+                                icon=True,
+                    )
 
                 solara.FileBrowser(
                                 can_select=False,
-                                on_file_open=set_file_browser
+                                on_file_open=set_file_browser,
+                                directory="./public"
                                 )
                 solara.Markdown("#### Run model") 
                 with solara.ColumnsResponsive(3, large=6):