Update main.py

main.py

@@ -112,18 +112,19 @@ async def convert_to_pdf(request: HTMLRequest):
         raise HTTPException(status_code=500, detail=str(e))
 
 
+from fastapi import FastAPI, HTTPException, Response
+from pydantic import BaseModel
 from html4docx import HtmlToDocx
-import io
-import tempfile
 import os
 
+
 class HTMLInput(BaseModel):
     html: str
 
 # Define the path to the temporary folder
 TEMP_FOLDER = "/app/temp"
 
-@app.post("/html_to_docx")
+@app.post("/convert")
 async def convert_html_to_docx(input_data: HTMLInput):
     temp_filename = None
     try:
@@ -144,14 +145,18 @@ async def convert_html_to_docx(input_data: HTMLInput):
             file_contents = file.read()
         
         # Return the DOCX file as a response
-        return fastapi.responses.Response(
+        return Response(
             content=file_contents, 
             media_type="application/vnd.openxmlformats-officedocument.wordprocessingml.document",
-            headers={"Content-Disposition": "attachment; filename=html-converted.docx"}
+            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)
+            os.remove(temp_filename)
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)