Update main.py

main.py

@@ -157,6 +157,39 @@ async def convert_html_to_docx(input_data: HTMLInput):
         if temp_filename and os.path.exists(temp_filename):
             os.remove(temp_filename)
 
+@app.post("/html_to_docx")
+async def convert_html_to_docx(input_data: HTMLRequest):
+    temp_filename = None
+    try:
+        # Create a new HtmlToDocx parser
+        parser = HtmlToDocx()
+        
+        # Parse the HTML string to DOCX
+        docx = parser.parse_html_string(input_data.html_content)
+        
+        # Create a unique filename in the temporary folder
+        temp_filename = os.path.join(TEMP_FOLDER, f"temp_{os.urandom(8).hex()}.docx")
+        
+        # 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 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: remove the temporary file
+        if temp_filename and os.path.exists(temp_filename):
+            os.remove(temp_filename)
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)