From 5dc5cf7988e55bcc994a166d148cf2737a3855ef Mon Sep 17 00:00:00 2001
From: Manuel Holtgrewe <manuel.holtgrewe@bih-charite.de>
Date: Wed, 13 Sep 2023 17:08:44 +0200
Subject: [PATCH] fix: properly parsing comma-separated list on REST API

---
 cada_prio/rest_server.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/cada_prio/rest_server.py b/cada_prio/rest_server.py
index 782ab86..1e60a9b 100644
--- a/cada_prio/rest_server.py
+++ b/cada_prio/rest_server.py
@@ -59,12 +59,14 @@ class PredictionResult(BaseModel):
 # Register endpoint for the prediction
 @app.get("/predict")
 async def handle_predict(
-    hpo_terms: typing.Annotated[typing.List[str], Query()],
-    genes: typing.Annotated[typing.Optional[typing.List[str]], Query()] = [],
+    hpo_terms: typing.Annotated[str, Query()],
+    genes: typing.Annotated[typing.Optional[str], Query()] = [],
 ):
+    hpo_terms_list = hpo_terms.split(",")
+    genes_list = genes.split(",")
     _, sorted_scores = predict.run_prediction(
-        hpo_terms,
-        genes,
+        hpo_terms_list,
+        genes_list,
         GLOBAL_STATIC["all_to_hgnc"],
         GLOBAL_STATIC["graph"],
         GLOBAL_STATIC["model"],