630 lines
29 KiB
Python
630 lines
29 KiB
Python
import asyncio
|
||
import io
|
||
import logging
|
||
import os
|
||
from pathlib import Path
|
||
from typing import AsyncGenerator
|
||
|
||
import uvicorn
|
||
from fastapi import FastAPI, File, Form, UploadFile, Request, HTTPException
|
||
from fastapi.responses import HTMLResponse, StreamingResponse, JSONResponse
|
||
from fastapi.templating import Jinja2Templates
|
||
|
||
# 假设这些导入能够正确找到你的库代码
|
||
from docutranslate import FileTranslater # Your existing FileTranslater
|
||
from docutranslate.logger import translater_logger # Your existing logger
|
||
|
||
os.environ["FASTAPI_RUNNING"] = "true"
|
||
app = FastAPI()
|
||
|
||
# --- 异步队列和自定义日志处理器设置 ---
|
||
log_queue = asyncio.Queue()
|
||
SHUTDOWN_SENTINEL = object() # 使用一个唯一的对象作为哨兵
|
||
|
||
|
||
class AsyncQueueHandler(logging.Handler):
|
||
def __init__(self, queue: asyncio.Queue):
|
||
super().__init__()
|
||
self.queue = queue
|
||
|
||
def emit(self, record: logging.LogRecord):
|
||
log_entry = self.format(record)
|
||
# 在 FastAPI 应用上下文中运行时，尝试使用 app.state.main_event_loop
|
||
main_loop = getattr(app.state, "main_event_loop", None)
|
||
if main_loop and main_loop.is_running():
|
||
main_loop.call_soon_threadsafe(self.queue.put_nowait, log_entry)
|
||
else:
|
||
# 如果主循环不可用或未运行（例如，在测试中或非常早期的启动/非常晚的关闭阶段）
|
||
# 这是一个备用方案，但不如 call_soon_threadsafe 安全
|
||
try:
|
||
# 如果在主事件循环上下文之外，或者事件循环已停止，
|
||
# put_nowait 可能仍然有效，因为它不依赖于正在运行的特定循环来放置项目
|
||
# 但理想情况下，日志记录应在主循环活跃时发生。
|
||
self.queue.put_nowait(log_entry)
|
||
except RuntimeError: # 例如，如果队列本身与已关闭的循环关联
|
||
print(f"Error putting log to queue (loop likely closed): {log_entry[:100]}...") # 记录部分日志以避免过长输出
|
||
self.handleError(record) # 调用基类的错误处理
|
||
except Exception as e:
|
||
print(f"Error putting log to queue (no main loop/not running): {e}")
|
||
self.handleError(record)
|
||
|
||
|
||
@app.on_event("startup")
|
||
async def startup_event():
|
||
app.state.main_event_loop = asyncio.get_running_loop()
|
||
queue_handler = AsyncQueueHandler(log_queue)
|
||
queue_handler.setLevel(logging.INFO)
|
||
ui_formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
|
||
queue_handler.setFormatter(ui_formatter)
|
||
# 检查 translater_logger 是否已经有这个类型的 handler，避免重复添加
|
||
if not any(isinstance(h, AsyncQueueHandler) for h in translater_logger.handlers):
|
||
translater_logger.addHandler(queue_handler)
|
||
translater_logger.info("Application startup complete. Log queue handler configured.")
|
||
|
||
|
||
@app.on_event("shutdown")
|
||
async def shutdown_event():
|
||
translater_logger.info("Application shutting down. Signaling log streamer to stop.")
|
||
# 向队列发送哨兵值以停止日志流生成器
|
||
await log_queue.put(SHUTDOWN_SENTINEL)
|
||
# (可选) 短暂等待，以允许生成器处理哨兵并退出
|
||
await asyncio.sleep(0.1)
|
||
translater_logger.info("Log streamer signaled.")
|
||
# (可选) 清空队列中剩余的日志，如果不想在关闭时处理它们
|
||
# while not log_queue.empty():
|
||
# try:
|
||
# log_queue.get_nowait()
|
||
# log_queue.task_done()
|
||
# except asyncio.QueueEmpty:
|
||
# break
|
||
# translater_logger.info("Log queue cleared during shutdown.")
|
||
|
||
|
||
# --- 全局状态 ---
|
||
current_translation_state = {
|
||
"markdown_content": None, "html_content": None, "original_filename_stem": None,
|
||
"error": None, "is_processing": False
|
||
}
|
||
templates = Jinja2Templates(directory=".") # 假设模板在当前目录或使用字符串模板
|
||
|
||
# --- HTML 模板字符串 ---
|
||
HTML_TEMPLATE_STR = """
|
||
<!DOCTYPE html>
|
||
<html lang="zh-CN">
|
||
<head>
|
||
<meta charset="UTF-8">
|
||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||
<title>DocuTranslate</title>
|
||
<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@picocss/pico@latest/css/pico.min.css">
|
||
<style>
|
||
:root {
|
||
--primary-color: #1e88e5;
|
||
--border-radius: 0.25rem;
|
||
}
|
||
body {
|
||
padding: 20px;
|
||
font-family: system-ui, -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
|
||
background-color: #f9f9f9;
|
||
}
|
||
.container {
|
||
max-width: 800px;
|
||
margin: auto;
|
||
background-color: white;
|
||
padding: 2rem;
|
||
border-radius: 8px;
|
||
box-shadow: 0 2px 10px rgba(0,0,0,0.05);
|
||
}
|
||
h1 {
|
||
font-size: 1.8rem;
|
||
margin-bottom: 1.5rem;
|
||
color: var(--primary-color);
|
||
display: flex;
|
||
align-items: center;
|
||
gap: 0.5rem;
|
||
}
|
||
h1 a {
|
||
text-decoration: none;
|
||
}
|
||
h1 a:hover {
|
||
text-decoration: underline;
|
||
}
|
||
.log-area {
|
||
background-color: #f5f5f5;
|
||
border: 1px solid #e0e0e0;
|
||
border-radius: var(--border-radius);
|
||
padding: 10px;
|
||
height: 200px;
|
||
overflow-y: scroll;
|
||
white-space: pre-wrap;
|
||
word-break: break-all;
|
||
font-family: monospace;
|
||
font-size: 0.85em;
|
||
line-height: 1.4;
|
||
margin-top: 1rem;
|
||
}
|
||
.error-message { color: #d32f2f; font-weight: 500; }
|
||
.success-message { color: #2e7d32; font-weight: 500; }
|
||
.form-group { margin-bottom: 1rem; }
|
||
.form-group label {
|
||
margin-bottom: 0.2rem;
|
||
display: block;
|
||
font-weight: 500;
|
||
font-size: 0.9rem;
|
||
}
|
||
.form-grid {
|
||
display: grid;
|
||
grid-template-columns: 1fr 1fr;
|
||
gap: 1rem;
|
||
}
|
||
.button-group {
|
||
margin-top: 1rem;
|
||
display: flex;
|
||
gap: 0.5rem;
|
||
flex-wrap: wrap;
|
||
}
|
||
summary {
|
||
font-weight: 500;
|
||
cursor: pointer;
|
||
padding: 0.5rem 0;
|
||
}
|
||
details {
|
||
border-bottom: 1px solid #eee;
|
||
margin-bottom: 1rem;
|
||
}
|
||
.checkbox-label {
|
||
display: flex;
|
||
align-items: center;
|
||
margin-right: 1rem;
|
||
margin-bottom: 0.5rem;
|
||
}
|
||
.checkbox-label input[type="checkbox"] {
|
||
margin-right: 0.5rem;
|
||
}
|
||
.checkbox-group {
|
||
display: flex;
|
||
flex-wrap: wrap;
|
||
margin-bottom: 1rem;
|
||
}
|
||
#resultArea {
|
||
margin-top: 1.5rem;
|
||
padding-top: 1rem;
|
||
border-top: 1px solid #eee;
|
||
}
|
||
#downloadButtons {
|
||
display: none;
|
||
margin-top: 1rem;
|
||
}
|
||
.section-header {
|
||
display: flex;
|
||
align-items: center;
|
||
margin-bottom: 0.5rem;
|
||
font-size: 1.1rem;
|
||
font-weight: 500;
|
||
}
|
||
select, input[type="text"], input[type="password"], input[type="file"] {
|
||
padding: 0.5rem;
|
||
border: 1px solid #ddd;
|
||
border-radius: var(--border-radius);
|
||
background-color: white;
|
||
}
|
||
button, a[role="button"] {
|
||
border-radius: var(--border-radius);
|
||
padding: 0.5rem 1rem;
|
||
}
|
||
.options-grid {
|
||
display: grid;
|
||
grid-template-columns: 1fr 1fr;
|
||
gap: 1rem;
|
||
margin-bottom: 1rem;
|
||
}
|
||
@media (max-width: 768px) {
|
||
.form-grid, .options-grid {
|
||
grid-template-columns: 1fr;
|
||
}
|
||
.container {
|
||
padding: 1rem;
|
||
}
|
||
}
|
||
</style>
|
||
</head>
|
||
<body>
|
||
<main class="container">
|
||
<h1>
|
||
<svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
|
||
<path d="M5 8l6 6"></path><path d="M4 14l6-6 2-3"></path><path d="M2 5l7 7v6"></path>
|
||
<path d="M15 10l6 6"></path><path d="M19 6l-7 7-6 2"></path><path d="M22 5l-7 7v6"></path>
|
||
</svg>
|
||
<a href="https://github.com/xunbu/docutranslate" target="_blank">DocuTranslate</a>
|
||
</h1>
|
||
|
||
<form id="translateForm">
|
||
<details>
|
||
<summary>API 配置</summary>
|
||
<div class="form-grid">
|
||
<div class="form-group">
|
||
<label for="base_url">API 地址</label>
|
||
<input type="text" id="base_url" name="base_url" placeholder="如: https://api.openai.com/v1" required>
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="apikey">API 密钥</label>
|
||
<input type="password" id="apikey" name="apikey" placeholder="sk-..." required>
|
||
</div>
|
||
</div>
|
||
<div class="form-group">
|
||
<label for="model_id">模型 ID</label>
|
||
<input type="text" id="model_id" name="model_id" placeholder="如: gpt-4-turbo-preview" required>
|
||
</div>
|
||
</details>
|
||
|
||
<div class="form-group">
|
||
<label for="file">文档选择</label>
|
||
<input type="file" id="file" name="file" required>
|
||
</div>
|
||
|
||
<div class="options-grid">
|
||
<div class="form-group">
|
||
<label for="to_lang">目标语言</label>
|
||
<select id="to_lang" name="to_lang">
|
||
<option value="中文">中文 (Chinese)</option>
|
||
<option value="English">英文 (English)</option>
|
||
<option value="日本語">日语 (Japanese)</option>
|
||
<option value="한국어">韩语 (Korean)</option>
|
||
<option value="Français">法语 (French)</option>
|
||
<option value="Deutsch">德语 (German)</option>
|
||
<option value="Español">西班牙语 (Spanish)</option>
|
||
<option value="Italiano">意大利语 (Italian)</option>
|
||
<option value="Português">葡萄牙语 (Portuguese)</option>
|
||
<option value="Русский">俄语 (Russian)</option>
|
||
<option value="العربية">阿拉伯语 (Arabic)</option>
|
||
<option value="हिन्दी">印地语 (Hindi)</option>
|
||
</select>
|
||
</div>
|
||
<div class="form-group">
|
||
<label>高级选项</label>
|
||
<div class="checkbox-group">
|
||
<label class="checkbox-label" for="formula_ocr">
|
||
<input type="checkbox" id="formula_ocr" name="formula_ocr">
|
||
公式识别
|
||
</label>
|
||
<label class="checkbox-label" for="code_ocr">
|
||
<input type="checkbox" id="code_ocr" name="code_ocr">
|
||
代码识别
|
||
</label>
|
||
<label class="checkbox-label" for="refine_markdown">
|
||
<input type="checkbox" id="refine_markdown" name="refine_markdown">
|
||
优化 Markdown
|
||
</label>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
|
||
<button type="submit" id="submitButton" class="primary">开始翻译</button>
|
||
</form>
|
||
|
||
<div id="resultArea">
|
||
<p id="statusMessage"></p>
|
||
<div id="downloadButtons" class="button-group">
|
||
<div class="section-header">翻译结果</div>
|
||
<a id="downloadMarkdown" href="#" role="button" class="outline">下载 Markdown</a>
|
||
<a id="downloadHtml" href="#" role="button" class="outline">下载 HTML</a>
|
||
</div>
|
||
</div>
|
||
|
||
<div class="section-header" style="margin-top: 1.5rem;">运行日志</div>
|
||
<div class="log-area" id="logArea"></div>
|
||
</main>
|
||
|
||
<script>
|
||
const base_url_input = document.getElementById('base_url');
|
||
const apikey_input = document.getElementById('apikey');
|
||
const model_id_input = document.getElementById('model_id');
|
||
const to_lang_input = document.getElementById('to_lang');
|
||
const formula_ocr_input = document.getElementById('formula_ocr');
|
||
const code_ocr_input = document.getElementById('code_ocr');
|
||
const refine_markdown_input = document.getElementById('refine_markdown');
|
||
|
||
// Load saved values from localStorage
|
||
if (localStorage.getItem('translator_base_url')) base_url_input.value = localStorage.getItem('translator_base_url');
|
||
if (localStorage.getItem('translator_apikey')) apikey_input.value = localStorage.getItem('translator_apikey');
|
||
if (localStorage.getItem('translator_model_id')) model_id_input.value = localStorage.getItem('translator_model_id');
|
||
if (localStorage.getItem('translator_to_lang')) {
|
||
const savedLang = localStorage.getItem('translator_to_lang');
|
||
// Find option with matching value
|
||
for (const option of to_lang_input.options) {
|
||
if (option.value === savedLang) {
|
||
option.selected = true;
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
formula_ocr_input.checked = localStorage.getItem('translator_formula_ocr') === 'true';
|
||
code_ocr_input.checked = localStorage.getItem('translator_code_ocr') === 'true';
|
||
refine_markdown_input.checked = localStorage.getItem('translator_refine_markdown') === 'true';
|
||
|
||
// Save to localStorage
|
||
function saveToLocalStorage(key, value) {
|
||
try {
|
||
localStorage.setItem(key, value);
|
||
} catch (e) {
|
||
console.error("Error saving to localStorage:", e);
|
||
}
|
||
}
|
||
|
||
base_url_input.addEventListener('input', () => saveToLocalStorage('translator_base_url', base_url_input.value));
|
||
apikey_input.addEventListener('input', () => saveToLocalStorage('translator_apikey', apikey_input.value));
|
||
model_id_input.addEventListener('input', () => saveToLocalStorage('translator_model_id', model_id_input.value));
|
||
to_lang_input.addEventListener('change', () => saveToLocalStorage('translator_to_lang', to_lang_input.value));
|
||
formula_ocr_input.addEventListener('change', () => saveToLocalStorage('translator_formula_ocr', formula_ocr_input.checked));
|
||
code_ocr_input.addEventListener('change', () => saveToLocalStorage('translator_code_ocr', code_ocr_input.checked));
|
||
refine_markdown_input.addEventListener('change', () => saveToLocalStorage('translator_refine_markdown', refine_markdown_input.checked));
|
||
|
||
const form = document.getElementById('translateForm');
|
||
const submitButton = document.getElementById('submitButton');
|
||
const logArea = document.getElementById('logArea');
|
||
const statusMessageElement = document.getElementById('statusMessage');
|
||
const downloadButtonsDiv = document.getElementById('downloadButtons');
|
||
const downloadMarkdownLink = document.getElementById('downloadMarkdown');
|
||
const downloadHtmlLink = document.getElementById('downloadHtml');
|
||
|
||
if (form && submitButton) {
|
||
form.addEventListener('submit', async function(event) {
|
||
event.preventDefault();
|
||
submitButton.disabled = true;
|
||
submitButton.setAttribute('aria-busy', 'true');
|
||
submitButton.textContent = '翻译中...';
|
||
if(logArea) logArea.innerHTML = '';
|
||
statusMessageElement.textContent = '';
|
||
statusMessageElement.className = '';
|
||
downloadButtonsDiv.style.display = 'none';
|
||
|
||
const formData = new FormData(form);
|
||
try {
|
||
const response = await fetch('/translate', { method: 'POST', body: formData });
|
||
const resultData = await response.json();
|
||
if (resultData.error) {
|
||
statusMessageElement.textContent = resultData.message;
|
||
statusMessageElement.className = 'error-message';
|
||
} else {
|
||
statusMessageElement.textContent = resultData.message;
|
||
statusMessageElement.className = 'success-message';
|
||
if (resultData.download_ready) {
|
||
downloadMarkdownLink.href = resultData.markdown_url;
|
||
downloadMarkdownLink.setAttribute('download', resultData.original_filename_stem + '_translated.md');
|
||
downloadHtmlLink.href = resultData.html_url;
|
||
downloadHtmlLink.setAttribute('download', resultData.original_filename_stem + '_translated.html');
|
||
downloadButtonsDiv.style.display = 'block';
|
||
}
|
||
}
|
||
} catch (error) {
|
||
console.error('Fetch error:', error);
|
||
statusMessageElement.textContent = '请求翻译失败，请检查网络或服务状态。';
|
||
statusMessageElement.className = 'error-message';
|
||
} finally {
|
||
submitButton.disabled = false;
|
||
submitButton.removeAttribute('aria-busy');
|
||
submitButton.textContent = '开始翻译';
|
||
}
|
||
});
|
||
}
|
||
|
||
if (typeof(EventSource) !== "undefined") {
|
||
let eventSource;
|
||
function connectEventSource() {
|
||
if (eventSource) {
|
||
eventSource.close();
|
||
}
|
||
eventSource = new EventSource("/stream-logs");
|
||
eventSource.onmessage = function(event) {
|
||
if (logArea && event.data !== ":heartbeat") { // Ignore heartbeat messages for display
|
||
logArea.innerHTML += event.data;
|
||
logArea.scrollTop = logArea.scrollHeight;
|
||
}
|
||
};
|
||
eventSource.onerror = function(err) {
|
||
console.error("EventSource failed:", err);
|
||
if (logArea) {
|
||
const errorMsgDiv = document.createElement('div');
|
||
errorMsgDiv.style.color = 'orange';
|
||
errorMsgDiv.textContent = '日志流连接暂时中断，尝试重连...';
|
||
logArea.appendChild(errorMsgDiv);
|
||
logArea.scrollTop = logArea.scrollHeight;
|
||
}
|
||
eventSource.close(); // Close the failed source
|
||
// Attempt to reconnect after a delay
|
||
setTimeout(connectEventSource, 5000); // Reconnect after 5 seconds
|
||
};
|
||
}
|
||
connectEventSource(); // Initial connection
|
||
} else {
|
||
if(logArea) logArea.innerHTML = "抱歉，您的浏览器不支持实时日志更新。";
|
||
}
|
||
</script>
|
||
</body>
|
||
</html>
|
||
"""
|
||
|
||
|
||
# --- FastAPI Endpoints ---
|
||
@app.get("/", response_class=HTMLResponse)
|
||
async def main_page_get_endpoint(request: Request):
|
||
# Clear log queue only if not processing, to avoid clearing logs of an ongoing task
|
||
# when page is reloaded. However, SSE should keep logs flowing.
|
||
# This logic might be redundant if SSE handles logs independently of page reloads.
|
||
if not current_translation_state["is_processing"]:
|
||
while not log_queue.empty():
|
||
try:
|
||
log_queue.get_nowait();
|
||
log_queue.task_done()
|
||
except asyncio.QueueEmpty:
|
||
break
|
||
|
||
context = {"request": request, "config": {}, "message": None, "error": False, "download_ready": False}
|
||
# If you are using Jinja2Templates with a file:
|
||
# return templates.TemplateResponse("your_template_name.html", context)
|
||
# If using the string template:
|
||
jinja_env = templates.env # Or initialize a Jinja2 Environment if not using FastAPI's templates
|
||
template_obj = jinja_env.from_string(HTML_TEMPLATE_STR)
|
||
return HTMLResponse(content=template_obj.render(context))
|
||
|
||
|
||
async def log_stream_generator() -> AsyncGenerator[str, None]:
|
||
last_heartbeat_time = asyncio.get_event_loop().time()
|
||
heartbeat_interval = 15 # Send heartbeat every 15 seconds
|
||
is_shutting_down = False
|
||
|
||
try:
|
||
while not is_shutting_down: # Loop until sentinel or cancellation
|
||
log_message = None
|
||
try:
|
||
# Wait for a log message with a timeout, so we can send heartbeats
|
||
# and check for shutdown sentinel periodically.
|
||
log_message = await asyncio.wait_for(log_queue.get(), timeout=1.0)
|
||
except asyncio.TimeoutError:
|
||
# No log message in this interval, proceed to check heartbeat
|
||
pass
|
||
except asyncio.CancelledError: # Handle cancellation if client disconnects
|
||
translater_logger.info("Log stream generator cancelled by client disconnect.")
|
||
raise # Re-raise to ensure task cleanup
|
||
|
||
if log_message is SHUTDOWN_SENTINEL:
|
||
translater_logger.info("Log stream generator received shutdown sentinel. Exiting.")
|
||
log_queue.task_done() # Mark sentinel as processed
|
||
is_shutting_down = True
|
||
break # Exit the loop
|
||
|
||
if log_message: # Process actual log message
|
||
# Basic HTML escaping for log messages to prevent XSS if logs contain HTML/JS
|
||
escaped_message = log_message.replace('&', '&').replace('<', '<').replace('>', '>')
|
||
yield f"data: {escaped_message}<br>\n\n"
|
||
log_queue.task_done()
|
||
last_heartbeat_time = asyncio.get_event_loop().time() # Reset heartbeat timer on actual data
|
||
|
||
current_time = asyncio.get_event_loop().time()
|
||
if current_time - last_heartbeat_time >= heartbeat_interval:
|
||
yield "data: :heartbeat\n\n"
|
||
last_heartbeat_time = current_time
|
||
|
||
except asyncio.CancelledError: # Catch again if cancellation happens outside the get()
|
||
translater_logger.info("Log stream generator task was cancelled externally.")
|
||
# Ensure any pending item in queue due to this generator is marked done IF it was fetched
|
||
# However, at this point, it's safer to just re-raise.
|
||
raise
|
||
finally:
|
||
translater_logger.info("Log stream generator finished.")
|
||
# Ensure the queue is not blocked if join() is ever used elsewhere for this queue.
|
||
# If a log_message was retrieved but not task_done'd before cancellation/sentinel,
|
||
# this could be an issue. The current logic should cover it.
|
||
|
||
|
||
@app.get("/stream-logs")
|
||
async def stream_logs_endpoint(request: Request):
|
||
return StreamingResponse(log_stream_generator(), media_type="text/event-stream")
|
||
|
||
|
||
@app.post("/translate", response_class=JSONResponse)
|
||
async def handle_translate_endpoint(
|
||
request: Request, # Keep request if needed for other things, like client IP
|
||
base_url: str = Form(...), apikey: str = Form(...), model_id: str = Form(...),
|
||
to_lang: str = Form("中文"), formula_ocr: bool = Form(False),
|
||
code_ocr: bool = Form(False), refine_markdown: bool = Form(False),
|
||
file: UploadFile = File(...)
|
||
):
|
||
if current_translation_state["is_processing"]:
|
||
return JSONResponse(status_code=429, content={"error": True, "message": "另一个翻译任务正在进行中，请稍后再试。"})
|
||
|
||
current_translation_state["is_processing"] = True
|
||
# It's good practice to clear the log queue for a new task if appropriate,
|
||
# or ensure old logs don't interfere. The AsyncQueueHandler means logs are
|
||
# continuously added, so clearing here makes sense for a "fresh" log view per task.
|
||
# However, the main page GET also clears it, so be mindful of desired behavior.
|
||
# For now, let's assume logs for a new task should start fresh.
|
||
while not log_queue.empty():
|
||
try:
|
||
item = log_queue.get_nowait()
|
||
if item is SHUTDOWN_SENTINEL: # Put sentinel back if accidentally removed
|
||
log_queue.put_nowait(SHUTDOWN_SENTINEL)
|
||
log_queue.task_done()
|
||
except asyncio.QueueEmpty:
|
||
break
|
||
|
||
translater_logger.info("收到翻译请求。")
|
||
response_data = {"error": False, "message": "", "download_ready": False, "markdown_url": None, "html_url": None,
|
||
"original_filename_stem": None}
|
||
|
||
file_contents = None # Initialize to ensure it's defined for finally block
|
||
try:
|
||
file_contents = await file.read() # Read file contents
|
||
original_filename = file.filename if file.filename else "uploaded_file"
|
||
current_translation_state["original_filename_stem"] = Path(original_filename).stem
|
||
response_data["original_filename_stem"] = current_translation_state["original_filename_stem"]
|
||
|
||
translater_logger.info(f"文件 '{original_filename}' 已上传, 大小: {len(file_contents)} 字节。")
|
||
|
||
ft = FileTranslater(base_url=base_url, key=apikey, model_id=model_id, tips=False)
|
||
|
||
# Run the blocking translation task in a separate thread
|
||
await asyncio.to_thread(
|
||
ft.translate_bytes, name=original_filename, file=file_contents, to_lang=to_lang,
|
||
formula=formula_ocr, code=code_ocr, refine=refine_markdown, save=False
|
||
# save=False if handling content in memory
|
||
)
|
||
|
||
# Assuming FileTranslater populates its internal state with translated content
|
||
current_translation_state["markdown_content"] = ft.export_to_markdown()
|
||
current_translation_state["html_content"] = ft.export_to_html(
|
||
title=current_translation_state["original_filename_stem"]) # Pass title if your method supports it
|
||
|
||
response_data["message"] = "翻译成功！下载链接已生成。"
|
||
response_data["download_ready"] = True
|
||
response_data["markdown_url"] = f"/download/markdown/{response_data['original_filename_stem']}_translated.md"
|
||
response_data["html_url"] = f"/download/html/{response_data['original_filename_stem']}_translated.html"
|
||
translater_logger.info("翻译流程处理完毕。")
|
||
|
||
except Exception as e:
|
||
translater_logger.error(f"翻译失败: {e}", exc_info=True) # exc_info=True for traceback
|
||
response_data["error"] = True
|
||
response_data["message"] = f"翻译过程中发生错误: {str(e)}"
|
||
finally:
|
||
current_translation_state["is_processing"] = False
|
||
if file: # Ensure file object exists
|
||
await file.close() # Close the UploadFile object
|
||
# Do not clear file_contents here as it's used by translate_bytes
|
||
# The content is in memory; if it were a temp file, you'd delete it here.
|
||
|
||
return JSONResponse(content=response_data)
|
||
|
||
|
||
# --- 下载接口 ---
|
||
@app.get("/download/markdown/{filename_with_ext}")
|
||
async def download_markdown_endpoint(filename_with_ext: str): # filename_with_ext from URL
|
||
# Use original_filename_stem from state to construct the expected filename for security/consistency
|
||
if current_translation_state["markdown_content"] and current_translation_state["original_filename_stem"]:
|
||
# Compare requested filename stem with stored stem if necessary, or just use stored stem
|
||
actual_filename = f"{current_translation_state['original_filename_stem']}_translated.md"
|
||
# if Path(filename_with_ext).stem != Path(actual_filename).stem:
|
||
# raise HTTPException(status_code=404, detail="文件名不匹配或内容不可用。")
|
||
|
||
return StreamingResponse(io.StringIO(current_translation_state["markdown_content"]), media_type="text/markdown",
|
||
headers={"Content-Disposition": f"attachment; filename=\"{actual_filename}\""})
|
||
raise HTTPException(status_code=404, detail="无 Markdown 翻译内容可用。")
|
||
|
||
|
||
@app.get("/download/html/{filename_with_ext}")
|
||
async def download_html_endpoint(filename_with_ext: str):
|
||
if current_translation_state["html_content"] and current_translation_state["original_filename_stem"]:
|
||
actual_filename = f"{current_translation_state['original_filename_stem']}_translated.html"
|
||
# if Path(filename_with_ext).stem != Path(actual_filename).stem:
|
||
# raise HTTPException(status_code=404, detail="文件名不匹配或内容不可用。")
|
||
|
||
return HTMLResponse(content=current_translation_state["html_content"], media_type="text/html",
|
||
headers={"Content-Disposition": f"attachment; filename=\"{actual_filename}\""})
|
||
raise HTTPException(status_code=404, detail="无 HTML 翻译内容可用。")
|
||
|
||
|
||
# --- Uvicorn 启动 ---
|
||
if __name__ == "__main__":
|
||
print("正在启动 FastAPI 文档翻译服务 (使用 asyncio.Queue 和 SSE)...") # Updated message
|
||
print("请访问 http://127.0.0.1:8010")
|
||
# Consider adding reload_dirs if you have other modules like docutranslate in development
|
||
uvicorn.run(app, host="127.0.0.1", port=8010) # Removed reload=True for this specific test
|
||
# Add it back if you are actively developing |