server.py

from flask import Flask, render_template, jsonify, request
import serial
import serial.tools.list_ports
import time
import json
import os

app = Flask(__name__)

BAUD_RATE = 115200
SERIAL_TIMEOUT = 5.0

serial_conn = None
homed = False
position = {"x": 0, "y": 0, "z": 0}
speed = 1000

DATA_DIR = os.path.dirname(os.path.abspath(__file__))
SEQ_FILE = os.path.join(DATA_DIR, "sequences.json")


def connect_serial():
    global serial_conn
    if serial_conn and serial_conn.is_open:
        return True
    ports = list(serial.tools.list_ports.comports())
    for p in ports:
        desc = (p.description or "").upper()
        name = (p.name or "").upper()
        if "USB" in desc or "ACM" in name:
            try:
                serial_conn = serial.Serial(p.device, BAUD_RATE, timeout=SERIAL_TIMEOUT)
                return True
            except serial.SerialException:
                pass
    if ports:
        try:
            serial_conn = serial.Serial(ports[0].device, BAUD_RATE, timeout=SERIAL_TIMEOUT)
            return True
        except serial.SerialException:
            pass
    return False


def disconnect_serial():
    global serial_conn
    if serial_conn and serial_conn.is_open:
        serial_conn.close()
    serial_conn = None


def send_gcode(cmd):
    global serial_conn
    if not serial_conn or not serial_conn.is_open:
        if not connect_serial():
            return False
    try:
        serial_conn.write((cmd + "\n").encode())
        serial_conn.flush()
        return True
    except:
        return False


@app.route("/")
def index():
    return render_template("index.html")


@app.route("/api/connect", methods=["POST"])
def api_connect():
    if connect_serial():
        return jsonify({"status": "ok", "port": serial_conn.name})
    return jsonify({"status": "error", "message": "No serial port found"}), 400


@app.route("/api/disconnect", methods=["POST"])
def api_disconnect():
    disconnect_serial()
    return jsonify({"status": "ok"})


@app.route("/api/home", methods=["POST"])
def api_home():
    global homed, position
    if send_gcode("G28"):
        homed = True
        position = {"x": 700, "y": 0, "z": 0}
        return jsonify({"status": "ok"})
    return jsonify({"status": "error"}), 400


@app.route("/api/move", methods=["POST"])
def api_move():
    global position
    req = request.get_json()
    x = req.get("x")
    y = req.get("y")
    z = req.get("z")
    spd = req.get("speed", speed)

    cmd = "G0"
    if x is not None:
        cmd += " X{}".format(x - 700)
        position["x"] = x
    if y is not None:
        cmd += " Y{}".format(-y)
        position["y"] = y
    if z is not None:
        cmd += " Z{}".format(-z)
        position["z"] = z
    cmd += " F{}".format(spd)

    if send_gcode(cmd):
        return jsonify({"status": "ok"})
    return jsonify({"status": "error"}), 400


@app.route("/api/gcode", methods=["POST"])
def api_gcode():
    req = request.get_json()
    cmd = req.get("cmd", "").strip()
    if not cmd:
        return jsonify({"status": "error"}), 400
    if send_gcode(cmd):
        resp = ""
        if serial_conn and serial_conn.is_open:
            time.sleep(0.3)
            if serial_conn.in_waiting:
                resp = serial_conn.read(serial_conn.in_waiting).decode(errors="ignore")
        return jsonify({"status": "ok", "response": resp.strip()})
    return jsonify({"status": "error"}), 400


if __name__ == "__main__":
    app.run(host="0.0.0.0", port=5000, debug=True)