Update document_generator.py

document_generator.py

@@ -456,8 +456,8 @@ class JsonDocumentResponse(BaseModel):
 class JsonDocumentRequest(BaseModel):
     query: str
     template: bool = False
-    images: Optional[List[UploadFile]] = None
-    documents: Optional[List[UploadFile]] = None
+    images: Optional[List[Union[UploadFile, str]]] = None
+    documents: Optional[List[Union[UploadFile, str]]] = None
     conversation_id: str = ""
 
     class Config:
@@ -535,8 +535,10 @@ async def generate_document_stream(document_generator: DocumentGenerator, docume
 
     db_manager.update_database("elevatics", query, markdown_document)
 
-@router.post("/generate-document/markdown-stream")
-async def generate_markdown_document_stream_endpoint(request: MarkdownDocumentRequest):
+@router.post("/generate-document/json", response_model=JsonDocumentResponse)
+async def generate_document_outline_endpoint(
+    request: JsonDocumentRequest = Depends()
+):
     ai_client = AIClient()
     document_generator = DocumentGenerator(ai_client)