AadityaPawarx1's picture
Upload 3 files
a154891 verified
from flask import Flask, render_template, request, jsonify
import os
import secrets
from werkzeug.utils import secure_filename
import solver
app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = 'uploads'
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024 # 16MB max
# Ensure upload dir exists
os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
@app.route('/')
def index():
return render_template('index.html')
@app.route('/api/solve', methods=['POST'])
def solve():
if 'image' not in request.files:
return jsonify({"error": "No image uploaded"}), 400
file = request.files['image']
clues = request.form.get('clues', '')
if file.filename == '':
return jsonify({"error": "No image selected"}), 400
if file:
filename = secure_filename(f"{secrets.token_hex(8)}_{file.filename}")
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
file.save(filepath)
try:
# Call the solver
result = solver.solve_challenge(filepath, clues)
# Clean up uploaded file
try:
os.remove(filepath)
except:
pass
return jsonify(result)
except Exception as e:
return jsonify({"error": str(e)}), 500
if __name__ == '__main__':
port = int(os.environ.get('PORT', 7860))
app.run(debug=False, host='0.0.0.0', port=port)