Update app.py

app.py

@@ -9,18 +9,16 @@ import time
 
 # Configuration
 CONFIG_FILE = "config.yaml"
-DATA_FILE = "/data/status.json"  # Persistent storage path
+DATA_FILE = "/data/status.json"
 
 # 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)
@@ -28,19 +26,15 @@ def load_data():
         return {}
 
 def save_data(data):
-    """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)
@@ -53,7 +47,6 @@ 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()
@@ -72,7 +65,6 @@ 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 {}
@@ -86,7 +78,6 @@ 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()
@@ -95,17 +86,19 @@ def start_background_updates():
     thread = threading.Thread(target=worker, daemon=True)
     thread.start()
 
-# Gradio Interface
+# Start background updates when app launches
+start_background_updates()
+
+# Create Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Website Status Monitor")
     status_output = gr.Markdown()
     
-    # Initial load and start background updates
+    # Initial load
     demo.load(
-        fn=lambda: [update_dashboard(), start_background_updates()],
+        fn=update_dashboard,
         outputs=status_output
     )
 
-# Launch the app
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", 7860)))