Make project selectable

app.py

@@ -2,7 +2,7 @@ import gradio as gr, pytesseract, cv2, os
 from annif_client import AnnifClient
 
 
-def process(image: str, proj: str, lang: str = "eng"):
+def process(image: str, project_num: int, lang: str = "eng"):
     try:
         img = cv2.imread(image)
         gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
@@ -11,12 +11,8 @@ def process(image: str, proj: str, lang: str = "eng"):
         )[1]
         text = pytesseract.image_to_string(threshold_img, lang=lang)
         os.remove(image)
-        results = annif.suggest(
-            project_id='yso-en',
-            text=text
-        )
-        results = [
-            f"{res['label']} {res['score']}" for res in results]
+        results = annif.suggest(project_id=proj_ids[project_num], text=text)
+        results = [f"{res['label']} {res['score']}" for res in results]
         return text, results
     except Exception as e:
         return str(e), []
@@ -24,14 +20,15 @@ def process(image: str, proj: str, lang: str = "eng"):
 
 annif = AnnifClient()
 langs = pytesseract.get_languages()
-projs = [project['name'] for project in annif.projects]
+proj_ids = [project["project_id"] for project in annif.projects]
+proj_names = [project["name"] for project in annif.projects]
 
 
 interface = gr.Interface(
     process,
     [
         gr.Image(type="filepath", mirror_webcam=False),
-        gr.Dropdown(label="Select Project", choices=projs, type="value"),
+        gr.Dropdown(label="Select Project", choices=proj_names, type="index"),
         gr.Dropdown(label="Select Language", choices=langs, type="value"),
     ],
     outputs=["text", "text"],