图片预览更新,验证码接口更新 #20

Merged
ppy4sjqvf merged 1 commits from ybw-branch into develop 1 month ago

@ -54,7 +54,7 @@ def register():
username = data.get('username')
password = data.get('password')
email = data.get('email')
# code = data.get('code')
code = data.get('code')
# 验证输入
if not username or not password or not email:
return jsonify({'error': '用户名、密码和邮箱不能为空'}), 400
@ -168,7 +168,7 @@ def change_email(current_user_id):
user = User.query.filter_by(current_user_id)
data = request.get_json()
new_email = data.get('new_email')
# code = data.get('code')
code = data.get('code')
if not new_email:
return jsonify({'error': '新邮箱不能为空'}), 400

@ -534,8 +534,6 @@ class ImageService:
def _get_evaluate_preview(task):
"""获取评估任务的预览图片"""
images = {
'original_generate': [],
'perturbed_generate': [],
'report': []
}
@ -547,17 +545,5 @@ class ImageService:
image_types_id=report_type.image_types_id
).all()
images['report'] = [ImageService.image_to_base64(img) for img in reports]
# 获取关联的微调任务生成图
if task.evaluation and task.evaluation.finetune_task_id:
finetune_task_id = task.evaluation.finetune_task_id
for type_code in ['original_generate', 'perturbed_generate']:
img_type = ImageType.query.filter_by(image_code=type_code).first()
if img_type:
generated = Image.query.filter_by(
task_id=finetune_task_id,
image_types_id=img_type.image_types_id
).all()
images[type_code] = [ImageService.image_to_base64(img) for img in generated]
return images
Loading…
Cancel
Save