Anthony Truchet commited on
Commit ·
92c5ce5
1
Parent(s): d0ef630
feat: more robust persistent cache
Browse files- Dockerfile +1 -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=
|
| 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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:
|