multi index support + limit increase

app.py

@@ -25,7 +25,7 @@ def search_and_display(seq, n_res, index_selection):
 
 
 def limit_n_results(n):
-    return max(min(n, 20), 1)
+    return max(min(n, 100), 1)
 
 
 def update_dropdown_menu(search_res):
@@ -117,7 +117,7 @@ with gr.Blocks() as demo:
                     n_results = gr.Number(10, label="N Results")
                     index_selector = gr.Dropdown(
                         choices=available_indexes,
-                        value="Pdb",
+                        value="PDB",
                         multiselect=False,
                         visible=True,
                         label="Index",

get_index.py

@@ -24,6 +24,6 @@ def get_engines(index_repo: str, model_repo: str):
     subindex_paths = glob(str(index_path / "*/"))
     engines = {}
     for subindex_path in subindex_paths:
-        subindex_name = os.path.basename(subindex_path).title()
+        subindex_name = os.path.basename(subindex_path)
         engines[subindex_name] = ProteinSearchEngine(data_path=Path(subindex_path))
     return engines