Update main.py

main.py

@@ -109,4 +109,49 @@ async def convert_to_pdf(request: HTMLRequest):
         pdf = pdfkit.from_string(request.html_content, False, options=options)
         return Response(content=pdf, media_type="application/pdf")
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+from html4docx import HtmlToDocx
+import io
+import tempfile
+import os
+
+class HTMLInput(BaseModel):
+    html: str
+
+@app.post("/convert")
+async def convert_html_to_docx(input_data: HTMLInput):
+    temp_file = None
+    try:
+        # Create a new HtmlToDocx parser
+        parser = HtmlToDocx()
+        
+        # Parse the HTML string to DOCX
+        docx = parser.parse_html_string(input_data.html)
+        
+        # Create a temporary file
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.docx')
+        temp_filename = temp_file.name
+        
+        # Save the DOCX to the temporary file
+        docx.save(temp_filename)
+        
+        # Open the file and read its contents
+        with open(temp_filename, 'rb') as file:
+            file_contents = file.read()
+        
+        # Return the DOCX file as a response
+        return fastapi.responses.Response(
+            content=file_contents, 
+            media_type="application/vnd.openxmlformats-officedocument.wordprocessingml.document",
+            headers={"Content-Disposition": "attachment; filename=converted.docx"}
+        )
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+    finally:
+        # Clean up: close and remove the temporary file
+        if temp_file:
+            temp_file.close()
+            os.unlink(temp_file.name)
+