From 2286108be1b093070ea20ff8cb078494b9151109 Mon Sep 17 00:00:00 2001
From: Laurent <laurent0206.cr@gmail.com>
Date: Wed, 31 Dec 2025 14:19:00 +0100
Subject: [PATCH] Try stuff

---
 api/mosquitto/data/mosquitto.db   | Bin 164 -> 164 bytes
 rpi/main.py                       |  15 ++++++++++-
 rpi/services/detection_service.py |  40 +++++++++++++-----------------
 rpi/services/game_service.py      |  16 +++---------
 4 files changed, 34 insertions(+), 37 deletions(-)

diff --git a/api/mosquitto/data/mosquitto.db b/api/mosquitto/data/mosquitto.db
index ba9bb7a0c18c0208e6936327ba0a5efe3c0895a3..61ef173bd5c4a775dc3057f70243294f470f131e 100644
GIT binary patch
delta 11
ScmZ3&xP)<nJmZ^*3M~K^!vq`v

delta 11
ScmZ3&xP)<nJfp`%g%$u6GXr=4

diff --git a/rpi/main.py b/rpi/main.py
index bb445030..5b9f0e18 100644
--- a/rpi/main.py
+++ b/rpi/main.py
@@ -1,6 +1,8 @@
 #!/usr/bin/env python3
 import os
 
+import cv2
+import requests
 from dotenv import load_dotenv
 from flask import Flask, jsonify
 
@@ -62,7 +64,18 @@ def start_party():
 @app.route("/command/party/play", methods=['POST'])
 def make_move():
     try:
-        game_service.make_move()
+        frame, fen = game_service.make_move()
+        print(fen)
+
+        encoded_frame = cv2.imencode('.jpg', frame)
+        image_bytes = encoded_frame[1].tobytes()
+        headers = {'Content-Type': 'image/jpeg'}
+        response = requests.post(
+            "https://192.168.15.125:1880/party/image",
+            data=image_bytes,
+            headers=headers,
+            verify=False)
+        print(response.status_code)
         return jsonify({"status": "ok", "message": "Party started"}), 200
     except Exception as ex:
         print(ex)
diff --git a/rpi/services/detection_service.py b/rpi/services/detection_service.py
index 9d0d96c0..4b98cb25 100644
--- a/rpi/services/detection_service.py
+++ b/rpi/services/detection_service.py
@@ -40,11 +40,25 @@ class DetectionService:
     def stop(self):
         self.camera.close()
 
-    def analyze_single_frame(self) -> tuple[np.ndarray, str | None]:
+    def analyze_single_frame(self) -> tuple[bytes, str | None]:
         frame = self.camera.take_photo()
-        fen = self.__get_fen(frame)
-        return frame, fen
+        result = self.__run_complete_detection(frame)
 
+        edges_prediction = result["edges"]
+        pieces_prediction = result["pieces"]
+
+        processed_frame = self.board_manager.process_frame(edges_prediction[0], frame, self.scale_size)
+        if processed_frame is None:
+            return frame.tobytes(), None
+
+        warped_corners = processed_frame[0]
+        matrix = processed_frame[1]
+
+        detections = self.pieces_manager.extract_pieces(pieces_prediction)
+
+        board = self.pieces_manager.pieces_to_board(detections, warped_corners, matrix, self.scale_size)
+
+        return pieces_prediction[0].plot().tobytes(), self.pieces_manager.board_to_fen(board)
 
     def __run_complete_detection(self, frame : np.ndarray, display=False) -> dict[str, list[Results]] :
         pieces_prediction = self.__run_pieces_detection(frame)
@@ -71,26 +85,6 @@ class DetectionService:
             self.__display_frame(prediction[0].plot())
         return prediction
 
-
-    def __get_fen(self, frame : np.ndarray) -> str | None:
-        result = self.__run_complete_detection(frame)
-
-        edges_prediction = result["edges"]
-        pieces_prediction = result["pieces"]
-
-        processed_frame = self.board_manager.process_frame(edges_prediction[0], frame, self.scale_size)
-        if processed_frame is None:
-            return None
-
-        warped_corners = processed_frame[0]
-        matrix = processed_frame[1]
-
-        detections = self.pieces_manager.extract_pieces(pieces_prediction)
-
-        board = self.pieces_manager.pieces_to_board(detections, warped_corners, matrix, self.scale_size)
-
-        return self.pieces_manager.board_to_fen(board)
-
     def __display_frame(self, frame : np.ndarray):
         cv2.namedWindow("Frame", cv2.WINDOW_NORMAL)
         cv2.resizeWindow("Frame", self.scale_size[0], self.scale_size[1])
diff --git a/rpi/services/game_service.py b/rpi/services/game_service.py
index 71060443..524cf9ce 100644
--- a/rpi/services/game_service.py
+++ b/rpi/services/game_service.py
@@ -1,4 +1,5 @@
 import cv2
+import numpy as np
 import requests
 
 from services.clock_service import ClockService
@@ -22,16 +23,5 @@ class GameService:
         self.clock_service.stop()
         self.detection_service.stop()
 
-    def make_move(self) -> None:
-        img, fen = self.detection_service.analyze_single_frame()
-        print(fen)
-
-        encoded_frame = cv2.imencode('.jpg', img)
-        image_bytes = encoded_frame[1].tobytes()
-        headers = {'Content-Type': 'image/jpeg'}
-        response = requests.post(
-            "https://192.168.15.125:1880/party/image",
-            data=image_bytes,
-            headers=headers,
-            verify=False)
-        print(response.status_code)
\ No newline at end of file
+    def make_move(self) -> tuple[bytes, str]:
+        return self.detection_service.analyze_single_frame()
\ No newline at end of file