Update app.py

app.py

@@ -1,220 +1,32 @@
-from ultralytics import YOLO
-import time
-import numpy as np
-import mediapipe as mp
-
+from flask import Flask, Response, render_template
 import cv2
-from flask import Flask, render_template, request, Response, session, redirect, url_for
-from flask_socketio import SocketIO
-import yt_dlp as youtube_dl
-
-model_object_detection = YOLO("bisindov2.pt")
+import base64
 
 app = Flask(__name__)
-app.config['SECRET_KEY'] = 'secret!'
-socketio = SocketIO(app, async_mode='threading')
-stop_flag = False
-
-# Translation mappings
-classes_translation = {
-    # Add your translations here
-}
-
-class VideoStreaming(object):
-    def __init__(self):
-        print ("===== Video Streaming =====")
-        self._preview = False
-        self._flipH = False
-        self._detect = False
-        self._mediaPipe = False
-        self._confidence = 75.0
-        self.mp_hands = mp.solutions.hands
-        self.hands = self.mp_hands.Hands()
-
-    @property
-    def confidence(self):
-        return self._confidence
-
-    @confidence.setter
-    def confidence(self, value):
-        self._confidence = int(value)
-
-    @property
-    def preview(self):
-        return self._preview
-
-    @preview.setter
-    def preview(self, value):
-        self._preview = bool(value)
-
-    @property
-    def flipH(self):
-        return self._flipH
-
-    @flipH.setter
-    def flipH(self, value):
-        self._flipH = bool(value)
-
-    @property
-    def detect(self):
-        return self._detect
-
-    @detect.setter
-    def detect(self, value):
-        self._detect = bool(value)
-
-    @property
-    def mediaPipe(self):
-        return self._mediaPipe
-
-    @mediaPipe.setter
-    def mediaPipe(self, value):
-        self._mediaPipe = bool(value)
-
-    def show(self, url):
-        print(url)
-        self._preview = False
-        self._flipH = False
-        self._detect = False
-        self._mediaPipe = False
-        self._confidence = 75.0
-
-        # Video capture setup
-        ydl_opts = {
-            "quiet": True,
-            "no_warnings": True,
-            "format": "best",
-            "forceurl": True,
-        }
+camera = cv2.VideoCapture(0)
 
-        #cap = cv2.VideoCapture(0 if url == '0' else url)
-        print("ayoub")
-        print(url)
-        cap = cv2.VideoCapture(0)
-        
-        while True:
-            if True:
-            #if self._preview:
-                print("ayoub2")
-                if stop_flag:
-                    print("Process Stopped")
-                    return
-                
-                grabbed, frame = cap.read()
-                
-                if not grabbed:
-                    break
+def generate_frames():
+    while True:
+        success, frame = camera.read()
+        if not success:
+            break
+        else:
+            # Preprocess the frame here (e.g., resize, color conversion)
+            processed_frame = frame  # Placeholder for preprocessing
 
-                if self.flipH:
-                    frame = cv2.flip(frame, 1)
+            _, buffer = cv2.imencode('.jpg', processed_frame)
+            frame_bytes = base64.b64encode(buffer.tobytes()).decode('utf-8')
 
-                # Perform YOLO object detection
-                if self.detect:
-                    frame_yolo = frame.copy()
-                    results_yolo = model_object_detection.predict(frame_yolo, conf=self._confidence / 100)
-                    frame_yolo, labels = results_yolo[0].plot()
-                    
-                    for label in labels:
-                        confidence = label.split(" ")[-1]
-                        label_name = " ".join(label.split(" ")[:-1])
-                        translated_label = classes_translation.get(label_name, label_name)
-                        list_labels = [translated_label, confidence]
-                        socketio.emit('label', list_labels)
+            yield frame_bytes
 
-                # Process hand landmarks with MediaPipe
-                if self.mediaPipe:
-                    image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-                    results = self.hands.process(image)
-                    
-                    if results.multi_hand_landmarks:
-                        for hand_landmarks in results.multi_hand_landmarks:
-                            mp.solutions.drawing_utils.draw_landmarks(
-                                frame,
-                                hand_landmarks,
-                                self.mp_hands.HAND_CONNECTIONS,
-                                landmark_drawing_spec=mp.solutions.drawing_utils.DrawingSpec(color=(255, 0, 0), thickness=4, circle_radius=3),
-                                connection_drawing_spec=mp.solutions.drawing_utils.DrawingSpec(color=(255, 255, 255), thickness=2, circle_radius=2),
-                            )
-
-                frame = cv2.imencode(".jpg", frame)[1].tobytes()
-                yield (b'--frame\r\nContent-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
-            else:
-                snap = np.zeros((1000, 1000), np.uint8)
-                label = "Streaming Off"
-                H, W = snap.shape
-                font = cv2.FONT_HERSHEY_PLAIN
-                color = (255, 255, 255)
-                cv2.putText(snap, label, (W//2 - 100, H//2), font, 2, color, 2)
-                frame = cv2.imencode(".jpg", snap)[1].tobytes()
-                yield (b'--frame\r\nContent-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
-
-
-# Initialize VideoStreaming object
-VIDEO = VideoStreaming()
-
-# Flask routes for the web application
-@app.route('/', methods=['GET', 'POST'])
-def homepage():
-    return render_template('hompage.html')
-
-@app.route('/index', methods=['GET', 'POST'])
+@app.route('/')
 def index():
-    global stop_flag
-    stop_flag = False
-    if request.method == 'POST':
-        url = request.form['url']
-        session['url'] = url
-        return redirect(url_for('index'))
     return render_template('index.html')
 
-@app.route('/video_feed')
-def video_feed():
-    url = session.get('url', None)
-    if url is None:
-        url = "0"
-        return Response(VIDEO.show(url), mimetype='multipart/x-mixed-replace; boundary=frame')
-    return Response(VIDEO.show(url), mimetype='multipart/x-mixed-replace; boundary=frame')
-
-# Button request routes for controlling settings
-@app.route("/request_preview_switch")
-def request_preview_switch():
-    VIDEO.preview = not VIDEO.preview
-    return "nothing"
-
-@app.route("/request_flipH_switch")
-def request_flipH_switch():
-    VIDEO.flipH = not VIDEO.flipH
-    return "nothing"
-
-@app.route("/request_run_model_switch")
-def request_run_model_switch():
-    VIDEO.detect = not VIDEO.detect
-    return "nothing"
-
-@app.route("/request_mediapipe_switch")
-def request_mediapipe_switch():
-    VIDEO.mediaPipe = not VIDEO.mediaPipe
-    return "nothing"
-
-@app.route('/update_slider_value', methods=['POST'])
-def update_slider_value():
-    slider_value = request.form['sliderValue']
-    VIDEO.confidence = slider_value
-    return 'OK'
-
-@app.route('/stop_process')
-def stop_process():
-    global stop_flag
-    stop_flag = True
-    return 'Process Stop Request'
-
-@socketio.on('connect')
-def test_connect():
-    print('Connected')
-
-
-
-
+@app.route('/capture_frame', methods=['GET', 'POST'])
+def capture_frame():
+    frame = next(generate_frames())
+    return frame
 
 if __name__ == '__main__':
-    socketio.run(app, host="0.0.0.0", allow_unsafe_werkzeug=True,port=7860)
+    app.run(debug=True)