update logger

distance_calculate.py

@@ -8,6 +8,23 @@ from pydantic import BaseModel
 from ultralytics import YOLO
 from PIL import Image
 import traceback
+import logging
+
+# Create a logger
+logger = logging.getLogger("hots-pot")
+logger.setLevel(logging.DEBUG)
+
+# Create a file handler
+file_handler = logging.FileHandler("hots-pot.log")
+file_handler.setLevel(logging.DEBUG)
+
+# Create a formatter and add it to the handler
+# formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+formatter = logging.Formatter('%(asctime)s %(levelname)s - %(message)s')
+file_handler.setFormatter(formatter)
+
+# Add the handler to the logger
+logger.addHandler(file_handler)
 
 app = FastAPI()
 
@@ -109,19 +126,19 @@ def find_closest_points(lst):
             closest_points.update({(list(closest_objects_pair[0])[0], list(closest_objects_pair[1])[0]): round(min_distance, 2)})
     return closest_points
 
-def defult_response():
-    return {
-        "object_positions": {
-            "Regular Modular FnS": {"top": "107.55 cm", "bottom": "0.0 cm", "left": "103.1 cm", "right": "33.44 cm"},
-            "Focused Medium Core": {"top": "101.69 cm", "bottom": "0.0 cm", "left": "47.57 cm", "right": "85.74 cm"},
-            "Modular Small FnS": {"top": "83.9 cm", "bottom": "34.75 cm", "left": "49.15 cm", "right": "86.44 cm"}
-        },
-        "actual_distances": [
-            {"object": ["Regular Modular FnS", "Focused Medium Core"], "distances": "8.73 cm"},
-            {"object": ["Regular Modular FnS", "Modular Small FnS"], "distances": "12.97 cm"},
-            {"object": ["Focused Medium Core", "Modular Small FnS"], "distances": "1.42 cm"}
-        ]
-    }
+# def defult_response():
+#     return {
+#         "object_positions": {
+#             "Regular Modular FnS": {"top": "107.55 cm", "bottom": "0.0 cm", "left": "103.1 cm", "right": "33.44 cm"},
+#             "Focused Medium Core": {"top": "101.69 cm", "bottom": "0.0 cm", "left": "47.57 cm", "right": "85.74 cm"},
+#             "Modular Small FnS": {"top": "83.9 cm", "bottom": "34.75 cm", "left": "49.15 cm", "right": "86.44 cm"}
+#         },
+#         "actual_distances": [
+#             {"object": ["Regular Modular FnS", "Focused Medium Core"], "distances": "8.73 cm"},
+#             {"object": ["Regular Modular FnS", "Modular Small FnS"], "distances": "12.97 cm"},
+#             {"object": ["Focused Medium Core", "Modular Small FnS"], "distances": "1.42 cm"}
+#         ]
+#     }
 
 
 
@@ -134,13 +151,13 @@ def process_image(request: ImageRequest):
         image.thumbnail(size)
     except Exception:
         print(traceback.format_exc())
-        return defult_response()
+        # return defult_response()
     
     try:
         res = model(image)
     except Exception:
         print(traceback.format_exc())
-        return defult_response()
+        # return defult_response()
     
     try:
         objects_names_points, objects_names_width_pix = get_points_n_names(res)
@@ -150,7 +167,7 @@ def process_image(request: ImageRequest):
         object_positions = find_position(objects_names_points, par_pix_cm, image)
     except Exception:
         print(traceback.format_exc())
-        return defult_response()
+        # return defult_response()
     
     try:
         # Remove the distance between the same object
@@ -159,14 +176,21 @@ def process_image(request: ImageRequest):
                 actual_distances.remove(item)
     except Exception:
         print(traceback.format_exc())
-        return defult_response()
+        # return defult_response()
     
     try:
         # Convert the object names to the system names
         object_positions_sys, actual_distances_sys = name_conversion(actual_distances,object_positions, name_convert)
+        logger.info(json.dumps({'Position':object_positions_sys, 'Distance':actual_distances_sys}))
     except Exception:
         print(traceback.format_exc())
-        return defult_response()
+        # return defult_response()
+    
+    print("#" * 50)
+    print("object_positions_sys", object_positions_sys)
+    print("-" * 50)
+    print("actual_distances_sys", actual_distances_sys)
+    print("#" * 50)
     
     return {
         "object_positions": object_positions_sys,

hots-pot.log

@@ -0,0 +1,4 @@
+hots-pot - INFO - {"Position": {"Modular Small Core": {"top": "68.12 cm", "bottom": "66.92 cm", "left": "23.03 cm", "right": "50.37 cm"}, "Modular Small FnS": {"top": "70.21 cm", "bottom": "75.08 cm", "left": "70.73 cm", "right": "11.02 cm"}}, "Distance": [{"object": ["Modular Small Core", "Modular Small FnS"], "distances": "2.89 cm"}]}
+2025-04-13 13:03:53,222 INFO - {"Position": {"Modular Small Core": {"top": "68.12 cm", "bottom": "66.92 cm", "left": "23.03 cm", "right": "50.37 cm"}, "Modular Small FnS": {"top": "70.21 cm", "bottom": "75.08 cm", "left": "70.73 cm", "right": "11.02 cm"}}, "Distance": [{"object": ["Modular Small Core", "Modular Small FnS"], "distances": "2.89 cm"}]}
+2025-04-13 13:04:00,698 INFO - {"Position": {"Modular Small Core": {"top": "68.12 cm", "bottom": "66.92 cm", "left": "23.03 cm", "right": "50.37 cm"}, "Modular Small FnS": {"top": "70.21 cm", "bottom": "75.08 cm", "left": "70.73 cm", "right": "11.02 cm"}}, "Distance": [{"object": ["Modular Small Core", "Modular Small FnS"], "distances": "2.89 cm"}]}
+2025-04-13 13:04:04,891 INFO - {"Position": {"Modular Small Core": {"top": "68.12 cm", "bottom": "66.92 cm", "left": "23.03 cm", "right": "50.37 cm"}, "Modular Small FnS": {"top": "70.21 cm", "bottom": "75.08 cm", "left": "70.73 cm", "right": "11.02 cm"}}, "Distance": [{"object": ["Modular Small Core", "Modular Small FnS"], "distances": "2.89 cm"}]}

test.ipynb

@@ -1,50 +1,31 @@
 {
  "cells": [
-  {
-   "cell_type": "code",
-   "execution_count": 26,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "actual_distances = [{'object': ('focused_medium_core', 'regular_modular_fns'), 'distances': 14.7}, {'object': ('focused_medium_core', 'focused_medium_fns'), 'distances': 9.04}, {'object': ('focused_medium_core', 'modular_small_fns'), 'distances': 8.12}, {'object': ('regular_modular_fns', 'regular_modular_fns'), 'distances': 0.39}, {'object': ('regular_modular_fns', 'focused_medium_fns'), 'distances': 1.71}, {'object': ('regular_modular_fns', 'modular_small_fns'), 'distances': 10.6}, {'object': ('focused_medium_fns', 'modular_small_fns'), 'distances': 9.73}, {'object': ('modular_small_fns', 'modular_small_fns'), 'distances': 0.0}]"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 27,
-   "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[{'object': ('focused_medium_core', 'regular_modular_fns'), 'distances': 14.7},\n",
-       " {'object': ('focused_medium_core', 'focused_medium_fns'), 'distances': 9.04},\n",
-       " {'object': ('focused_medium_core', 'modular_small_fns'), 'distances': 8.12},\n",
-       " {'object': ('regular_modular_fns', 'regular_modular_fns'), 'distances': 0.39},\n",
-       " {'object': ('regular_modular_fns', 'focused_medium_fns'), 'distances': 1.71},\n",
-       " {'object': ('regular_modular_fns', 'modular_small_fns'), 'distances': 10.6},\n",
-       " {'object': ('focused_medium_fns', 'modular_small_fns'), 'distances': 9.73},\n",
-       " {'object': ('modular_small_fns', 'modular_small_fns'), 'distances': 0.0}]"
-      ]
-     },
-     "execution_count": 27,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
-   "source": [
-    "actual_distances"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
-    "for item in actual_distances[:]:\n",
-    "    if item['object'][0] == item['object'][1]:\n",
-    "        actual_distances.remove(item)"
+    "import logging\n",
+    "\n",
+    "# Create a logger\n",
+    "logger = logging.getLogger(\"hots-pot\")\n",
+    "logger.setLevel(logging.DEBUG)\n",
+    "\n",
+    "# Create a file handler\n",
+    "file_handler = logging.FileHandler(\"hots-pot.log\")\n",
+    "file_handler.setLevel(logging.DEBUG)\n",
+    "\n",
+    "# Create a formatter and add it to the handler\n",
+    "formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')\n",
+    "file_handler.setFormatter(formatter)\n",
+    "\n",
+    "# Add the handler to the logger\n",
+    "logger.addHandler(file_handler)\n",
+    "\n",
+    "# Example usage\n",
+    "logger.info(\"This is an info message\")\n",
+    "logger.error(\"This is an error\")\n"
    ]
   }
  ],