|
from flask import Flask, request, jsonify, send_from_directory |
|
from flask_cors import CORS |
|
import os |
|
|
|
from puzzle_dataset import get_puzzle_by_index |
|
from solver import solve_puzzle |
|
|
|
BASE_DIR = os.path.dirname(os.path.abspath(__file__)) |
|
example_path = os.path.join(BASE_DIR, "Example.txt") |
|
with open(example_path, "r", encoding="utf-8") as f: |
|
DEFAULT_SYS_CONTENT = f.read() |
|
sat_cnt_path = os.path.join(BASE_DIR, "Sat_cnt.txt") |
|
with open(sat_cnt_path, "r", encoding="utf-8") as f: |
|
SAT_CNT_CONTENT = f.read() |
|
|
|
|
|
app = Flask(__name__, static_folder='static', static_url_path='') |
|
CORS(app) |
|
|
|
@app.route('/') |
|
def index(): |
|
|
|
return send_from_directory(app.static_folder, 'index.html') |
|
|
|
@app.route("/get_puzzle", methods=["GET"]) |
|
def get_puzzle(): |
|
idx_str = request.args.get("index", "0") |
|
try: |
|
idx = int(idx_str) |
|
except ValueError: |
|
return jsonify({"success": False, "error": "Index must be an integer"}), 400 |
|
|
|
puzzle, solution = get_puzzle_by_index(idx) |
|
if puzzle is None or solution is None: |
|
return jsonify({"success": False, "error": "Invalid puzzle index"}), 404 |
|
|
|
return jsonify({ |
|
"success": True, |
|
"index": idx, |
|
"puzzle": puzzle, |
|
"expected_solution": solution |
|
}) |
|
|
|
@app.route("/solve", methods=["POST"]) |
|
def solve(): |
|
data = request.get_json() |
|
puzzle_index = data.get("index") |
|
puzzle_text = data.get("puzzle") |
|
expected_solution = data.get("expected_solution") |
|
sys_content = data.get("sys_content", DEFAULT_SYS_CONTENT) |
|
|
|
if puzzle_index is None or puzzle_text is None or expected_solution is None: |
|
return jsonify({"success": False, "error": "Missing puzzle data"}), 400 |
|
|
|
result = solve_puzzle(puzzle_index, puzzle_text, expected_solution, sys_content, SAT_CNT_CONTENT) |
|
return jsonify({"success": True, "result": result}) |
|
|
|
@app.route("/default_sys_content", methods=["GET"]) |
|
def get_default_sys_content(): |
|
return jsonify({"success": True, "sysContent": DEFAULT_SYS_CONTENT}) |
|
|
|
if __name__ == "__main__": |
|
app.run(host="0.0.0.0", port=7860, debug=False) |
|
|