EL GHAFRAOUI AYOUB commited on
Commit
43a4aa3
·
1 Parent(s): 826d6c1
Files changed (2) hide show
  1. main.py +8 -17
  2. templates/index.html +2 -1
main.py CHANGED
@@ -8,11 +8,6 @@ from socketio import ASGIApp
8
 
9
  import cv2
10
  from flask import Flask, render_template, request, Response, session, redirect, url_for
11
- from flask import Flask, render_template
12
-
13
- from flask_socketio import SocketIO
14
- from flask_socketio import emit
15
- from flask_cors import CORS
16
 
17
  from flask_socketio import SocketIO
18
  import yt_dlp as youtube_dl
@@ -20,16 +15,11 @@ import uvicorn
20
 
21
  model_object_detection = YOLO("bisindov2.pt")
22
 
23
-
24
-
25
-
26
  app = Flask(__name__)
27
 
28
- socketio = SocketIO(app, cors_allowed_origins="*")
29
-
30
- CORS(app)
31
-
32
- app.secret_key = 'flask-sockets-builds'
33
 
34
 
35
  ######################################################
@@ -344,9 +334,10 @@ def stop_process():
344
  @socketio.on('connect')
345
  def test_connect():
346
  print('Connected')
347
- #emit('message', data, broadcast=True)
348
-
349
 
 
350
 
351
- if __name__ == '__main__':
352
- socketio.run(app, host='0.0.0.0', debug=True, allow_unsafe_werkzeug=True)
 
 
 
8
 
9
  import cv2
10
  from flask import Flask, render_template, request, Response, session, redirect, url_for
 
 
 
 
 
11
 
12
  from flask_socketio import SocketIO
13
  import yt_dlp as youtube_dl
 
15
 
16
  model_object_detection = YOLO("bisindov2.pt")
17
 
 
 
 
18
  app = Flask(__name__)
19
 
20
+ app.config['SECRET_KEY'] = 'secret!'
21
+ socketio = SocketIO(app, async_mode='threading')
22
+ stop_flag = False
 
 
23
 
24
 
25
  ######################################################
 
334
  @socketio.on('connect')
335
  def test_connect():
336
  print('Connected')
 
 
337
 
338
+ if __name__ == "__main__":
339
 
340
+ asgi_app = ASGIApp(app, socketio)
341
+ uvicorn.run(asgi_app, host='0.0.0.0', port=8000)
342
+
343
+
templates/index.html CHANGED
@@ -97,9 +97,10 @@
97
  <!-- Video -->
98
  <div class="col-span-8 mx-4 mt-3" style="background-color: lightgrey; border-radius: 10px;"> <!-- أضفت لون خلفية ونصبت دائرة لـ div الفيديو -->
99
  <div id="container">
100
- <img class="center" src="/video_feed" id="videoElement">
101
  </div>
102
  </div>
 
103
 
104
  <!-- Terminal -->
105
  <div class="col-span-2 mr-4" style="background-color: lightgrey; border-radius: 10px;"> <!-- أضفت لون خلفية ونصبت دائرة لـ div الطرفية -->
 
97
  <!-- Video -->
98
  <div class="col-span-8 mx-4 mt-3" style="background-color: lightgrey; border-radius: 10px;"> <!-- أضفت لون خلفية ونصبت دائرة لـ div الفيديو -->
99
  <div id="container">
100
+ <img class="center" src="{{ url_for('video_feed') }}" id="videoElement" alt="Video Feed">
101
  </div>
102
  </div>
103
+
104
 
105
  <!-- Terminal -->
106
  <div class="col-span-2 mr-4" style="background-color: lightgrey; border-radius: 10px;"> <!-- أضفت لون خلفية ونصبت دائرة لـ div الطرفية -->