| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670 | 
							- #
 - #  Copyright 2024 The InfiniFlow Authors. All Rights Reserved.
 - #
 - #  Licensed under the Apache License, Version 2.0 (the "License");
 - #  you may not use this file except in compliance with the License.
 - #  You may obtain a copy of the License at
 - #
 - #      http://www.apache.org/licenses/LICENSE-2.0
 - #
 - #  Unless required by applicable law or agreed to in writing, software
 - #  distributed under the License is distributed on an "AS IS" BASIS,
 - #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 - #  See the License for the specific language governing permissions and
 - #  limitations under the License
 - #
 - import json
 - import os.path
 - import pathlib
 - import re
 - from pathlib import Path
 - 
 - import flask
 - from flask import request
 - from flask_login import current_user, login_required
 - 
 - from api import settings
 - from api.constants import FILE_NAME_LEN_LIMIT, IMG_BASE64_PREFIX
 - from api.db import VALID_FILE_TYPES, VALID_TASK_STATUS, FileSource, FileType, ParserType, TaskStatus
 - from api.db.db_models import File, Task
 - from api.db.services import duplicate_name
 - from api.db.services.document_service import DocumentService, doc_upload_and_parse
 - from api.db.services.file2document_service import File2DocumentService
 - from api.db.services.file_service import FileService
 - from api.db.services.knowledgebase_service import KnowledgebaseService
 - from api.db.services.task_service import TaskService, queue_tasks
 - from api.db.services.user_service import UserTenantService
 - from api.utils import get_uuid
 - from api.utils.api_utils import (
 -     get_data_error_result,
 -     get_json_result,
 -     server_error_response,
 -     validate_request,
 - )
 - from api.utils.file_utils import filename_type, get_project_base_directory, thumbnail
 - from api.utils.web_utils import html2pdf, is_valid_url
 - from deepdoc.parser.html_parser import RAGFlowHtmlParser
 - from rag.nlp import search
 - from rag.utils.storage_factory import STORAGE_IMPL
 - 
 - 
 - @manager.route("/upload", methods=["POST"])  # noqa: F821
 - @login_required
 - @validate_request("kb_id")
 - def upload():
 -     kb_id = request.form.get("kb_id")
 -     if not kb_id:
 -         return get_json_result(data=False, message='Lack of "KB ID"', code=settings.RetCode.ARGUMENT_ERROR)
 -     if "file" not in request.files:
 -         return get_json_result(data=False, message="No file part!", code=settings.RetCode.ARGUMENT_ERROR)
 - 
 -     file_objs = request.files.getlist("file")
 -     for file_obj in file_objs:
 -         if file_obj.filename == "":
 -             return get_json_result(data=False, message="No file selected!", code=settings.RetCode.ARGUMENT_ERROR)
 -         if len(file_obj.filename.encode("utf-8")) > FILE_NAME_LEN_LIMIT:
 -             return get_json_result(data=False, message=f"File name must be {FILE_NAME_LEN_LIMIT} bytes or less.", code=settings.RetCode.ARGUMENT_ERROR)
 - 
 -     e, kb = KnowledgebaseService.get_by_id(kb_id)
 -     if not e:
 -         raise LookupError("Can't find this knowledgebase!")
 -     err, files = FileService.upload_document(kb, file_objs, current_user.id)
 - 
 -     if err:
 -         return get_json_result(data=files, message="\n".join(err), code=settings.RetCode.SERVER_ERROR)
 - 
 -     if not files:
 -         return get_json_result(data=files, message="There seems to be an issue with your file format. Please verify it is correct and not corrupted.", code=settings.RetCode.DATA_ERROR)
 -     files = [f[0] for f in files]  # remove the blob
 - 
 -     return get_json_result(data=files)
 - 
 - 
 - @manager.route("/web_crawl", methods=["POST"])  # noqa: F821
 - @login_required
 - @validate_request("kb_id", "name", "url")
 - def web_crawl():
 -     kb_id = request.form.get("kb_id")
 -     if not kb_id:
 -         return get_json_result(data=False, message='Lack of "KB ID"', code=settings.RetCode.ARGUMENT_ERROR)
 -     name = request.form.get("name")
 -     url = request.form.get("url")
 -     if not is_valid_url(url):
 -         return get_json_result(data=False, message="The URL format is invalid", code=settings.RetCode.ARGUMENT_ERROR)
 -     e, kb = KnowledgebaseService.get_by_id(kb_id)
 -     if not e:
 -         raise LookupError("Can't find this knowledgebase!")
 - 
 -     blob = html2pdf(url)
 -     if not blob:
 -         return server_error_response(ValueError("Download failure."))
 - 
 -     root_folder = FileService.get_root_folder(current_user.id)
 -     pf_id = root_folder["id"]
 -     FileService.init_knowledgebase_docs(pf_id, current_user.id)
 -     kb_root_folder = FileService.get_kb_folder(current_user.id)
 -     kb_folder = FileService.new_a_file_from_kb(kb.tenant_id, kb.name, kb_root_folder["id"])
 - 
 -     try:
 -         filename = duplicate_name(DocumentService.query, name=name + ".pdf", kb_id=kb.id)
 -         filetype = filename_type(filename)
 -         if filetype == FileType.OTHER.value:
 -             raise RuntimeError("This type of file has not been supported yet!")
 - 
 -         location = filename
 -         while STORAGE_IMPL.obj_exist(kb_id, location):
 -             location += "_"
 -         STORAGE_IMPL.put(kb_id, location, blob)
 -         doc = {
 -             "id": get_uuid(),
 -             "kb_id": kb.id,
 -             "parser_id": kb.parser_id,
 -             "parser_config": kb.parser_config,
 -             "created_by": current_user.id,
 -             "type": filetype,
 -             "name": filename,
 -             "location": location,
 -             "size": len(blob),
 -             "thumbnail": thumbnail(filename, blob),
 -             "suffix": Path(filename).suffix.lstrip("."),
 -         }
 -         if doc["type"] == FileType.VISUAL:
 -             doc["parser_id"] = ParserType.PICTURE.value
 -         if doc["type"] == FileType.AURAL:
 -             doc["parser_id"] = ParserType.AUDIO.value
 -         if re.search(r"\.(ppt|pptx|pages)$", filename):
 -             doc["parser_id"] = ParserType.PRESENTATION.value
 -         if re.search(r"\.(eml)$", filename):
 -             doc["parser_id"] = ParserType.EMAIL.value
 -         DocumentService.insert(doc)
 -         FileService.add_file_from_kb(doc, kb_folder["id"], kb.tenant_id)
 -     except Exception as e:
 -         return server_error_response(e)
 -     return get_json_result(data=True)
 - 
 - 
 - @manager.route("/create", methods=["POST"])  # noqa: F821
 - @login_required
 - @validate_request("name", "kb_id")
 - def create():
 -     req = request.json
 -     kb_id = req["kb_id"]
 -     if not kb_id:
 -         return get_json_result(data=False, message='Lack of "KB ID"', code=settings.RetCode.ARGUMENT_ERROR)
 -     if len(req["name"].encode("utf-8")) > FILE_NAME_LEN_LIMIT:
 -         return get_json_result(data=False, message=f"File name must be {FILE_NAME_LEN_LIMIT} bytes or less.", code=settings.RetCode.ARGUMENT_ERROR)
 - 
 -     if req["name"].strip() == "":
 -         return get_json_result(data=False, message="File name can't be empty.", code=settings.RetCode.ARGUMENT_ERROR)
 -     req["name"] = req["name"].strip()
 - 
 -     try:
 -         e, kb = KnowledgebaseService.get_by_id(kb_id)
 -         if not e:
 -             return get_data_error_result(message="Can't find this knowledgebase!")
 - 
 -         if DocumentService.query(name=req["name"], kb_id=kb_id):
 -             return get_data_error_result(message="Duplicated document name in the same knowledgebase.")
 - 
 -         doc = DocumentService.insert(
 -             {
 -                 "id": get_uuid(),
 -                 "kb_id": kb.id,
 -                 "parser_id": kb.parser_id,
 -                 "parser_config": kb.parser_config,
 -                 "created_by": current_user.id,
 -                 "type": FileType.VIRTUAL,
 -                 "name": req["name"],
 -                 "suffix": Path(req["name"]).suffix.lstrip("."),
 -                 "location": "",
 -                 "size": 0,
 -             }
 -         )
 -         return get_json_result(data=doc.to_json())
 -     except Exception as e:
 -         return server_error_response(e)
 - 
 - 
 - @manager.route("/list", methods=["POST"])  # noqa: F821
 - @login_required
 - def list_docs():
 -     kb_id = request.args.get("kb_id")
 -     if not kb_id:
 -         return get_json_result(data=False, message='Lack of "KB ID"', code=settings.RetCode.ARGUMENT_ERROR)
 -     tenants = UserTenantService.query(user_id=current_user.id)
 -     for tenant in tenants:
 -         if KnowledgebaseService.query(tenant_id=tenant.tenant_id, id=kb_id):
 -             break
 -     else:
 -         return get_json_result(data=False, message="Only owner of knowledgebase authorized for this operation.", code=settings.RetCode.OPERATING_ERROR)
 -     keywords = request.args.get("keywords", "")
 - 
 -     page_number = int(request.args.get("page", 0))
 -     items_per_page = int(request.args.get("page_size", 0))
 -     orderby = request.args.get("orderby", "create_time")
 -     if request.args.get("desc", "true").lower() == "false":
 -         desc = False
 -     else:
 -         desc = True
 - 
 -     req = request.get_json()
 - 
 -     run_status = req.get("run_status", [])
 -     if run_status:
 -         invalid_status = {s for s in run_status if s not in VALID_TASK_STATUS}
 -         if invalid_status:
 -             return get_data_error_result(message=f"Invalid filter run status conditions: {', '.join(invalid_status)}")
 - 
 -     types = req.get("types", [])
 -     if types:
 -         invalid_types = {t for t in types if t not in VALID_FILE_TYPES}
 -         if invalid_types:
 -             return get_data_error_result(message=f"Invalid filter conditions: {', '.join(invalid_types)} type{'s' if len(invalid_types) > 1 else ''}")
 - 
 -     suffix = req.get("suffix", [])
 - 
 -     try:
 -         docs, tol = DocumentService.get_by_kb_id(kb_id, page_number, items_per_page, orderby, desc, keywords, run_status, types, suffix)
 - 
 -         for doc_item in docs:
 -             if doc_item["thumbnail"] and not doc_item["thumbnail"].startswith(IMG_BASE64_PREFIX):
 -                 doc_item["thumbnail"] = f"/v1/document/image/{kb_id}-{doc_item['thumbnail']}"
 - 
 -         return get_json_result(data={"total": tol, "docs": docs})
 -     except Exception as e:
 -         return server_error_response(e)
 - 
 - 
 - @manager.route("/filter", methods=["POST"])  # noqa: F821
 - @login_required
 - def get_filter():
 -     req = request.get_json()
 - 
 -     kb_id = req.get("kb_id")
 -     if not kb_id:
 -         return get_json_result(data=False, message='Lack of "KB ID"', code=settings.RetCode.ARGUMENT_ERROR)
 -     tenants = UserTenantService.query(user_id=current_user.id)
 -     for tenant in tenants:
 -         if KnowledgebaseService.query(tenant_id=tenant.tenant_id, id=kb_id):
 -             break
 -     else:
 -         return get_json_result(data=False, message="Only owner of knowledgebase authorized for this operation.", code=settings.RetCode.OPERATING_ERROR)
 - 
 -     keywords = req.get("keywords", "")
 - 
 -     suffix = req.get("suffix", [])
 - 
 -     run_status = req.get("run_status", [])
 -     if run_status:
 -         invalid_status = {s for s in run_status if s not in VALID_TASK_STATUS}
 -         if invalid_status:
 -             return get_data_error_result(message=f"Invalid filter run status conditions: {', '.join(invalid_status)}")
 - 
 -     types = req.get("types", [])
 -     if types:
 -         invalid_types = {t for t in types if t not in VALID_FILE_TYPES}
 -         if invalid_types:
 -             return get_data_error_result(message=f"Invalid filter conditions: {', '.join(invalid_types)} type{'s' if len(invalid_types) > 1 else ''}")
 - 
 -     try:
 -         filter, total = DocumentService.get_filter_by_kb_id(kb_id, keywords, run_status, types, suffix)
 -         return get_json_result(data={"total": total, "filter": filter})
 -     except Exception as e:
 -         return server_error_response(e)
 - 
 - 
 - @manager.route("/infos", methods=["POST"])  # noqa: F821
 - @login_required
 - def docinfos():
 -     req = request.json
 -     doc_ids = req["doc_ids"]
 -     for doc_id in doc_ids:
 -         if not DocumentService.accessible(doc_id, current_user.id):
 -             return get_json_result(data=False, message="No authorization.", code=settings.RetCode.AUTHENTICATION_ERROR)
 -     docs = DocumentService.get_by_ids(doc_ids)
 -     return get_json_result(data=list(docs.dicts()))
 - 
 - 
 - @manager.route("/thumbnails", methods=["GET"])  # noqa: F821
 - # @login_required
 - def thumbnails():
 -     doc_ids = request.args.get("doc_ids").split(",")
 -     if not doc_ids:
 -         return get_json_result(data=False, message='Lack of "Document ID"', code=settings.RetCode.ARGUMENT_ERROR)
 - 
 -     try:
 -         docs = DocumentService.get_thumbnails(doc_ids)
 - 
 -         for doc_item in docs:
 -             if doc_item["thumbnail"] and not doc_item["thumbnail"].startswith(IMG_BASE64_PREFIX):
 -                 doc_item["thumbnail"] = f"/v1/document/image/{doc_item['kb_id']}-{doc_item['thumbnail']}"
 - 
 -         return get_json_result(data={d["id"]: d["thumbnail"] for d in docs})
 -     except Exception as e:
 -         return server_error_response(e)
 - 
 - 
 - @manager.route("/change_status", methods=["POST"])  # noqa: F821
 - @login_required
 - @validate_request("doc_ids", "status")
 - def change_status():
 -     req = request.get_json()
 -     doc_ids = req.get("doc_ids", [])
 -     status = str(req.get("status", ""))
 - 
 -     if status not in ["0", "1"]:
 -         return get_json_result(data=False, message='"Status" must be either 0 or 1!', code=settings.RetCode.ARGUMENT_ERROR)
 - 
 -     result = {}
 -     for doc_id in doc_ids:
 -         if not DocumentService.accessible(doc_id, current_user.id):
 -             result[doc_id] = {"error": "No authorization."}
 -             continue
 - 
 -         try:
 -             e, doc = DocumentService.get_by_id(doc_id)
 -             if not e:
 -                 result[doc_id] = {"error": "No authorization."}
 -                 continue
 -             e, kb = KnowledgebaseService.get_by_id(doc.kb_id)
 -             if not e:
 -                 result[doc_id] = {"error": "Can't find this knowledgebase!"}
 -                 continue
 -             if not DocumentService.update_by_id(doc_id, {"status": str(status)}):
 -                 result[doc_id] = {"error": "Database error (Document update)!"}
 -                 continue
 - 
 -             status_int = int(status)
 -             if not settings.docStoreConn.update({"doc_id": doc_id}, {"available_int": status_int}, search.index_name(kb.tenant_id), doc.kb_id):
 -                 result[doc_id] = {"error": "Database error (docStore update)!"}
 -             result[doc_id] = {"status": status}
 -         except Exception as e:
 -             result[doc_id] = {"error": f"Internal server error: {str(e)}"}
 - 
 -     return get_json_result(data=result)
 - 
 - 
 - @manager.route("/rm", methods=["POST"])  # noqa: F821
 - @login_required
 - @validate_request("doc_id")
 - def rm():
 -     req = request.json
 -     doc_ids = req["doc_id"]
 -     if isinstance(doc_ids, str):
 -         doc_ids = [doc_ids]
 - 
 -     for doc_id in doc_ids:
 -         if not DocumentService.accessible4deletion(doc_id, current_user.id):
 -             return get_json_result(data=False, message="No authorization.", code=settings.RetCode.AUTHENTICATION_ERROR)
 - 
 -     root_folder = FileService.get_root_folder(current_user.id)
 -     pf_id = root_folder["id"]
 -     FileService.init_knowledgebase_docs(pf_id, current_user.id)
 -     errors = ""
 -     kb_table_num_map = {}
 -     for doc_id in doc_ids:
 -         try:
 -             e, doc = DocumentService.get_by_id(doc_id)
 -             if not e:
 -                 return get_data_error_result(message="Document not found!")
 -             tenant_id = DocumentService.get_tenant_id(doc_id)
 -             if not tenant_id:
 -                 return get_data_error_result(message="Tenant not found!")
 - 
 -             b, n = File2DocumentService.get_storage_address(doc_id=doc_id)
 - 
 -             TaskService.filter_delete([Task.doc_id == doc_id])
 -             if not DocumentService.remove_document(doc, tenant_id):
 -                 return get_data_error_result(message="Database error (Document removal)!")
 - 
 -             f2d = File2DocumentService.get_by_document_id(doc_id)
 -             deleted_file_count = 0
 -             if f2d:
 -                 deleted_file_count = FileService.filter_delete([File.source_type == FileSource.KNOWLEDGEBASE, File.id == f2d[0].file_id])
 -             File2DocumentService.delete_by_document_id(doc_id)
 -             if deleted_file_count > 0:
 -                 STORAGE_IMPL.rm(b, n)
 - 
 -             doc_parser = doc.parser_id
 -             if doc_parser == ParserType.TABLE:
 -                 kb_id = doc.kb_id
 -                 if kb_id not in kb_table_num_map:
 -                     counts = DocumentService.count_by_kb_id(kb_id=kb_id, keywords="", run_status=[TaskStatus.DONE], types=[])
 -                     kb_table_num_map[kb_id] = counts
 -                 kb_table_num_map[kb_id] -= 1
 -                 if kb_table_num_map[kb_id] <= 0:
 -                     KnowledgebaseService.delete_field_map(kb_id)
 -         except Exception as e:
 -             errors += str(e)
 - 
 -     if errors:
 -         return get_json_result(data=False, message=errors, code=settings.RetCode.SERVER_ERROR)
 - 
 -     return get_json_result(data=True)
 - 
 - 
 - @manager.route("/run", methods=["POST"])  # noqa: F821
 - @login_required
 - @validate_request("doc_ids", "run")
 - def run():
 -     req = request.json
 -     for doc_id in req["doc_ids"]:
 -         if not DocumentService.accessible(doc_id, current_user.id):
 -             return get_json_result(data=False, message="No authorization.", code=settings.RetCode.AUTHENTICATION_ERROR)
 -     try:
 -         kb_table_num_map = {}
 -         for id in req["doc_ids"]:
 -             info = {"run": str(req["run"]), "progress": 0}
 -             if str(req["run"]) == TaskStatus.RUNNING.value and req.get("delete", False):
 -                 info["progress_msg"] = ""
 -                 info["chunk_num"] = 0
 -                 info["token_num"] = 0
 - 
 -             e, doc = DocumentService.get_by_id(id)
 -             if not e:
 -                 return get_data_error_result(message="Document not found!")
 -             if doc.run == TaskStatus.DONE.value:
 -                 DocumentService.clear_chunk_num_when_rerun(doc.id)
 - 
 -             DocumentService.update_by_id(id, info)
 -             tenant_id = DocumentService.get_tenant_id(id)
 -             if not tenant_id:
 -                 return get_data_error_result(message="Tenant not found!")
 -             e, doc = DocumentService.get_by_id(id)
 -             if not e:
 -                 return get_data_error_result(message="Document not found!")
 -             if req.get("delete", False):
 -                 TaskService.filter_delete([Task.doc_id == id])
 -                 if settings.docStoreConn.indexExist(search.index_name(tenant_id), doc.kb_id):
 -                     settings.docStoreConn.delete({"doc_id": id}, search.index_name(tenant_id), doc.kb_id)
 - 
 -             if str(req["run"]) == TaskStatus.RUNNING.value:
 -                 e, doc = DocumentService.get_by_id(id)
 -                 doc = doc.to_dict()
 -                 doc["tenant_id"] = tenant_id
 - 
 -                 doc_parser = doc.get("parser_id", ParserType.NAIVE)
 -                 if doc_parser == ParserType.TABLE:
 -                     kb_id = doc.get("kb_id")
 -                     if not kb_id:
 -                         continue
 -                     if kb_id not in kb_table_num_map:
 -                         count = DocumentService.count_by_kb_id(kb_id=kb_id, keywords="", run_status=[TaskStatus.DONE], types=[])
 -                         kb_table_num_map[kb_id] = count
 -                         if kb_table_num_map[kb_id] <= 0:
 -                             KnowledgebaseService.delete_field_map(kb_id)
 -                 bucket, name = File2DocumentService.get_storage_address(doc_id=doc["id"])
 -                 queue_tasks(doc, bucket, name, 0)
 - 
 -         return get_json_result(data=True)
 -     except Exception as e:
 -         return server_error_response(e)
 - 
 - 
 - @manager.route("/rename", methods=["POST"])  # noqa: F821
 - @login_required
 - @validate_request("doc_id", "name")
 - def rename():
 -     req = request.json
 -     if not DocumentService.accessible(req["doc_id"], current_user.id):
 -         return get_json_result(data=False, message="No authorization.", code=settings.RetCode.AUTHENTICATION_ERROR)
 -     try:
 -         e, doc = DocumentService.get_by_id(req["doc_id"])
 -         if not e:
 -             return get_data_error_result(message="Document not found!")
 -         if pathlib.Path(req["name"].lower()).suffix != pathlib.Path(doc.name.lower()).suffix:
 -             return get_json_result(data=False, message="The extension of file can't be changed", code=settings.RetCode.ARGUMENT_ERROR)
 -         if len(req["name"].encode("utf-8")) > FILE_NAME_LEN_LIMIT:
 -             return get_json_result(data=False, message=f"File name must be {FILE_NAME_LEN_LIMIT} bytes or less.", code=settings.RetCode.ARGUMENT_ERROR)
 - 
 -         for d in DocumentService.query(name=req["name"], kb_id=doc.kb_id):
 -             if d.name == req["name"]:
 -                 return get_data_error_result(message="Duplicated document name in the same knowledgebase.")
 - 
 -         if not DocumentService.update_by_id(req["doc_id"], {"name": req["name"]}):
 -             return get_data_error_result(message="Database error (Document rename)!")
 - 
 -         informs = File2DocumentService.get_by_document_id(req["doc_id"])
 -         if informs:
 -             e, file = FileService.get_by_id(informs[0].file_id)
 -             FileService.update_by_id(file.id, {"name": req["name"]})
 - 
 -         return get_json_result(data=True)
 -     except Exception as e:
 -         return server_error_response(e)
 - 
 - 
 - @manager.route("/get/<doc_id>", methods=["GET"])  # noqa: F821
 - # @login_required
 - def get(doc_id):
 -     try:
 -         e, doc = DocumentService.get_by_id(doc_id)
 -         if not e:
 -             return get_data_error_result(message="Document not found!")
 - 
 -         b, n = File2DocumentService.get_storage_address(doc_id=doc_id)
 -         response = flask.make_response(STORAGE_IMPL.get(b, n))
 - 
 -         ext = re.search(r"\.([^.]+)$", doc.name)
 -         if ext:
 -             if doc.type == FileType.VISUAL.value:
 -                 response.headers.set("Content-Type", "image/%s" % ext.group(1))
 -             else:
 -                 response.headers.set("Content-Type", "application/%s" % ext.group(1))
 -         return response
 -     except Exception as e:
 -         return server_error_response(e)
 - 
 - 
 - @manager.route("/change_parser", methods=["POST"])  # noqa: F821
 - @login_required
 - @validate_request("doc_id", "parser_id")
 - def change_parser():
 -     req = request.json
 - 
 -     if not DocumentService.accessible(req["doc_id"], current_user.id):
 -         return get_json_result(data=False, message="No authorization.", code=settings.RetCode.AUTHENTICATION_ERROR)
 -     try:
 -         e, doc = DocumentService.get_by_id(req["doc_id"])
 -         if not e:
 -             return get_data_error_result(message="Document not found!")
 -         if doc.parser_id.lower() == req["parser_id"].lower():
 -             if "parser_config" in req:
 -                 if req["parser_config"] == doc.parser_config:
 -                     return get_json_result(data=True)
 -             else:
 -                 return get_json_result(data=True)
 - 
 -         if (doc.type == FileType.VISUAL and req["parser_id"] != "picture") or (re.search(r"\.(ppt|pptx|pages)$", doc.name) and req["parser_id"] != "presentation"):
 -             return get_data_error_result(message="Not supported yet!")
 - 
 -         e = DocumentService.update_by_id(doc.id, {"parser_id": req["parser_id"], "progress": 0, "progress_msg": "", "run": TaskStatus.UNSTART.value})
 -         if not e:
 -             return get_data_error_result(message="Document not found!")
 -         if "parser_config" in req:
 -             DocumentService.update_parser_config(doc.id, req["parser_config"])
 -         if doc.token_num > 0:
 -             e = DocumentService.increment_chunk_num(doc.id, doc.kb_id, doc.token_num * -1, doc.chunk_num * -1, doc.process_duration * -1)
 -             if not e:
 -                 return get_data_error_result(message="Document not found!")
 -             tenant_id = DocumentService.get_tenant_id(req["doc_id"])
 -             if not tenant_id:
 -                 return get_data_error_result(message="Tenant not found!")
 -             if settings.docStoreConn.indexExist(search.index_name(tenant_id), doc.kb_id):
 -                 settings.docStoreConn.delete({"doc_id": doc.id}, search.index_name(tenant_id), doc.kb_id)
 - 
 -         return get_json_result(data=True)
 -     except Exception as e:
 -         return server_error_response(e)
 - 
 - 
 - @manager.route("/image/<image_id>", methods=["GET"])  # noqa: F821
 - # @login_required
 - def get_image(image_id):
 -     try:
 -         arr = image_id.split("-")
 -         if len(arr) != 2:
 -             return get_data_error_result(message="Image not found.")
 -         bkt, nm = image_id.split("-")
 -         response = flask.make_response(STORAGE_IMPL.get(bkt, nm))
 -         response.headers.set("Content-Type", "image/JPEG")
 -         return response
 -     except Exception as e:
 -         return server_error_response(e)
 - 
 - 
 - @manager.route("/upload_and_parse", methods=["POST"])  # noqa: F821
 - @login_required
 - @validate_request("conversation_id")
 - def upload_and_parse():
 -     if "file" not in request.files:
 -         return get_json_result(data=False, message="No file part!", code=settings.RetCode.ARGUMENT_ERROR)
 - 
 -     file_objs = request.files.getlist("file")
 -     for file_obj in file_objs:
 -         if file_obj.filename == "":
 -             return get_json_result(data=False, message="No file selected!", code=settings.RetCode.ARGUMENT_ERROR)
 - 
 -     doc_ids = doc_upload_and_parse(request.form.get("conversation_id"), file_objs, current_user.id)
 - 
 -     return get_json_result(data=doc_ids)
 - 
 - 
 - @manager.route("/parse", methods=["POST"])  # noqa: F821
 - @login_required
 - def parse():
 -     url = request.json.get("url") if request.json else ""
 -     if url:
 -         if not is_valid_url(url):
 -             return get_json_result(data=False, message="The URL format is invalid", code=settings.RetCode.ARGUMENT_ERROR)
 -         download_path = os.path.join(get_project_base_directory(), "logs/downloads")
 -         os.makedirs(download_path, exist_ok=True)
 -         from seleniumwire.webdriver import Chrome, ChromeOptions
 - 
 -         options = ChromeOptions()
 -         options.add_argument("--headless")
 -         options.add_argument("--disable-gpu")
 -         options.add_argument("--no-sandbox")
 -         options.add_argument("--disable-dev-shm-usage")
 -         options.add_experimental_option("prefs", {"download.default_directory": download_path, "download.prompt_for_download": False, "download.directory_upgrade": True, "safebrowsing.enabled": True})
 -         driver = Chrome(options=options)
 -         driver.get(url)
 -         res_headers = [r.response.headers for r in driver.requests if r and r.response]
 -         if len(res_headers) > 1:
 -             sections = RAGFlowHtmlParser().parser_txt(driver.page_source)
 -             driver.quit()
 -             return get_json_result(data="\n".join(sections))
 - 
 -         class File:
 -             filename: str
 -             filepath: str
 - 
 -             def __init__(self, filename, filepath):
 -                 self.filename = filename
 -                 self.filepath = filepath
 - 
 -             def read(self):
 -                 with open(self.filepath, "rb") as f:
 -                     return f.read()
 - 
 -         r = re.search(r"filename=\"([^\"]+)\"", str(res_headers))
 -         if not r or not r.group(1):
 -             return get_json_result(data=False, message="Can't not identify downloaded file", code=settings.RetCode.ARGUMENT_ERROR)
 -         f = File(r.group(1), os.path.join(download_path, r.group(1)))
 -         txt = FileService.parse_docs([f], current_user.id)
 -         return get_json_result(data=txt)
 - 
 -     if "file" not in request.files:
 -         return get_json_result(data=False, message="No file part!", code=settings.RetCode.ARGUMENT_ERROR)
 - 
 -     file_objs = request.files.getlist("file")
 -     txt = FileService.parse_docs(file_objs, current_user.id)
 - 
 -     return get_json_result(data=txt)
 - 
 - 
 - @manager.route("/set_meta", methods=["POST"])  # noqa: F821
 - @login_required
 - @validate_request("doc_id", "meta")
 - def set_meta():
 -     req = request.json
 -     if not DocumentService.accessible(req["doc_id"], current_user.id):
 -         return get_json_result(data=False, message="No authorization.", code=settings.RetCode.AUTHENTICATION_ERROR)
 -     try:
 -         meta = json.loads(req["meta"])
 -     except Exception as e:
 -         return get_json_result(data=False, message=f"Json syntax error: {e}", code=settings.RetCode.ARGUMENT_ERROR)
 -     if not isinstance(meta, dict):
 -         return get_json_result(data=False, message='Meta data should be in Json map format, like {"key": "value"}', code=settings.RetCode.ARGUMENT_ERROR)
 - 
 -     try:
 -         e, doc = DocumentService.get_by_id(req["doc_id"])
 -         if not e:
 -             return get_data_error_result(message="Document not found!")
 - 
 -         if not DocumentService.update_by_id(req["doc_id"], {"meta_fields": meta}):
 -             return get_data_error_result(message="Database error (meta updates)!")
 - 
 -         return get_json_result(data=True)
 -     except Exception as e:
 -         return server_error_response(e)
 
 
  |