debugging cuda issues and adding basic tests for api

Dockerfile

@@ -1,9 +1,10 @@
-FROM python:3.12-slim
+FROM nvidia/cuda:12.1.1-base-ubuntu22.04
 
-# Install CUDA dependencies
-RUN apt-get update && apt-get install -y \
-    cuda-cudart-12-1 \
-    cuda-libraries-12-1 \
+# Install Python 3.12
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3.12 \
+    python3-pip \
+    && ln -s /usr/bin/python3.12 /usr/bin/python \
     && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /code

answers.md

@@ -0,0 +1,13 @@
+1. **Technical Choices:** What influenced your decision on the specific tools and models used?
+
+
+2. **Entity Contextualization:** How did you approach the problem of providing context for each identified entity?
+
+
+3. **Error Handling:** Can you describe how your API handles potential errors?
+
+
+4. **Challenges and Learnings:** What were the top challenges faced, and what did you learn from them?
+
+
+5. **Improvement Propositions:** Given more time, what improvements or additional features would you consider adding?

tests/test_api.py

@@ -0,0 +1,43 @@
+import pytest
+from fastapi.testclient import TestClient
+from everycure.app import app
+import os
+import tempfile
+
+client = TestClient(app)
+
+def create_test_pdf():
+    # Create a temporary PDF file for testing
+    with tempfile.NamedTemporaryFile(suffix=".pdf", delete=False) as tmp:
+        tmp.write(b"%PDF-1.5\nTest PDF content")
+        return tmp.name
+
+@pytest.fixture
+def test_pdf():
+    pdf_path = create_test_pdf()
+    yield pdf_path
+    # Cleanup after test
+    os.unlink(pdf_path)
+
+
+def test_extract_entities_invalid_file():
+    # Test with non-PDF file
+    with tempfile.NamedTemporaryFile(suffix=".txt") as tmp:
+        tmp.write(b"Not a PDF file")
+        tmp.seek(0)
+        response = client.post(
+            "/extract",
+            files={"file": ("test.txt", tmp, "text/plain")}
+        )
+    
+    assert response.status_code == 400
+    assert "Invalid file type" in response.json()["detail"]
+
+def test_extract_entities_empty_file(test_pdf):
+    with open(test_pdf, "rb") as f:
+        response = client.post(
+            "/extract",
+            files={}  # No file provided
+        )
+    
+    assert response.status_code == 422  # FastAPI's validation error