Spaces:
Sleeping
Sleeping
File size: 3,850 Bytes
0387db2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 |
import os
import json
import uuid
from flask import Flask, render_template, request, jsonify, send_file, send_from_directory
from werkzeug.utils import secure_filename
app = Flask(__name__)
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024 # 16MB max limit
app.config['UPLOAD_FOLDER'] = os.path.join(os.path.dirname(__file__), 'data', 'uploads')
# Ensure data directories exist
DATA_DIR = os.path.join(os.path.dirname(__file__), 'data')
os.makedirs(DATA_DIR, exist_ok=True)
os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif', 'json'}
def allowed_file(filename):
return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
@app.errorhandler(404)
def not_found(e):
if request.path.startswith('/api/'):
return jsonify({'error': 'Not found'}), 404
return render_template('index.html'), 404
@app.errorhandler(500)
def server_error(e):
if request.path.startswith('/api/'):
return jsonify({'error': 'Internal Server Error'}), 500
return "<h1>500 Internal Server Error</h1><p>Something went wrong.</p>", 500
@app.route('/')
def index():
return render_template('index.html')
@app.route('/uploads/<filename>')
def uploaded_file(filename):
return send_from_directory(app.config['UPLOAD_FOLDER'], filename)
@app.route('/api/upload', methods=['POST'])
def upload_file():
if 'file' not in request.files:
return jsonify({'error': 'No file part'}), 400
file = request.files['file']
if file.filename == '':
return jsonify({'error': 'No selected file'}), 400
if file and allowed_file(file.filename):
filename = secure_filename(file.filename)
# Unique filename to avoid overwrites
unique_filename = f"{uuid.uuid4()}_{filename}"
file.save(os.path.join(app.config['UPLOAD_FOLDER'], unique_filename))
return jsonify({'success': True, 'url': f'/uploads/{unique_filename}', 'filename': unique_filename})
return jsonify({'error': 'File type not allowed'}), 400
@app.route('/api/projects', methods=['GET'])
def list_projects():
projects = []
for filename in os.listdir(DATA_DIR):
if filename.endswith('.json'):
try:
with open(os.path.join(DATA_DIR, filename), 'r', encoding='utf-8') as f:
data = json.load(f)
projects.append({
'id': filename.replace('.json', ''),
'name': data.get('name', 'Untitled'),
'updated_at': data.get('updated_at', '')
})
except:
pass
return jsonify(projects)
@app.route('/api/projects', methods=['POST'])
def save_project():
data = request.json
project_id = data.get('id')
if not project_id:
project_id = str(uuid.uuid4())
data['id'] = project_id
file_path = os.path.join(DATA_DIR, f"{project_id}.json")
with open(file_path, 'w', encoding='utf-8') as f:
json.dump(data, f, ensure_ascii=False, indent=2)
return jsonify({'success': True, 'id': project_id})
@app.route('/api/projects/<project_id>', methods=['GET'])
def get_project(project_id):
file_path = os.path.join(DATA_DIR, f"{project_id}.json")
if os.path.exists(file_path):
with open(file_path, 'r', encoding='utf-8') as f:
return jsonify(json.load(f))
return jsonify({'error': 'Not found'}), 404
@app.route('/api/projects/<project_id>', methods=['DELETE'])
def delete_project(project_id):
file_path = os.path.join(DATA_DIR, f"{project_id}.json")
if os.path.exists(file_path):
os.remove(file_path)
return jsonify({'success': True})
return jsonify({'error': 'Not found'}), 404
if __name__ == '__main__':
app.run(host='0.0.0.0', port=7860, debug=True)
|