Update app.py

app.py

@@ -3,12 +3,17 @@ import os
 import psutil
 import time
 import datetime
+import json
 
 app = Flask(__name__)
 
 # Define the path to static files
 static_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)))
 
+def write_visitor_count_to_json(data):
+    with open('visit_data.json', 'w') as f:
+        json.dump(data, f)
+
 @app.route('/css/<path:filename>')
 def css(filename):
     return send_from_directory(os.path.join(static_dir, 'css'), filename)
@@ -51,10 +56,18 @@ def update_visitor_counts():
             last_update_date = current_date
         visitor_count += 1
         visitor_today += 1
+        data = {
+            'visitor_count': visitor_count,
+            'visitor_today': visitor_today,
+            'last_update_date': last_update_date.strftime('%Y-%m-%d')
+        }
+        write_visitor_count_to_json(data)
 
 @app.route('/count')
 def count():
-    return jsonify({'visitor_count': visitor_count, 'visitor_today': visitor_today})
+    with open('visit_data.json', 'r') as f:
+        data = json.load(f)
+    return jsonify(data)
 
 # Define the status route
 @app.route('/status')
@@ -66,7 +79,7 @@ def status():
     return jsonify({'runtime': uptime, 'memory': f'{memory_free} / {memory_total}'})
 
 # Handle 404 errors
-@app.errorhandler(404)
+@app.errorhandler(40)
 def page_not_found(e):
     return send_from_directory(static_dir, '404.html'), 404