Update main.py

main.py

@@ -1,14 +1,10 @@
-from fastapi import FastAPI, Query
-from pydantic import BaseModel
+from fastapi import FastAPI, Query, Response
 import requests
 import json
 
 app = FastAPI()
 
-class EatcHeadersResponse(BaseModel):
-    report: str
-
-@app.get("/eatc-headers", response_class=HTMLResponse)
+@app.get("/eatc-headers")
 async def get_eatc_headers(eatc_code: str = Query(..., description="EATC code to retrieve headers for")):
     url = f"https://donantes.eatcloud.info/api/abaco/eatc_dona_headers?eatc-code={eatc_code}"
     response = requests.get(url)
@@ -20,8 +16,8 @@ async def get_eatc_headers(eatc_code: str = Query(..., description="EATC code to
             
             report = f"<p>eatc_code: {eatc_code_value}</p><p>eatc_match_asignation_rule: {eatc_match_asignation_rule_value}</p>"
             
-            return HTMLResponse(content=report, status_code=200)
+            return Response(content=report, media_type="text/html")
         else:
-            return HTMLResponse(content="<p>Error: Invalid response from API</p>", status_code=400)
+            return Response(content="<p>Error: Invalid response from API</p>", media_type="text/html", status_code=400)
     else:
-        return HTMLResponse(content="<p>Error: API returned error status code</p>", status_code=400)
+        return Response(content="<p>Error: API returned error status code</p>", media_type="text/html", status_code=400)