|
|
@@ -5,6 +5,7 @@ from pydantic import BaseModel
|
|
|
from tool.pdf2md import pdf2ai
|
|
|
from utils import dmQuery
|
|
|
from utils.shellUtil import execute_command_on_linux
|
|
|
+from utils.text_to_html import save_html_to_file
|
|
|
|
|
|
app = FastAPI() # 创建API实例
|
|
|
|
|
|
@@ -27,6 +28,11 @@ class Pdf2md(BaseModel):
|
|
|
pageNum: int
|
|
|
|
|
|
|
|
|
+class Saved2text(BaseModel):
|
|
|
+ htmlContent: str
|
|
|
+ fileName: str
|
|
|
+
|
|
|
+
|
|
|
@app.put("/exsql")
|
|
|
async def create_item( item: Item, q: str = None):
|
|
|
result = {**item.dict()}
|
|
|
@@ -57,6 +63,15 @@ async def pdf2md(item: Pdf2md, q: str = None):
|
|
|
|
|
|
result = pdf2ai(item.pdfUrl, item.prompt, item.pageNum)
|
|
|
|
|
|
+ return result @ app.put("/pdf2md")
|
|
|
+
|
|
|
+
|
|
|
+async def saved2text(item: Saved2text, q: str = None):
|
|
|
+ pdf_url = 'http://42.194.163.46:9007/ywd/%E6%89%AB%E6%8F%8F%E5%85%A8%E8%83%BD%E7%8E%8B%202025-03-20%2021.38.pdf'
|
|
|
+ prompt = "提取有效信息，只输出图片有效信息无效信息不用。"
|
|
|
+
|
|
|
+ result = save_html_to_file(item.htmlContent, item.fileName)
|
|
|
+
|
|
|
return result
|
|
|
|
|
|
|
