Update app.py

app.py

@@ -1,44 +1,55 @@
-from flask import Flask, render_template, Response
-import cv2
-import mediapipe as mp
-
-app = Flask(__name__)
-
-# Initialize webcam and Mediapipe Pose detection
-mp_pose = mp.solutions.pose
-pose = mp_pose.Pose()
-cap = cv2.VideoCapture(0, cv2.CAP_DSHOW)  # Use DirectShow for better camera compatibility
-
-def generate_frames():
-    while True:
-        success, frame = cap.read()
-        if not success:
-            break
-        else:
-            # Convert BGR to RGB for Mediapipe processing
-            rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-            results = pose.process(rgb_frame)
-
-            # Draw pose landmarks
-            if results.pose_landmarks:
-                mp.solutions.drawing_utils.draw_landmarks(
-                    frame, results.pose_landmarks, mp_pose.POSE_CONNECTIONS
-                )
-
-            # Encode frame to JPEG format
-            ret, buffer = cv2.imencode('.jpg', frame)
-            frame = buffer.tobytes()
-
-        yield (b'--frame\r\n'
-               b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
-
-@app.route('/')
-def index():
-    return render_template('index.html')  # HTML page to display video
-
-@app.route('/video')
-def video():
-    return Response(generate_frames(), mimetype='multipart/x-mixed-replace; boundary=frame')
-
-if __name__ == "__main__":
-    app.run(debug=True)
+import sys
+
+import qdarktheme
+from PyQt5.QtWidgets import QApplication
+
+from app_controllers.controller import Controller
+from app_models.model import Model
+from app_views.view import View
+
+model_name = "small640.pt"
+
+
+class App:
+    def __init__(self):
+        super().__init__()
+        self.model = Model(model_name)
+        self.view = View(self.model)
+        self.controller = Controller(self.model, self.view)
+        print("All modules loaded")
+
+
+if __name__ == '__main__':
+    if len(sys.argv) > 2:
+        print("Error: Exactly one argument is expected")
+        sys.exit(1)
+    elif len(sys.argv) == 1:
+        print("Info: Loading default inference model: {}".format(model_name))
+    else:
+        model_name = sys.argv[1]
+    qdarktheme.enable_hi_dpi()
+    app = QApplication([])
+    qdarktheme.setup_theme('dark')
+    # update global stylesheet
+    current_stylesheet = app.styleSheet()
+    updated_stylesheet = current_stylesheet + """
+        QSlider {background-color: #323844;}
+        QGroupBox {background-color: #323844;}
+        QLabel {font-weight: bold;}
+        QPushButton{font-weight: bold;}
+        QCheckBox {font-weight: bold;}
+        QRadioButton {font-weight: bold;}
+        QComboBox {font-weight: bold;}
+        QToolTip {
+            background-color: #323844;
+            color: white;
+            font-size: 12px;
+            border: 1px solid #aaa;
+            border-radius: 4px;
+            padding: 5px;
+        }
+    """
+    app.setStyleSheet(updated_stylesheet)
+    window = App()
+    window.view.show()
+    sys.exit(app.exec())