| from fastapi import APIRouter, HTTPException |
| from fastapi.responses import FileResponse |
| from pydantic import BaseModel |
| from typing import Dict, Optional |
| import json |
| import os |
| from datetime import datetime |
| import uuid |
| import tempfile |
|
|
| router = APIRouter() |
|
|
| BASE_URL = "https://pvanand-doc-maker.hf.space/api/v1" |
| DOCS_DIR = os.path.join(tempfile.gettempdir(), "doc_maker_documents") |
|
|
| class DocumentPart(BaseModel): |
| pageNo: int |
| content: str |
| docId: Optional[str] = None |
|
|
| class DocumentResponse(BaseModel): |
| docId: str |
| message: str |
| totalPages: int |
| download_url: Optional[str] = None |
|
|
| def get_download_url(file_id: str) -> str: |
| return f"{BASE_URL}/download/{file_id}" |
|
|
| def get_document_path(doc_id: str) -> str: |
| os.makedirs(DOCS_DIR, exist_ok=True) |
| return os.path.join(DOCS_DIR, f"{doc_id}.json") |
|
|
| def update_document(doc_id: str, page_no: int, content: str) -> Dict: |
| filepath = get_document_path(doc_id) |
| |
| if os.path.exists(filepath): |
| with open(filepath, 'r', encoding='utf-8') as f: |
| doc = json.load(f) |
| doc['updatedAt'] = datetime.utcnow().isoformat() |
| else: |
| doc = { |
| 'docId': doc_id, |
| 'createdAt': datetime.utcnow().isoformat(), |
| 'updatedAt': datetime.utcnow().isoformat(), |
| 'content': {} |
| } |
| |
| doc['content'][str(page_no)] = content |
| |
| with open(filepath, 'w', encoding='utf-8') as f: |
| json.dump(doc, f, ensure_ascii=False, indent=2) |
| |
| return doc |
|
|
| @router.post("/document/update", response_model=DocumentResponse) |
| async def add_document_part(part: DocumentPart): |
| doc_id = part.docId or str(uuid.uuid4()) |
| |
| try: |
| doc = update_document(doc_id, part.pageNo, part.content) |
| return DocumentResponse( |
| docId=doc_id, |
| message=f"Page {part.pageNo} added successfully", |
| totalPages=len(doc['content']), |
| download_url=get_download_url(doc_id) |
| ) |
| except Exception as e: |
| raise HTTPException( |
| status_code=500, |
| detail=f"Error updating document: {str(e)}" |
| ) |
|
|
| @router.get("/document/{doc_id}") |
| async def get_document(doc_id: str): |
| filepath = get_document_path(doc_id) |
| if not os.path.exists(filepath): |
| raise HTTPException(status_code=404, detail="Document not found") |
| |
| with open(filepath, 'r', encoding='utf-8') as f: |
| return json.load(f) |
|
|
| @router.get("/download/{doc_id}") |
| async def download_document(doc_id: str): |
| filepath = get_document_path(doc_id) |
| if not os.path.exists(filepath): |
| raise HTTPException(status_code=404, detail="Document not found") |
| |
| return FileResponse( |
| filepath, |
| media_type='application/json', |
| filename=f"document_{doc_id}.json" |
| ) |
|
|
| @router.delete("/document/{doc_id}") |
| async def delete_document(doc_id: str): |
| filepath = get_document_path(doc_id) |
| if not os.path.exists(filepath): |
| raise HTTPException(status_code=404, detail="Document not found") |
| |
| os.remove(filepath) |
| return {"message": "Document deleted successfully"} |
