Spaces:

habulaj
/

pdf-api

Sleeping

App Files Files Community

habulaj commited on Dec 10, 2025

Commit

2ce54f6

verified ·

1 Parent(s): 51eb27e

Update app.py

Browse files

Files changed (1) hide show

app.py +58 -38

app.py CHANGED Viewed

@@ -2,8 +2,10 @@ from fastapi import FastAPI, HTTPException
 from fastapi.responses import Response
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from typing import Optional
-import httpx
 app = FastAPI()
@@ -16,9 +18,6 @@ app.add_middleware(
     allow_headers=["*"],
 )
-# Endpoint da API externa
-EXTERNAL_API_URL = "https://habulaj-pdf-api.hf.space"
 # Modelo para criar posts (gerar imagem a partir de HTML)
 class PostRequest(BaseModel):
     html: str
@@ -34,7 +33,6 @@ async def create_post(request: PostRequest):
     """
     Cria uma imagem a partir de HTML para posts.
     Recebe HTML e retorna uma imagem PNG.
-    Faz proxy para a API externa: https://habulaj-pdf-api.hf.space
     Exemplo de uso:
     {
@@ -43,49 +41,71 @@ async def create_post(request: PostRequest):
         "height": 1350
     }
     """
     try:
-        # Prepara o payload para a API externa
-        payload = {
-            "html": request.html,
-            "width": request.width,
-            "height": request.height
-        }
-        # Faz requisição para a API externa
-        async with httpx.AsyncClient(timeout=60.0) as client:
-            response = await client.post(
-                f"{EXTERNAL_API_URL}/create-post",
-                json=payload,
-                headers={"Content-Type": "application/json"}
-            )
-            if response.status_code != 200:
-                error_message = f"Erro na API externa: {response.status_code}"
-                try:
-                    error_data = response.json()
-                    if "detail" in error_data:
-                        error_message = error_data["detail"]
-                    elif "message" in error_data:
-                        error_message = error_data["message"]
-                except:
-                    error_message = f"Erro na API externa: {response.status_code} {response.text}"
-                raise HTTPException(status_code=response.status_code, detail=error_message)
-            # Retorna a imagem recebida da API externa
             return Response(
-                content=response.content,
                 media_type="image/png",
                 headers={
                     "Content-Disposition": "attachment; filename=post.png"
                 }
             )
-    except httpx.TimeoutException:
-        raise HTTPException(status_code=504, detail="Timeout ao comunicar com a API externa")
-    except httpx.RequestError as e:
-        raise HTTPException(status_code=502, detail=f"Erro ao comunicar com a API externa: {str(e)}")
-    except HTTPException:
-        raise
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Erro ao gerar imagem do post: {str(e)}")

 from fastapi.responses import Response
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
+from playwright.sync_api import sync_playwright
 from typing import Optional
+import tempfile
+import os
 app = FastAPI()
     allow_headers=["*"],
 )
 # Modelo para criar posts (gerar imagem a partir de HTML)
 class PostRequest(BaseModel):
     html: str
     """
     Cria uma imagem a partir de HTML para posts.
     Recebe HTML e retorna uma imagem PNG.
     Exemplo de uso:
     {
         "height": 1350
     }
     """
+    html_file = None
     try:
+        # Garantir que o HTML está completo
+        html_content = request.html
+        # Se o HTML não tiver estrutura completa, adicionar
+        if "<!DOCTYPE" not in html_content and "<html" not in html_content:
+            html_content = f"""<!DOCTYPE html>
+<html lang="pt-BR">
+<head>
+  <meta charset="UTF-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+</head>
+<body>
+{html_content}
+</body>
+</html>"""
+        # Criar arquivo temporário para o HTML
+        with tempfile.NamedTemporaryFile(mode='w', suffix='.html', delete=False, encoding='utf-8') as f:
+            f.write(html_content)
+            html_file = f.name
+        try:
+            # Usar Playwright para renderizar HTML e gerar screenshot
+            with sync_playwright() as p:
+                browser = p.chromium.launch(headless=True)
+                page = browser.new_page(viewport={'width': request.width, 'height': request.height})
+                # Carrega o HTML do arquivo temporário
+                page.goto(f'file://{html_file}')
+                # Espera o conteúdo carregar (especialmente imagens base64)
+                page.wait_for_timeout(2000)  # Aguarda 2 segundos para garantir que imagens e fontes carreguem
+                # Tira screenshot
+                screenshot_bytes = page.screenshot(full_page=True, type='png')
+                browser.close()
+            # Remove arquivo temporário
+            if os.path.exists(html_file):
+                os.unlink(html_file)
+                html_file = None
+            # Retorna a imagem
             return Response(
+                content=screenshot_bytes,
                 media_type="image/png",
                 headers={
                     "Content-Disposition": "attachment; filename=post.png"
                 }
             )
+        except Exception as e:
+            # Remove arquivo temporário em caso de erro
+            if html_file and os.path.exists(html_file):
+                os.unlink(html_file)
+            raise e
     except Exception as e:
+        # Garantir limpeza do arquivo temporário
+        if html_file and os.path.exists(html_file):
+            try:
+                os.unlink(html_file)
+            except:
+                pass
         raise HTTPException(status_code=500, detail=f"Erro ao gerar imagem do post: {str(e)}")