Update doc_maker.py

doc_maker.py

@@ -8,11 +8,9 @@ from datetime import datetime
 import uuid
 import tempfile
 
-router = APIRouter(prefix="/api/v1")
+router = APIRouter()  # Add prefix here
 
-BASE_URL = "https://pvanand-doc-maker.hf.space/api/v1/"
-
-# Use system temp directory for file storage
+BASE_URL = "https://pvanand-doc-maker.hf.space/api/v1"
 DOCS_DIR = os.path.join(tempfile.gettempdir(), "doc_maker_documents")
 
 class DocumentPart(BaseModel):
@@ -27,15 +25,13 @@ class DocumentResponse(BaseModel):
     download_url: Optional[str] = None
 
 def get_download_url(file_id: str) -> str:
-    return f"{BASE_URL}download/{file_id}"
+    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(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:
-    """Update or create document with new page content"""
     filepath = get_document_path(doc_id)
     
     if os.path.exists(filepath):
@@ -75,7 +71,7 @@ async def add_document_part(part: DocumentPart):
             detail=f"Error updating document: {str(e)}"
         )
 
-@router.get("/document/{doc_id}", response_model=Dict)
+@router.get("/document/{doc_id}")
 async def get_document(doc_id: str):
     filepath = get_document_path(doc_id)
     if not os.path.exists(filepath):