Create doc_maker.py

doc_maker.py

@@ -0,0 +1,99 @@
+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
+
+router = APIRouter()
+
+BASE_URL = "https://pvanand-doc-maker.hf.space/api/v1/"
+
+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:
+    """Get the full path for a document file"""
+    os.makedirs("documents", exist_ok=True)
+    return f"documents/{doc_id}.json"
+
+def update_document(doc_id: str, page_no: int, content: str) -> Dict:
+    """Update or create document with new page content"""
+    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/part", 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}", response_model=Dict)
+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("/document/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"}