Spaces:
Running
Running
File size: 3,714 Bytes
a935d6c 2463d3e a935d6c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 | from fastapi import FastAPI, File, UploadFile, Form, HTTPException
from fastapi.responses import FileResponse, HTMLResponse
from fastapi.middleware.cors import CORSMiddleware
from fastapi import BackgroundTasks
from fastapi.responses import FileResponse
import os
import tempfile
import re
from pathlib import Path
# Import your conversion function
from statis import process_excel_to_word
app = FastAPI(title="QCM Converter API")
# Enable CORS for all origins (you can restrict this in production)
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
def validate_hex_color(color: str) -> bool:
"""Validate hex color format"""
pattern = r'^[0-9A-Fa-f]{6}$'
return bool(re.match(pattern, color))
@app.get("/", response_class=HTMLResponse)
async def root():
"""Serve the HTML interface"""
html_path = Path(__file__).parent / "index.html"
if html_path.exists():
return html_path.read_text()
return """
<html>
<body>
<h1>QCM Converter API</h1>
<p>Upload your Excel files at <a href="/docs">/docs</a></p>
</body>
</html>
"""
@app.post("/convert")
async def convert_file(
background_tasks: BackgroundTasks,
file: UploadFile = File(...),
use_two_columns: bool = Form(True),
add_separator_line: bool = Form(True),
theme_color: str = Form("5FFFDF")
):
"""
Convert Excel QCM file to Word document
Parameters:
- file: Excel file (.xlsx)
- use_two_columns: Use two-column layout
- add_separator_line: Add separator line between columns
- theme_color: Hex color code (without #) e.g., "5FFFDF"
"""
# Validate file extension
if not file.filename.endswith('.xlsx'):
raise HTTPException(status_code=400, detail="Only .xlsx files are supported")
# Validate color
if not validate_hex_color(theme_color):
raise HTTPException(
status_code=400,
detail="Invalid color format. Use 6-character hex code (e.g., '5FFFDF')"
)
with tempfile.NamedTemporaryFile(delete=False, suffix='.xlsx') as temp_input:
content = await file.read()
temp_input.write(content)
temp_input_path = temp_input.name
output_filename = file.filename.replace('.xlsx', '_converted.docx')
temp_output_path = tempfile.mktemp(suffix='.docx')
try:
process_excel_to_word(
excel_file_path=temp_input_path,
output_word_path=temp_output_path,
theme_hex=theme_color
)
# ✅ Schedule cleanup as a background task
background_tasks.add_task(cleanup_files, temp_input_path, temp_output_path)
return FileResponse(
temp_output_path,
media_type="application/vnd.openxmlformats-officedocument.wordprocessingml.document",
filename=output_filename,
background=None # <-- No lambda here
)
except Exception as e:
cleanup_files(temp_input_path, temp_output_path)
raise HTTPException(status_code=500, detail=f"Conversion failed: {str(e)}")
def cleanup_files(*file_paths):
"""Clean up temporary files"""
for file_path in file_paths:
try:
if os.path.exists(file_path):
os.unlink(file_path)
except Exception as e:
print(f"Error cleaning up {file_path}: {e}")
@app.get("/health")
async def health_check():
"""Health check endpoint"""
return {"status": "healthy", "message": "QCM Converter API is running"}
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=7860) |