Enhance image before ai processing

cloud/app.py

@@ -17,6 +17,7 @@ from APIs.mqtt import send_command
 from APIs.webex import WebexManager
 from APIs.shodan import ShodanAuditor
 from APIs.twilio import send_alert_sms
+from cloud.image_enhancer import enhance_image_for_ai
 from microwaveCookPlanner import MicrowaveCookPlanner
 import safety_checker
 from jobs import job_server_cve_audit, job_client_ip_audit, job_request_telemetry
@@ -153,7 +154,7 @@ start_scheduler_once()
 # ---------------------------------------------------------
 # Authentication Middleware
 # ---------------------------------------------------------
-EXEMPT_ROUTES = {'hello_world', 'oauth_callback', 'odata_metadata', 'debug_telemetryrequest', 'trigger_job_manually'}
+EXEMPT_ROUTES = {'hello_world', 'oauth_callback', 'odata_metadata', 'debug_telemetryrequest', 'trigger_job_manually', 'debug_unsafe_dish', 'debug_safe_dish'}
 
 @app.before_request
 def authenticate_request():
@@ -245,12 +246,14 @@ async def cooking_params():
         with open(filepath, "wb") as f:
             f.write(base64.b64decode(camera_image_b64))
         data["camera_image"] = filepath
+        
+        enhanced_filepath = enhance_image_for_ai(filepath)
 
         # Concurrent AI Execution
-        safety_task = asyncio.to_thread(safety_checker.check_dish_safety, filepath)
+        safety_task = asyncio.to_thread(safety_checker.check_dish_safety, enhanced_filepath)
         planner_task = asyncio.to_thread(
             microwave_cook_planner.generate_plan,
-            image_path=filepath,
+            image_path=enhanced_filepath,
             height_cm=height_cm,
             initial_temp_c=initial_temp_c,
             microwave_wattage=microwave_wattage,
@@ -584,6 +587,24 @@ def trigger_job_manually(job_id):
             "status": "error",
             "message": f"Job execution failed: {str(e)}"
         }), 500
+        
+@app.route("/debug/dishsafety/not_safe", methods=["GET"])
+def debug_unsafe_dish():
+    """Debug endpoint to simulate an unsafe dish scenario."""
+    # Simulated unsafe dish data
+    unsafe_dish = "storage/dishPhotos/dish_78827473286c4f67855f3fbd0ccb9bb4.jpg"
+    
+    # Call the cooking_params endpoint logic directly
+    return safety_checker.check_dish_safety(unsafe_dish)
+
+@app.route("/debug/dishsafety/safe", methods=["GET"])
+def debug_safe_dish():
+    """Debug endpoint to simulate a safe dish scenario."""
+    # Simulated safe dish data
+    unsafe_dish = "storage/dishPhotos/dish_972251744d034b32bb5134ff9d0f071c.jpg"
+    
+    # Call the cooking_params endpoint logic directly
+    return safety_checker.check_dish_safety(unsafe_dish)
 
 # ---------------------------------------------------------
 # OData Metadata Definition