Update app.py

app.py

@@ -7,11 +7,11 @@ from flask import Flask, Response, render_template, stream_with_context
 
 app = Flask(__name__)
 random.seed()  # Initialize the random number generator
-global reqs
-reqs = 0
 
+# Initialize a dictionary to store requests per second
+requests_per_second = {}
 
-@app.route('/dstat')
+@app.route('/')
 def index():
   with open("index.html", "r") as f:
     html = f.read()
@@ -21,10 +21,16 @@ def index():
 def chart_data():
     def generate_random_data():
         while True:
+            current_time = datetime.now().strftime('%M:%S')
+            if current_time not in requests_per_second:
+                requests_per_second[current_time] = 0
+            
             json_data = json.dumps(
-                {'time': datetime.now().strftime('%M:%S'), 'value': reqs})
+                {'time': current_time, 'value': requests_per_second[current_time]})
             yield f"data:{json_data}\n\n"
-            time.sleep(1)
+            time.sleep(1)  # Wait for 1 second
+            # Reset the count for the next second
+            requests_per_second[current_time] = 0
 
     response = Response(stream_with_context(generate_random_data()), mimetype="text/event-stream")
     response.headers["Cache-Control"] = "no-cache"
@@ -33,9 +39,13 @@ def chart_data():
 
 @app.route("/attack")
 def attack():
-  global reqs
-  reqs = reqs + 1
-  return "200"
+    # Increment the request count for the current second
+    current_time = datetime.now().strftime('%M:%S')
+    if current_time in requests_per_second:
+        requests_per_second[current_time] += 1
+    else:
+        requests_per_second[current_time] = 1
+    return "200"
 
 
 if __name__ == '__main__':