Update app.py

app.py

@@ -10,10 +10,6 @@ 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)
@@ -44,10 +40,11 @@ def info():
 visitor_count = 0
 visitor_today = 0
 last_update_date = datetime.datetime.now().date()
+visitor_total = 0
 
 @app.before_request
 def update_visitor_counts():
-    global visitor_count, visitor_today, last_update_date
+    global visitor_count, visitor_today, last_update_date, visitor_total
     allowed_paths = ['/ai', '/api', '/tool']
     if request.path.startswith(tuple(allowed_paths)):
         current_date = datetime.datetime.now().date()
@@ -56,18 +53,25 @@ 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)
+        visitor_total += 1
+        if datetime.datetime.now().hour == 0 and datetime.datetime.now().minute == 0:
+            reset_visitor_count()
 
 @app.route('/count')
 def count():
-    with open('visit_data.json', 'r') as f:
-        data = json.load(f)
-    return jsonify(data)
+    return jsonify({
+        'visitor_count': visitor_count,
+        'visitor_today': visitor_today,
+        'visitor_total': visitor_total
+    })
+
+@app.route('/reset')
+def reset_visitor_count():
+    global visitor_count, visitor_today, last_update_date
+    visitor_count = 0
+    visitor_today = 0
+    last_update_date = datetime.datetime.now().date()
+    return jsonify({'visitor_count': visitor_count, 'visitor_today': visitor_today})
 
 # Define the status route
 @app.route('/status')