Update document_generator.py

document_generator.py

@@ -281,6 +281,31 @@ async def generate_document_endpoint(request: DocumentRequest):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
+
+@app.post("/generate-document-test", response_model=DocumentResponse)
+async def test_generate_document_endpoint(request: DocumentRequest):
+    try:
+        # Load JSON document from file
+        json_path = os.path.join("output/document_generator", "ai-chatbot-prd.json")
+        with open(json_path, "r") as json_file:
+            json_document = json.load(json_file)
+
+        # Load Markdown document from file
+        md_path = os.path.join("output/document_generator", "ai-chatbot-prd.md")
+        with open(md_path, "r") as md_file:
+            markdown_document = md_file.read()
+
+        return DocumentResponse(
+            json_document=json_document,
+            markdown_document=markdown_document
+        )
+    except FileNotFoundError:
+        raise HTTPException(status_code=404, detail="Test files not found")
+    except json.JSONDecodeError:
+        raise HTTPException(status_code=500, detail="Error parsing JSON file")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
 class CacheTestResponse(BaseModel):
     result: str
     execution_time: float