Update main.py

main.py

@@ -1,37 +1,28 @@
 import os
 import glob
-import re
-import pandas as pd, numpy as np
 import comtypes.client
 import docx
-from docx.document import Document
-from docx.oxml.table import CT_Tbl
-from docx.oxml.text.paragraph import CT_P
-from docx.table import _Cell, Table
-from docx.text.paragraph import Paragraph
-from flask import Flask, request, send_file
-from werkzeug.utils import secure_filename
+from fastapi import FastAPI, File, UploadFile
+from pydantic import BaseModel
+from starlette.responses import FileResponse
 import zipfile
 import io
 
-app = Flask(__name__)
+app = FastAPI()
 
 UPLOAD_FOLDER = 'uploads/'
 if not os.path.exists(UPLOAD_FOLDER):
     os.makedirs(UPLOAD_FOLDER)
 
-@app.route('/convert', methods=['POST'])
-def convert_pdfs():
-    if 'pdf_files' not in request.files:
-        return 'No file part', 400
-
-    pdf_files = request.files.getlist('pdf_files')
+@app.post('/convert')
+async def convert_pdfs(pdf_files: list[UploadFile] = File(...)):
     if not pdf_files:
-        return 'No selected file', 400
+        return {"detail": "No selected file"}
 
     for pdf_file in pdf_files:
-        filename = secure_filename(pdf_file.filename)
-        pdf_file.save(os.path.join(UPLOAD_FOLDER, filename))
+        filename = pdf_file.filename
+        with open(os.path.join(UPLOAD_FOLDER, filename), "wb") as f:
+            f.write(await pdf_file.read())
 
     path_pdf = UPLOAD_FOLDER
     path_docx = UPLOAD_FOLDER
@@ -58,7 +49,8 @@ def convert_pdfs():
                     zf.write(filepath, filename)
 
     memory_file.seek(0)
-    return send_file(memory_file, attachment_filename='converted_docx.zip', as_attachment=True)
+    return FileResponse(memory_file, filename='converted_docx.zip', media_type='application/zip')
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=5000)
+    import uvicorn
+    uvicorn.run(app, host='0.0.0.0', port=7860)