Update main.py

main.py

@@ -120,9 +120,12 @@ import os
 class HTMLInput(BaseModel):
     html: str
 
+# Define the path to the temporary folder
+TEMP_FOLDER = "/app/temp"
+
 @app.post("/convert")
 async def convert_html_to_docx(input_data: HTMLInput):
-    temp_file = None
+    temp_filename = None
     try:
         # Create a new HtmlToDocx parser
         parser = HtmlToDocx()
@@ -130,9 +133,8 @@ async def convert_html_to_docx(input_data: HTMLInput):
         # 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
+        # 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)
@@ -150,8 +152,6 @@ async def convert_html_to_docx(input_data: HTMLInput):
     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)
-        
+        # Clean up: remove the temporary file
+        if temp_filename and os.path.exists(temp_filename):
+            os.remove(temp_filename)