Upload folder using huggingface_hub

app.py

@@ -1,10 +1,11 @@
-from fastapi import FastAPI, HTTPException
-from fastapi.responses import HTMLResponse
+from fastapi import FastAPI, HTTPException, BackgroundTasks
+from fastapi.responses import HTMLResponse, FileResponse
 from pydantic import BaseModel
 from pathlib import Path
 import uuid
 import shutil
-from validator import validate_rmd
+import subprocess
+from validator import validate_rmd, render_rmd_format
 
 app = FastAPI(
     title="RMD Validator API",
@@ -156,6 +157,107 @@ async def validate_rmd_and_cleanup(rmd_data: RmdContent):
         shutil.rmtree(temp_dir, ignore_errors=True)
 
 
+def _prepare_rmd(rmd_data, temp_dir):
+    if not rmd_data.filename.endswith('.rmd'):
+        rmd_data.filename += '.rmd'
+    rmd_file = temp_dir / rmd_data.filename
+    rmd_file.write_text(rmd_data.content, encoding='utf-8')
+    return rmd_file
+
+
+@app.post("/download/html")
+async def download_html(rmd_data: RmdContent, background_tasks: BackgroundTasks):
+    request_id = str(uuid.uuid4())
+    temp_dir = Path(f"/app/temp/{request_id}")
+    temp_dir.mkdir(parents=True, exist_ok=True)
+
+    try:
+        rmd_file = _prepare_rmd(rmd_data, temp_dir)
+        render_rmd_format(str(rmd_file), 'html_document')
+
+        html_file = temp_dir / f"{rmd_file.stem}.html"
+        if not html_file.exists():
+            raise HTTPException(status_code=500, detail="HTML non generato")
+
+        background_tasks.add_task(shutil.rmtree, temp_dir, True)
+        return FileResponse(
+            path=str(html_file),
+            filename=f"{rmd_file.stem}.html",
+            media_type="text/html",
+        )
+
+    except subprocess.CalledProcessError as e:
+        shutil.rmtree(temp_dir, ignore_errors=True)
+        raise HTTPException(status_code=422, detail=f"Errore rendering: {e.stderr}")
+    except HTTPException:
+        raise
+    except Exception as e:
+        shutil.rmtree(temp_dir, ignore_errors=True)
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+@app.post("/download/pdf")
+async def download_pdf(rmd_data: RmdContent, background_tasks: BackgroundTasks):
+    request_id = str(uuid.uuid4())
+    temp_dir = Path(f"/app/temp/{request_id}")
+    temp_dir.mkdir(parents=True, exist_ok=True)
+
+    try:
+        rmd_file = _prepare_rmd(rmd_data, temp_dir)
+        render_rmd_format(str(rmd_file), 'pdf_document')
+
+        pdf_file = temp_dir / f"{rmd_file.stem}.pdf"
+        if not pdf_file.exists():
+            raise HTTPException(status_code=500, detail="PDF non generato")
+
+        background_tasks.add_task(shutil.rmtree, temp_dir, True)
+        return FileResponse(
+            path=str(pdf_file),
+            filename=f"{rmd_file.stem}.pdf",
+            media_type="application/pdf",
+        )
+
+    except subprocess.CalledProcessError as e:
+        shutil.rmtree(temp_dir, ignore_errors=True)
+        raise HTTPException(status_code=422, detail=f"Errore rendering: {e.stderr}")
+    except HTTPException:
+        raise
+    except Exception as e:
+        shutil.rmtree(temp_dir, ignore_errors=True)
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+@app.post("/download/tex")
+async def download_tex(rmd_data: RmdContent, background_tasks: BackgroundTasks):
+    request_id = str(uuid.uuid4())
+    temp_dir = Path(f"/app/temp/{request_id}")
+    temp_dir.mkdir(parents=True, exist_ok=True)
+
+    try:
+        rmd_file = _prepare_rmd(rmd_data, temp_dir)
+        render_rmd_format(str(rmd_file), 'latex_document')
+
+        tex_file = temp_dir / f"{rmd_file.stem}.tex"
+        if not tex_file.exists():
+            raise HTTPException(status_code=500, detail="LaTeX non generato")
+
+        background_tasks.add_task(shutil.rmtree, temp_dir, True)
+        return FileResponse(
+            path=str(tex_file),
+            filename=f"{rmd_file.stem}.tex",
+            media_type="application/x-tex",
+        )
+
+    except subprocess.CalledProcessError as e:
+        shutil.rmtree(temp_dir, ignore_errors=True)
+        raise HTTPException(status_code=422, detail=f"Errore rendering: {e.stderr}")
+    except HTTPException:
+        raise
+    except Exception as e:
+        shutil.rmtree(temp_dir, ignore_errors=True)
+        raise HTTPException(status_code=500, detail=str(e))
+
+
 @app.delete("/cleanup/{request_id}")
 async def cleanup_request(request_id: str):
     temp_dir = Path(f"/app/temp/{request_id}")

validator.py

@@ -49,3 +49,29 @@ def validate_rmd(path):
             f.write(error_message)
 
     return status, error_message if status == 404 else "OK"
+
+
+def render_rmd_format(path, output_format):
+    rmd_path = Path(path)
+    if not rmd_path.is_file():
+        raise FileNotFoundError(f"File non trovato: {path}")
+
+    report_dir = rmd_path.parent
+
+    render_cmd = [
+        "Rscript", "--vanilla", "-e",
+        f"rmarkdown::render("
+        f"input = '{rmd_path.as_posix()}', "
+        f"output_format = '{output_format}', "
+        f"output_dir = '{report_dir.as_posix()}')"
+    ]
+
+    result = subprocess.run(
+        render_cmd,
+        check=True,
+        capture_output=True,
+        text=True,
+        encoding='utf-8'
+    )
+
+    return result