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)