diff --git a/__pycache__/main.cpython-312.pyc b/__pycache__/main.cpython-312.pyc index d8bdf90..a5ad2d8 100644 Binary files a/__pycache__/main.cpython-312.pyc and b/__pycache__/main.cpython-312.pyc differ diff --git a/app/__pycache__/main.cpython-312.pyc b/app/__pycache__/main.cpython-312.pyc deleted file mode 100644 index 31996bd..0000000 Binary files a/app/__pycache__/main.cpython-312.pyc and /dev/null differ diff --git a/app/__pycache__/main.cpython-313.pyc b/app/__pycache__/main.cpython-313.pyc deleted file mode 100644 index fbc960b..0000000 Binary files a/app/__pycache__/main.cpython-313.pyc and /dev/null differ diff --git a/app/main.py b/app/main.py deleted file mode 100644 index 4e9616f..0000000 --- a/app/main.py +++ /dev/null @@ -1,131 +0,0 @@ -# from docx import Document -import os -from tempfile import NamedTemporaryFile - -import thulac -from fastapi import FastAPI, Form, UploadFile, File, Request -from fastapi.responses import FileResponse -from fastapi.responses import HTMLResponse -from fastapi.templating import Jinja2Templates -from pydantic import BaseModel -from reportlab.lib.pagesizes import letter -from reportlab.pdfgen import canvas - - -class ExportRequest(BaseModel): - format: str # 导出格式（html、pdf、rtf） - content: str # 要导出的内容 - -# 初始化 THULAC 分词工具 -thu = thulac.thulac() - -# 初始化 FastAPI 应用和模板 -app = FastAPI() -templates = Jinja2Templates(directory="templates") - - -@app.get("/", response_class=HTMLResponse) -async def home(request: Request): - return templates.TemplateResponse("index.html", {"request": request}) - - -@app.post("/process", response_class=HTMLResponse) -async def process(request: Request, file: UploadFile = File(None), text: str = Form(None)): - """ - 统一处理上传文件和粘贴文本，返回标注结果页面 - """ - if file: - # 读取上传文件内容 - content = await file.read() - text_content = content.decode("utf-8") - elif text: - # 读取粘贴文本内容 - text_content = text - else: - return templates.TemplateResponse( - "error.html", {"request": request, "message": "未提供文件或文本"} - ) - - # 处理文本，返回标注结果 - processed_text = process_text(text) - - return templates.TemplateResponse("result.html", {"request": request, "content": processed_text}) - - -def process_text(text: str) -> str: - """处理文本并为每种词性添加 CSS 类，同时保留段落换行""" - words = thu.cut(text) - colored_text = "" - - for word, tag in words: - if tag: # 确保 tag 不为空 - colored_text += f"{word}" - else: - colored_text += word # 没有词性时直接添加词语 - - # 替换换行符为
- return colored_text.replace("\n", "
") - - - -@app.post("/export") -async def export_file(request: ExportRequest): - format = request.format - content = request.content - - if format == "html": - file_name = "exported_result.html" - file_content = f"{content}" - return create_file_response(file_name, file_content.encode("utf-8"), "text/html") - - elif format == "pdf": - with NamedTemporaryFile(delete=False, suffix=".pdf") as temp_file: - generate_pdf(temp_file.name, content) - return FileResponse(temp_file.name, filename="exported_result.pdf", media_type="application/pdf") - - elif format == "rtf": - file_name = "exported_result.rtf" - file_content = generate_rtf(content) - return create_file_response(file_name, file_content.encode("utf-8"), "application/rtf") - - return {"error": "Unsupported format"} - - -def create_file_response(file_name, file_content, media_type): - """ - 创建文件响应 - """ - with NamedTemporaryFile(delete=False, suffix=os.path.splitext(file_name)[1]) as temp_file: - temp_file.write(file_content) - temp_file.flush() - return FileResponse(temp_file.name, filename=file_name, media_type=media_type) - - -def generate_pdf(file_path, content): - """ - 生成 PDF 文件 - """ - c = canvas.Canvas(file_path, pagesize=letter) - c.setFont("Helvetica", 12) - width, height = letter - y = height - 40 - - for line in content.split("
"): - c.drawString(40, y, line.strip()) - y -= 20 - if y < 40: # 换页条件 - c.showPage() - c.setFont("Helvetica", 12) - y = height - 40 - c.save() - - -def generate_rtf(content): - """ - 生成 RTF 文件 - """ - rtf_header = r"{\rtf1\ansi\deff0" - rtf_footer = r"}" - rtf_body = content.replace("
", r"\line ") - return rtf_header + rtf_body + rtf_footer - diff --git a/app/templates/error.html b/app/templates/error.html deleted file mode 100644 index e69de29..0000000 diff --git a/app/templates/index.html b/app/templates/index.html deleted file mode 100644 index ef8dd38..0000000 --- a/app/templates/index.html +++ /dev/null @@ -1,156 +0,0 @@ - - - - - - 词性标注工具 - - - -
-

词性标注工具

- - -
- - -
- - -
-
- - -
- -
-
-
- - -
-
- - -
- -
-
-
-
- - - - diff --git a/app/templates/result.html b/app/templates/result.html deleted file mode 100644 index b67b313..0000000 --- a/app/templates/result.html +++ /dev/null @@ -1,164 +0,0 @@ - - - - - - 标注结果 - - - -

标注结果

- - - 返回 - - -
- 词性显示控制： - - - - - -
- - -
{{ content | safe }}
- -
- 导出结果： - - - -
- - - - - - diff --git a/main.py b/main.py index fb182fd..4e9616f 100644 --- a/main.py +++ b/main.py @@ -1,7 +1,131 @@ -def main(): - print("Starting mark-word-fastapi...") - print("Hello from mark-word-fastapi!") +# from docx import Document +import os +from tempfile import NamedTemporaryFile + +import thulac +from fastapi import FastAPI, Form, UploadFile, File, Request +from fastapi.responses import FileResponse +from fastapi.responses import HTMLResponse +from fastapi.templating import Jinja2Templates +from pydantic import BaseModel +from reportlab.lib.pagesizes import letter +from reportlab.pdfgen import canvas -if __name__ == "__main__": - main() +class ExportRequest(BaseModel): + format: str # 导出格式（html、pdf、rtf） + content: str # 要导出的内容 + +# 初始化 THULAC 分词工具 +thu = thulac.thulac() + +# 初始化 FastAPI 应用和模板 +app = FastAPI() +templates = Jinja2Templates(directory="templates") + + +@app.get("/", response_class=HTMLResponse) +async def home(request: Request): + return templates.TemplateResponse("index.html", {"request": request}) + + +@app.post("/process", response_class=HTMLResponse) +async def process(request: Request, file: UploadFile = File(None), text: str = Form(None)): + """ + 统一处理上传文件和粘贴文本，返回标注结果页面 + """ + if file: + # 读取上传文件内容 + content = await file.read() + text_content = content.decode("utf-8") + elif text: + # 读取粘贴文本内容 + text_content = text + else: + return templates.TemplateResponse( + "error.html", {"request": request, "message": "未提供文件或文本"} + ) + + # 处理文本，返回标注结果 + processed_text = process_text(text) + + return templates.TemplateResponse("result.html", {"request": request, "content": processed_text}) + + +def process_text(text: str) -> str: + """处理文本并为每种词性添加 CSS 类，同时保留段落换行""" + words = thu.cut(text) + colored_text = "" + + for word, tag in words: + if tag: # 确保 tag 不为空 + colored_text += f"{word}" + else: + colored_text += word # 没有词性时直接添加词语 + + # 替换换行符为
+ return colored_text.replace("\n", "
") + + + +@app.post("/export") +async def export_file(request: ExportRequest): + format = request.format + content = request.content + + if format == "html": + file_name = "exported_result.html" + file_content = f"{content}" + return create_file_response(file_name, file_content.encode("utf-8"), "text/html") + + elif format == "pdf": + with NamedTemporaryFile(delete=False, suffix=".pdf") as temp_file: + generate_pdf(temp_file.name, content) + return FileResponse(temp_file.name, filename="exported_result.pdf", media_type="application/pdf") + + elif format == "rtf": + file_name = "exported_result.rtf" + file_content = generate_rtf(content) + return create_file_response(file_name, file_content.encode("utf-8"), "application/rtf") + + return {"error": "Unsupported format"} + + +def create_file_response(file_name, file_content, media_type): + """ + 创建文件响应 + """ + with NamedTemporaryFile(delete=False, suffix=os.path.splitext(file_name)[1]) as temp_file: + temp_file.write(file_content) + temp_file.flush() + return FileResponse(temp_file.name, filename=file_name, media_type=media_type) + + +def generate_pdf(file_path, content): + """ + 生成 PDF 文件 + """ + c = canvas.Canvas(file_path, pagesize=letter) + c.setFont("Helvetica", 12) + width, height = letter + y = height - 40 + + for line in content.split("
"): + c.drawString(40, y, line.strip()) + y -= 20 + if y < 40: # 换页条件 + c.showPage() + c.setFont("Helvetica", 12) + y = height - 40 + c.save() + + +def generate_rtf(content): + """ + 生成 RTF 文件 + """ + rtf_header = r"{\rtf1\ansi\deff0" + rtf_footer = r"}" + rtf_body = content.replace("
", r"\line ") + return rtf_header + rtf_body + rtf_footer +