Added some error logging
This commit is contained in:
+68
-63
@@ -54,71 +54,76 @@ def hello_world():
|
|||||||
|
|
||||||
@app.route("/cooking-params", methods=["POST"])
|
@app.route("/cooking-params", methods=["POST"])
|
||||||
async def cooking_params():
|
async def cooking_params():
|
||||||
data = request.get_json()
|
|
||||||
|
|
||||||
if not data:
|
|
||||||
return jsonify({"error": "Invalid or missing JSON payload"}), 400
|
|
||||||
|
|
||||||
# Extract user or device parameters
|
|
||||||
height_cm = float(data.get("dish_height", 4.0))
|
|
||||||
initial_temp_c = float(data.get("ir_initial_temp", 20.0))
|
|
||||||
microwave_wattage = int(data.get("microwave_wattage", 900))
|
|
||||||
defrost_mode = bool(data.get("defrost_mode", False))
|
|
||||||
|
|
||||||
# 1. Save Camera Image
|
|
||||||
camera_image_b64 = data.get("camera_image")
|
|
||||||
if not camera_image_b64:
|
|
||||||
return jsonify({"error": "Missing required field 'camera_image'"}), 400
|
|
||||||
|
|
||||||
filename = f"dish_{uuid.uuid4().hex}.jpg"
|
|
||||||
filepath = os.path.join(CAMERA_IMAGE_DIR, filename)
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
with open(filepath, "wb") as f:
|
data = request.get_json()
|
||||||
f.write(base64.b64decode(camera_image_b64))
|
|
||||||
data["camera_image"] = filepath
|
if not data:
|
||||||
|
return jsonify({"error": "Invalid or missing JSON payload"}), 400
|
||||||
|
|
||||||
|
# Extract user or device parameters
|
||||||
|
height_cm = float(data.get("dish_height", 4.0))
|
||||||
|
initial_temp_c = float(data.get("ir_initial_temp", 20.0))
|
||||||
|
microwave_wattage = int(data.get("microwave_wattage", 900))
|
||||||
|
defrost_mode = bool(data.get("defrost_mode", False))
|
||||||
|
|
||||||
|
# 1. Save Camera Image
|
||||||
|
camera_image_b64 = data.get("camera_image")
|
||||||
|
if not camera_image_b64:
|
||||||
|
return jsonify({"error": "Missing required field 'camera_image'"}), 400
|
||||||
|
|
||||||
|
filename = f"dish_{uuid.uuid4().hex}.jpg"
|
||||||
|
filepath = os.path.join(CAMERA_IMAGE_DIR, filename)
|
||||||
|
|
||||||
|
try:
|
||||||
|
with open(filepath, "wb") as f:
|
||||||
|
f.write(base64.b64decode(camera_image_b64))
|
||||||
|
data["camera_image"] = filepath
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[Error] Failed to save camera image: {str(e)}")
|
||||||
|
return jsonify({"error": f"Failed to save camera image: {str(e)}"}), 500
|
||||||
|
|
||||||
|
# 2. Run Vision Safety Check & Cook Planner Concurrently
|
||||||
|
try:
|
||||||
|
safety_task = asyncio.to_thread(safety_checker.check_dish_safety, filepath)
|
||||||
|
planner_task = asyncio.to_thread(
|
||||||
|
microwave_cook_planner.generate_plan,
|
||||||
|
image_path=filepath,
|
||||||
|
height_cm=height_cm,
|
||||||
|
initial_temp_c=initial_temp_c,
|
||||||
|
microwave_wattage=microwave_wattage,
|
||||||
|
defrost_mode=defrost_mode
|
||||||
|
)
|
||||||
|
|
||||||
|
# Execute both concurrently and await results
|
||||||
|
safety_result, cook_plan = await asyncio.gather(safety_task, planner_task)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[Error] Exception during safety check or cook planning: {str(e)}")
|
||||||
|
return jsonify({"error": f"Task execution failed: {str(e)}"}), 500
|
||||||
|
|
||||||
|
# 3. Evaluate Safety Result
|
||||||
|
data["safety_check"] = safety_result
|
||||||
|
if not safety_result.get("is_safe", True):
|
||||||
|
print(f"[Safety Warning] Unsafe dish detected: {safety_result}")
|
||||||
|
return jsonify({
|
||||||
|
"error": "Safety hazard detected in microwave area",
|
||||||
|
"is_safe": False,
|
||||||
|
"warning": safety_result.get("warning_message", "Unsafe materials detected."),
|
||||||
|
"detected_hazards": safety_result.get("detected_hazards", [])
|
||||||
|
}), 200
|
||||||
|
|
||||||
|
# 4. Attach Cooking Plan & Save to MongoDB
|
||||||
|
data["analysis_results"] = cook_plan
|
||||||
|
|
||||||
|
try:
|
||||||
|
cooking_collection.insert_one(data)
|
||||||
|
data.pop("_id", None)
|
||||||
|
except Exception as e:
|
||||||
|
return jsonify({"error": f"Database error: {str(e)}"}), 500
|
||||||
|
|
||||||
|
return jsonify(cook_plan), 201
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
return jsonify({"error": f"Failed to save camera image: {str(e)}"}), 500
|
print(f"[Error] Exception in /cooking-params: {str(e)}")
|
||||||
|
|
||||||
# 2. Run Vision Safety Check & Cook Planner Concurrently
|
|
||||||
try:
|
|
||||||
safety_task = asyncio.to_thread(safety_checker.check_dish_safety, filepath)
|
|
||||||
planner_task = asyncio.to_thread(
|
|
||||||
microwave_cook_planner.generate_plan,
|
|
||||||
image_path=filepath,
|
|
||||||
height_cm=height_cm,
|
|
||||||
initial_temp_c=initial_temp_c,
|
|
||||||
microwave_wattage=microwave_wattage,
|
|
||||||
defrost_mode=defrost_mode
|
|
||||||
)
|
|
||||||
|
|
||||||
# Execute both concurrently and await results
|
|
||||||
safety_result, cook_plan = await asyncio.gather(safety_task, planner_task)
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
return jsonify({"error": f"Task execution failed: {str(e)}"}), 500
|
|
||||||
|
|
||||||
# 3. Evaluate Safety Result
|
|
||||||
data["safety_check"] = safety_result
|
|
||||||
if not safety_result.get("is_safe", True):
|
|
||||||
print(f"[Safety Warning] Unsafe dish detected: {safety_result}")
|
|
||||||
return jsonify({
|
|
||||||
"error": "Safety hazard detected in microwave area",
|
|
||||||
"is_safe": False,
|
|
||||||
"warning": safety_result.get("warning_message", "Unsafe materials detected."),
|
|
||||||
"detected_hazards": safety_result.get("detected_hazards", [])
|
|
||||||
}), 200
|
|
||||||
|
|
||||||
# 4. Attach Cooking Plan & Save to MongoDB
|
|
||||||
data["analysis_results"] = cook_plan
|
|
||||||
|
|
||||||
try:
|
|
||||||
cooking_collection.insert_one(data)
|
|
||||||
data.pop("_id", None)
|
|
||||||
except Exception as e:
|
|
||||||
return jsonify({"error": f"Database error: {str(e)}"}), 500
|
|
||||||
|
|
||||||
return jsonify(cook_plan), 201
|
|
||||||
|
|
||||||
@app.route("/telemetry", methods=["POST"])
|
@app.route("/telemetry", methods=["POST"])
|
||||||
def telemetry():
|
def telemetry():
|
||||||
|
|||||||
Reference in New Issue
Block a user