Further debugging

Dockerfile

@@ -1,6 +1,6 @@
 FROM pytorch/pytorch:latest
 
-RUN apt update && apt install -y && apt install ffmpeg && rm -rf /var/lib/apt/lists/*
+RUN apt update && apt install -y && apt install -y ffmpeg&& rm -rf /var/lib/apt/lists/*
 
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
@@ -23,6 +23,13 @@ COPY --chown=user src ./src
 RUN python3 -m pip install .
 
 COPY --chown=user "demo.py" .
-EXPOSE 7860 8080
 
-CMD ["python3", "demo.py"]
+# EXPOSE 7860 8080
+
+# COPY --chown=user "start.sh" .
+# RUN chmod +x "start.sh"
+
+# CMD ["./start.sh"]
+
+EXPOSE 7860
+CMD ["python3", "./demo.py"]

demo.py

@@ -4,11 +4,14 @@ import gradio as gr
 
 client = httpx.Client()
 # TODO: update the tags somehow; re-fetching inside of check_status doesn't work
-tag_counts = {
-    key.strip("#"): val
-    for key, val in client.get("http://localhost:8080/tag_counts/").json().items()
-    if key.strip("#")
-}
+try:
+    tag_counts = {
+        key.strip("#"): val
+        for key, val in client.get("http://localhost:8080/tag_counts/").json().items()
+        if key.strip("#")
+    }
+except httpx.ConnectError:
+    tag_counts = {}
 
 
 def submit(inputs):
@@ -19,8 +22,12 @@ def submit(inputs):
 
 
 def check_status():
-    response = httpx.get("http://localhost:8080/check_job_status/")
-    return response.json()
+    try:
+        response = httpx.get("http://localhost:8080/check_job_status/")
+        result = response.json()
+    except httpx.ConnectError:
+        result = "Server could not be reached"
+    return result
 
 
 def get_results(inputs: list[str]):