malteos commited on
Commit
ceb63d6
·
verified ·
1 Parent(s): 5ee9649

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ from fastapi import FastAPI
2
  import os
3
  import duckdb
4
 
5
- data_dir = "./data"
6
  db_path = data_dir + "/db/cc_explorer.db"
7
 
8
  print("db_path", db_path)
@@ -28,5 +28,5 @@ def greet_json():
28
  except Exception:
29
  latest_crawl_id = "n/a"
30
 
31
- return {"Hello": "World!", "latest_crawl": latest_crawl_id, "exists": os.path.exists(db_path), "data_ls": os.listdir(data_dir)}
32
 
 
2
  import os
3
  import duckdb
4
 
5
+ data_dir = os.environ.get("DATA_DIR", "./data")
6
  db_path = data_dir + "/db/cc_explorer.db"
7
 
8
  print("db_path", db_path)
 
28
  except Exception:
29
  latest_crawl_id = "n/a"
30
 
31
+ return {"Hello": "World!", "latest_crawl": latest_crawl_id, "exists": os.path.exists(db_path), "data_ls": os.listdir(data_dir), "db_path": db_path}
32