Spaces:
Sleeping
Sleeping
File size: 2,993 Bytes
99eaf9e ebb7ae8 99eaf9e ebb7ae8 99eaf9e ebb7ae8 99eaf9e ebb7ae8 99eaf9e ebb7ae8 99eaf9e |
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 |
from fastapi import FastAPI
from fastapi.responses import FileResponse, JSONResponse
from pydantic import BaseModel
import uuid
import os
import glob
import re
import pypandoc
app = FastAPI()
# Control de descargas por archivo
descargas = {}
class MarkdownInput(BaseModel):
content: str
# Reemplaza '---' por saltos de línea
def limpiar_lineas_hr(markdown_text: str) -> str:
return re.sub(r'^\s*---\s*$', '\n', markdown_text, flags=re.MULTILINE)
# Convierte \[...\] y \(...\) a $$...$$ y $...$
def normalizar_ecuaciones(md: str) -> str:
md = re.sub(r'\\\[\s*(.*?)\s*\\\]', r'$$\1$$', md, flags=re.DOTALL)
md = re.sub(r'\\\(\s*(.*?)\s*\\\)', r'$\1$', md, flags=re.DOTALL)
return md
def save_markdown_to_file(content: str, path: str):
with open(path, "w", encoding="utf-8") as f:
f.write(content)
def convert_markdown_to_docx(input_md: str, output_docx: str):
pypandoc.convert_file(
source_file=input_md,
to="docx",
format="md",
outputfile=output_docx,
extra_args=["--standalone"]
)
def limpieza():
for ext in ["*.docx", "*.md"]:
for file in glob.glob(ext):
try:
os.remove(file)
except Exception as e:
print(f"No se pudo eliminar {file}: {e}")
@app.post("/convert/")
def convert(data: MarkdownInput):
try:
limpieza()
uid = str(uuid.uuid4())
input_md = f"{uid}.md"
output_docx = f"{uid}.docx"
contenido_limpio = limpiar_lineas_hr(data.content)
contenido_limpio = normalizar_ecuaciones(contenido_limpio)
save_markdown_to_file(contenido_limpio, input_md)
convert_markdown_to_docx(input_md, output_docx)
os.remove(input_md)
descargas[uid] = 0
return {
"message": "Archivo generado exitosamente, para descargar acceder a la URL",
"url": f"https://jairodanielmt-markdown-docx.hf.space/download/{uid}"
}
except Exception as e:
return JSONResponse(content={"error": str(e)}, status_code=500)
@app.get("/download/{file_id}")
def download(file_id: str):
output_docx = f"{file_id}.docx"
if not os.path.exists(output_docx):
return JSONResponse(
content={"error": "El archivo no existe o ya fue eliminado."},
status_code=404
)
if file_id not in descargas:
return JSONResponse(
content={"error": "ID de archivo no válido."},
status_code=400
)
if descargas[file_id] >= 2:
os.remove(output_docx)
del descargas[file_id]
return JSONResponse(
content={"error": "El archivo ya no está disponible. Genere uno nuevo."},
status_code=410
)
descargas[file_id] += 1
return FileResponse(
path=output_docx,
filename="material_educativo.docx",
media_type="application/vnd.openxmlformats-officedocument.wordprocessingml.document",
)
|