update app.py

app.py

@@ -2,7 +2,6 @@ import os
 import tempfile
 from fastapi import FastAPI, UploadFile, File, HTTPException, Header
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import FileResponse, JSONResponse
 from typhoon_ocr import ocr_document
 from PyPDF2 import PdfReader
 from docx import Document
@@ -69,10 +68,10 @@ async def ocr_endpoint(
     x_api_key: str | None = Header(None)
 ):
     if API_KEY and x_api_key != API_KEY:
-        return JSONResponse(status_code=401, content={"detail": "API key ผิดพ่อง"})
+        raise HTTPException(status_code=401, detail="API key ผิดพ่อง")
 
     if page_num < 0:
-        return JSONResponse(status_code=400, content={"detail": "เลขหน้าต้องเป็น 0 หรือบวกเท่านั้น"})
+        raise HTTPException(status_code=400, detail="เลขหน้าต้องเป็น 0 หรือบวกเท่านั้น")
 
     suffix = os.path.splitext(file.filename)[-1].lower()
     content = await file.read()
@@ -83,33 +82,15 @@ async def ocr_endpoint(
     try:
         md, md_path, docx_path = process_file_with_typhoon(tmp_path, task_type, page_num)
     except Exception as e:
-        return JSONResponse(status_code=400, content={"detail": str(e)})
+        raise HTTPException(status_code=400, detail=str(e))
 
-    # ส่งข้อมูล Markdown + ลิงก์ดาวน์โหลดไฟล์
     return {
         "markdown": md,
-        "md_download_url": f"/download/md/{os.path.basename(md_path)}",
-        "docx_download_url": f"/download/docx/{os.path.basename(docx_path)}",
+        "md_file": md_path,
+        "docx_file": docx_path,
         "task_type": task_type,
     }
 
-@app.get("/download/md/{filename}")
-async def download_md(filename: str):
-    tmp_dir = tempfile.gettempdir()
-    file_path = os.path.join(tmp_dir, filename)
-    if not os.path.exists(file_path):
-        raise HTTPException(status_code=404, detail="ไฟล์ .md ไม่เจอ")
-    return FileResponse(file_path, media_type="text/markdown", filename="ocr_extraction_peace.md")
-
-@app.get("/download/docx/{filename}")
-async def download_docx(filename: str):
-    tmp_dir = tempfile.gettempdir()
-    file_path = os.path.join(tmp_dir, filename)
-    if not os.path.exists(file_path):
-        raise HTTPException(status_code=404, detail="ไฟล์ .docx ไม่เจอ")
-    return FileResponse(file_path, media_type="application/vnd.openxmlformats-officedocument.wordprocessingml.document", filename="ocr_extraction_peace.docx")
-
-
 def gradio_handler(file, task_type, page_num):
     if file is None:
         return "❌ อัปโหลดไฟล์ก่อน....", None, None