Update app.py

app.py

@@ -9,13 +9,18 @@ import time
 
 # Configuration
 CONFIG_FILE = "config.yaml"
-DATA_FILE = "/data/status.json"
+DATA_FILE = "/data/status.json"  # Persistent storage path
+
+# Ensure the /data directory exists
+os.makedirs(os.path.dirname(DATA_FILE), exist_ok=True)
 
 def load_config():
+    """Load endpoints from config.yaml."""
     with open(CONFIG_FILE, "r") as f:
         return yaml.safe_load(f)["endpoints"]
 
 def load_data():
+    """Load status data from persistent storage."""
     try:
         with open(DATA_FILE, "r") as f:
             return json.load(f)
@@ -23,10 +28,19 @@ def load_data():
         return {}
 
 def save_data(data):
-    with open(DATA_FILE, "w") as f:
-        json.dump(data, f)
+    """Save status data to persistent storage."""
+    try:
+        with open(DATA_FILE, "w") as f:
+            json.dump(data, f)
+    except PermissionError:
+        # Fallback to temporary storage if /data isn't writable
+        temp_file = "/tmp/status.json"
+        with open(temp_file, "w") as f:
+            json.dump(data, f)
+        return temp_file
 
 def check_status(endpoint):
+    """Check the status of a single endpoint."""
     try:
         start = time.time()
         response = requests.get(endpoint["url"], timeout=10)
@@ -39,6 +53,7 @@ def check_status(endpoint):
         return {"status": "DOWN", "response_time": None, "error": str(e)}
 
 def update_dashboard():
+    """Update the dashboard with latest status."""
     endpoints = load_config()
     data = load_data()
     current_time = datetime.now().isoformat()
@@ -57,6 +72,7 @@ def update_dashboard():
     return format_dashboard(data)
 
 def format_dashboard(data):
+    """Format the dashboard output."""
     output = []
     for name, history in data.items():
         last_check = history[-1] if history else {}
@@ -70,6 +86,7 @@ def format_dashboard(data):
     return "\n".join(output)
 
 def start_background_updates():
+    """Start a background thread to periodically update the dashboard."""
     def worker():
         while True:
             update_dashboard()
@@ -78,6 +95,7 @@ def start_background_updates():
     thread = threading.Thread(target=worker, daemon=True)
     thread.start()
 
+# Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("# Website Status Monitor")
     status_output = gr.Markdown()
@@ -88,5 +106,6 @@ with gr.Blocks() as demo:
         outputs=status_output
     )
 
+# Launch the app
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", 7860)))