131 lines
4.2 KiB
Python
131 lines
4.2 KiB
Python
import os
|
|
import base64
|
|
import uuid
|
|
from flask import Flask, request, jsonify
|
|
from pymongo import MongoClient
|
|
from APIs import generate, EdamamAPI
|
|
import sys
|
|
from microwaveCookPlanner import MicrowaveCookPlanner
|
|
sys.path.insert(0, '..')
|
|
try:
|
|
from shared import config
|
|
except ImportError:
|
|
from ..shared import config
|
|
|
|
app = Flask(__name__)
|
|
|
|
# ---------------------------------------------------------
|
|
# Configuration & Setup
|
|
# ---------------------------------------------------------
|
|
|
|
# Configure MongoDB connection (adjust the URI as needed for your environment)
|
|
MONGO_URI = os.getenv("MONGO_URI", "mongodb://localhost:27017/")
|
|
client = MongoClient(MONGO_URI)
|
|
db = client["microwave_network_db"]
|
|
cooking_collection = db["cooking_parameters"]
|
|
device_network_collection = db["device_network"]
|
|
|
|
# Ensure the camera image storage directory exists when the app starts
|
|
CAMERA_IMAGE_DIR = "storage/dishCameraImages"
|
|
os.makedirs(CAMERA_IMAGE_DIR, exist_ok=True)
|
|
|
|
# ---------------------------------------------------------
|
|
# Classes
|
|
# ---------------------------------------------------------
|
|
microwave_cook_planner = MicrowaveCookPlanner()
|
|
|
|
# ---------------------------------------------------------
|
|
# Routes
|
|
# ---------------------------------------------------------
|
|
|
|
@app.route("/")
|
|
def hello_world():
|
|
gen = generate(prompt="Say Hello, to the user !")
|
|
print(gen)
|
|
return f"<p>{gen}</p>"
|
|
|
|
|
|
@app.route("/cooking-params", methods=["POST"])
|
|
def cooking_params():
|
|
data = request.get_json()
|
|
|
|
if not data:
|
|
return jsonify({"error": "Invalid or missing JSON payload"}), 400
|
|
|
|
# Extract user or device parameters (with fallback defaults)
|
|
height_cm = float(data.get("dish_height", 4.0))
|
|
initial_temp_c = float(data.get("ir_initial_temp", 20.0)) # e.g., 4.0 for fridge, -18.0 for freezer
|
|
microwave_wattage = int(data.get("microwave_wattage", 900)) # e.g., 900W
|
|
defrost_mode = bool(data.get("defrost_mode", False)) # True for defrost, False for cook/reheat
|
|
print("Parsed parameters - Height (cm):", height_cm, "Initial Temp (C):", initial_temp_c, "Microwave Wattage:", microwave_wattage, "Defrost Mode:", defrost_mode)
|
|
|
|
# 1. Handle the Camera Image
|
|
camera_image_b64 = data.get("camera_image")
|
|
filepath = None
|
|
|
|
if camera_image_b64:
|
|
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:
|
|
return jsonify({"error": f"Failed to save camera image: {str(e)}"}), 500
|
|
else:
|
|
return jsonify({"error": "Missing required field 'camera_image'"}), 400
|
|
|
|
# 2. Run the Cook Planning Engine
|
|
try:
|
|
cook_plan = 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
|
|
)
|
|
except Exception as e:
|
|
return jsonify({"error": f"Failed to compute cooking plan: {str(e)}"}), 500
|
|
|
|
# 3. Attach cooking parameters to database record
|
|
data["analysis_results"] = cook_plan
|
|
|
|
# 4. Save to MongoDB
|
|
try:
|
|
cooking_collection.insert_one(data)
|
|
data.pop("_id", None)
|
|
except Exception as e:
|
|
return jsonify({"error": f"Database error: {str(e)}"}), 500
|
|
|
|
# 5. Return complete output
|
|
return jsonify(cook_plan), 201
|
|
|
|
@app.route("/device-network", methods=["POST"])
|
|
def device_network():
|
|
data = request.get_json()
|
|
|
|
if not data:
|
|
return jsonify({"error": "Invalid or missing JSON payload"}), 400
|
|
|
|
|
|
# 2. Save to MongoDB
|
|
try:
|
|
# Insert the dictionary directly into Mongo (it will retain your exact JSON keys)
|
|
device_network_collection.insert_one(data)
|
|
|
|
return "", 200
|
|
|
|
except Exception as e:
|
|
return jsonify({"error": f"Database error: {str(e)}"}), 500
|
|
|
|
@app.route("/debug", methods=["GET"])
|
|
def debug():
|
|
image_path = "microwaveDish.jpg"
|
|
edamam = EdamamAPI()
|
|
return edamam.analyze_dish_image(image_path)
|
|
|
|
if __name__ == "__main__":
|
|
app.run(debug=config.DEBUG) |