start sh

Dockerfile

@@ -13,8 +13,6 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 COPY --chown=user . /app
 
-RUN hf-mount status
-
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["bash", "start.sh"]
 
 

app.py

@@ -14,14 +14,18 @@ def greet_json():
     con = duckdb.connect()
     con.execute("INSTALL sqlite; LOAD sqlite;")
 
-    row = con.execute("""
-        SELECT crawl_id
-        FROM sqlite_scan(?, 'crawls')
-        ORDER BY crawl_date DESC
-        LIMIT 1
-    """, [db_path]).fetchone()
+    try: 
+        row = con.execute("""
+            SELECT crawl_id
+            FROM sqlite_scan(?, 'crawls')
+            ORDER BY crawl_date DESC
+            LIMIT 1
+        """, [db_path]).fetchone()
 
-    latest_crawl_id = row[0]
+        latest_crawl_id = row[0]
 
-    return {"Hello": "World!", "latest_crawl": latest_crawl, "exists": os.path.exists(db_path)}
+    except Exception:
+        latest_crawl_id = "n/a"
+
+    return {"Hello": "World!", "latest_crawl": latest_crawl_id, "exists": os.path.exists(db_path)}
 

start.sh

@@ -0,0 +1,5 @@
+#!/bin/bash
+
+hf-mount status
+
+uvicorn app:app --host 0.0.0.0 --port 7860