Update app.py

app.py

@@ -14,16 +14,16 @@ app = Flask(__name__)
 app.secret_key = 'your_secret_key'  # Replace with a secure secret key
 
 # Global dictionary to hold job progress and results
-jobs = {}  # jobs[job_id] = {"progress": int, "result": {…}}
+jobs = {}  # jobs[job_id] = {"progress": int, "result": {...}}
 
 #########################################
 # 1. Initialize the Models
 #########################################
 
 # --- Roboflow Box Detection Model ---
-API_KEY = "wLjPoPYaLmrqCIOFA0RH"           # Your Roboflow API key
+API_KEY = "wLjPoPYaLmrqCIOFA0RH"  # Your Roboflow API key
 PROJECT_ID = "base-model-box-r4suo-8lkk1-6dbqh"  # Your Roboflow project ID
-VERSION_NUMBER = "2"                        # Your trained model version number
+VERSION_NUMBER = "2"  # Your trained model version number
 
 try:
     rf = roboflow.Roboflow(api_key=API_KEY)
@@ -146,16 +146,16 @@ def process_image(job_id, image_path, object_type, multiplier):
             jobs[job_id]['progress'] = 60
 
             # --- ArUco Marker Detection for Measurement ---
-            # We assume the printed marker is 5cm x 5cm.
+            # We assume the printed marker is 5 cm x 5 cm.
             marker_real_width_cm = 5.0
             try:
                 # Use a fresh copy for marker detection.
                 aruco_image = image.copy()
                 gray = cv2.cvtColor(aruco_image, cv2.COLOR_BGR2GRAY)
-                # Use the DICT_6X6_250 dictionary.
+                # Use DICT_6X6_250; your printed marker should be generated with this dictionary.
                 aruco_dict = cv2.aruco.getPredefinedDictionary(cv2.aruco.DICT_6X6_250)
                 aruco_params = cv2.aruco.DetectorParameters_create()
-                # Adjust parameters to help detect small markers.
+                # Tweak parameters for small markers.
                 aruco_params.adaptiveThreshWinSizeMin = 3
                 aruco_params.adaptiveThreshWinSizeMax = 23
                 aruco_params.adaptiveThreshWinSizeStep = 10
@@ -166,7 +166,7 @@ def process_image(job_id, image_path, object_type, multiplier):
                 print("DEBUG: Detected candidate IDs:", ids)
                 conversion_factor = None
                 if ids is not None and len(ids) > 0:
-                    # Look for marker with id 42.
+                    # Look for marker with ID 42
                     selected_index = None
                     for i, marker_id in enumerate(ids):
                         if marker_id[0] == 42:
@@ -176,7 +176,6 @@ def process_image(job_id, image_path, object_type, multiplier):
                         print("DEBUG: Marker with ID 42 not found among detected markers.")
                     else:
                         marker_corners = corners[selected_index].reshape((4, 2))
-                        # Draw only the selected marker
                         cv2.aruco.drawDetectedMarkers(image, [corners[selected_index]], [ids[selected_index]])
                         side_lengths = []
                         for i in range(4):
@@ -580,4 +579,3 @@ def result():
 
 if __name__ == '__main__':
     app.run(host="0.0.0.0", port=7860, threaded=True)
-