🐛

app.py

@@ -24,6 +24,18 @@ class Response(BaseModel):
 
 app = FastAPI(title="toxicity-api")
 
+@app.get("/")
+def root():
+    return {
+        "message": "Toxicity Prediction API",
+        "endpoints": {
+            "/metadata": "GET - API metadata and capabilities",
+            "/healthz": "GET - Health check",
+            "/predict": "POST - Predict toxicity for SMILES"
+        },
+        "usage": "Send POST to /predict with {'smiles': ['your_smiles_here']} and Authorization header"
+    }
+
 @app.get("/metadata")
 def metadata():
     return {

tests/test_app.py

@@ -15,6 +15,21 @@ def client():
     return TestClient(app)
 
 
+class TestRootEndpoint:
+    
+    def test_root(self, client):
+        response = client.get("/")
+        assert response.status_code == 200
+        data = response.json()
+        
+        assert data["message"] == "Toxicity Prediction API"
+        assert "endpoints" in data
+        assert "usage" in data
+        assert "/metadata" in data["endpoints"]
+        assert "/healthz" in data["endpoints"]
+        assert "/predict" in data["endpoints"]
+
+
 class TestMetadataEndpoint:
     
     def test_metadata(self, client):