Anthony Truchet commited on
Commit
92c5ce5
·
1 Parent(s): d0ef630

feat: more robust persistent cache

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -2
  2. src/app.py +6 -1
Dockerfile CHANGED
@@ -44,7 +44,6 @@ RUN mkdir -p $HF_HOME
44
  WORKDIR $APP_HOME
45
  EXPOSE 8501
46
  CMD streamlit run $APP_CODE/app.py \
47
- --logger.level=debug \
48
  --server.enableXsrfProtection=true \
49
  --server.fileWatcherType=auto
50
-
 
44
  WORKDIR $APP_HOME
45
  EXPOSE 8501
46
  CMD streamlit run $APP_CODE/app.py \
47
+ --logger.level=info \
48
  --server.enableXsrfProtection=true \
49
  --server.fileWatcherType=auto
 
src/app.py CHANGED
@@ -17,7 +17,12 @@ DATA_PATH = pathlib.Path(os.environ.get("APP_DATA"))
17
 
18
  def cached_download_csv(url : str, **kwargs) -> pd.DataFrame:
19
  shelf_filename = DATA_PATH.joinpath("shelf.dat").as_posix()
20
- with shelve.open(shelf_filename, writeback=True, flag='c') as shelf:
 
 
 
 
 
21
  st.info(f"Opened or created the shelf file '{shelf_filename}'")
22
  maybe_cached = shelf.get(url)
23
  if maybe_cached is None:
 
17
 
18
  def cached_download_csv(url : str, **kwargs) -> pd.DataFrame:
19
  shelf_filename = DATA_PATH.joinpath("shelf.dat").as_posix()
20
+ try:
21
+ s = shelve.open(shelf_filename, writeback=True, flag='c')
22
+ except Exception as exn:
23
+ st.warning("Unexpected exception, by-passing persistent cache :\n" + str(exn))
24
+ return pd.read_csv(url, **kwargs)
25
+ with s as shelf:
26
  st.info(f"Opened or created the shelf file '{shelf_filename}'")
27
  maybe_cached = shelf.get(url)
28
  if maybe_cached is None: