Update test_streaming.py

test_streaming.py

@@ -1,11 +1,61 @@
-from flask import Flask, render_template, request, Response
+# from flask import Flask, render_template, request, Response
+# import base64
+# import cv2
+# import numpy as np
+# from datetime import datetime
+
+# app = Flask(__name__)
+# latest_frame = None  # Store the latest frame
+
+# @app.route('/')
+# def index():
+#     return render_template('test_streaming_index.html')
+
+# @app.route('/upload_frame', methods=['POST'])
+# def upload_frame():
+#     global latest_frame
+#     data_url = request.json['image']
+#     header, encoded = data_url.split(",", 1)
+#     img_bytes = base64.b64decode(encoded)
+#     nparr = np.frombuffer(img_bytes, np.uint8)
+#     frame = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
+#     latest_frame = frame
+#     return '', 204
+
+# def stream():
+#     global latest_frame
+#     while True:
+#         if latest_frame is not None:
+#             _, buffer = cv2.imencode('.jpg', latest_frame)
+#             frame = buffer.tobytes()
+#             yield (b'--frame\r\n'
+#                    b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
+#         else:
+#             yield b''
+            
+# @app.route('/video_feed')
+# def video_feed():
+#     return Response(stream(), mimetype='multipart/x-mixed-replace; boundary=frame')
+
+# if __name__ == '__main__':
+#     app.run(host='0.0.0.0', port=7860)
+
+from flask import Flask, render_template, request, Response, session
 import base64
 import cv2
 import numpy as np
 from datetime import datetime
+from uuid import uuid4
 
 app = Flask(__name__)
-latest_frame = None  # Store the latest frame
+app.secret_key = 'your-secret-key'  # Needed for session management
+
+user_frames = {}  # Dictionary to store frames per user session
+
+@app.before_request
+def ensure_session():
+    if 'uid' not in session:
+        session['uid'] = str(uuid4())  # Unique ID per user
 
 @app.route('/')
 def index():
@@ -13,29 +63,30 @@ def index():
 
 @app.route('/upload_frame', methods=['POST'])
 def upload_frame():
-    global latest_frame
+    uid = session['uid']
     data_url = request.json['image']
     header, encoded = data_url.split(",", 1)
     img_bytes = base64.b64decode(encoded)
     nparr = np.frombuffer(img_bytes, np.uint8)
     frame = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
-    latest_frame = frame
+    user_frames[uid] = frame  # Store frame per user
     return '', 204
 
-def stream():
-    global latest_frame
+def stream(uid):
     while True:
-        if latest_frame is not None:
-            _, buffer = cv2.imencode('.jpg', latest_frame)
-            frame = buffer.tobytes()
+        frame = user_frames.get(uid)
+        if frame is not None:
+            _, buffer = cv2.imencode('.jpg', frame)
+            frame_bytes = buffer.tobytes()
             yield (b'--frame\r\n'
-                   b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
+                   b'Content-Type: image/jpeg\r\n\r\n' + frame_bytes + b'\r\n')
         else:
             yield b''
-            
+
 @app.route('/video_feed')
 def video_feed():
-    return Response(stream(), mimetype='multipart/x-mixed-replace; boundary=frame')
+    uid = session['uid']
+    return Response(stream(uid), mimetype='multipart/x-mixed-replace; boundary=frame')
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)