add logging

Dockerfile

@@ -1,13 +1,19 @@
+# Use an official Python runtime as a parent image
 FROM python:3.9
 
+# Create a user and set the working directory
 RUN useradd -m -u 1000 user
-
 WORKDIR /app
 
+# Copy the requirements file and install dependencies
 COPY --chown=user ./requirements.txt requirements.txt
-
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
+# Copy the rest of the application code
 COPY --chown=user . /app
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Expose the required port
+EXPOSE 7860
+
+# Run the FastAPI application with logging
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860", "--log-level", "debug"]

app.py

@@ -1,8 +1,13 @@
+import logging
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from bs4 import BeautifulSoup
 from typing import List, Dict
 
+# Configure logging
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
+
 app = FastAPI()
 
 all_html_tags = {
@@ -27,6 +32,7 @@ class HTMLOutput(BaseModel):
 
 
 def extract_html_tags(html_code: str) -> Dict[str, List[str]]:
+    logger.debug(f"Extracting HTML tags from code: {html_code}")
     soup = BeautifulSoup(html_code, "html.parser")
     tags_used = {tag.name for tag in soup.find_all()}
     tags_not_used = all_html_tags - tags_used
@@ -40,6 +46,16 @@ def extract_html_tags(html_code: str) -> Dict[str, List[str]]:
 async def extract_tags(input: HTMLInput):
     try:
         result = extract_html_tags(input.html_code)
+        logger.debug(f"Extraction result: {result}")
         return HTMLOutput(**result)
     except Exception as e:
+        logger.error(f"Error during extraction: {e}")
         raise HTTPException(status_code=500, detail=str(e))
+
+# Add a simple root endpoint to confirm the server is running
+
+
+@app.get("/")
+async def root():
+    logger.debug("Root endpoint called")
+    return {"message": "FastAPI application is running"}