restoring banned units

Author: syedzayyan

components/dataloader/CompoundGetter.tsx

@@ -4,20 +4,9 @@ import LigandContext from "../../context/LigandContext";
 import Link from "next/link";
 import FAQComp from "../ui-comps/FAQComp";
 
-// whitelist from your counts, filtered by count > 500
-const VALID_COMBOS: Record<string, string[]> = {
-  B: ["IC50", "Ki", "Potency", "EC50"],          // Binding
-  F: ["Potency", "IC50", "EC50", "ED50", "AC50", "XC50"], // Functional
-  ADMET: ["IC50"],
-  T: ["IC50"],
-  // P, U have no units with count > 500 -> all units disabled
-};
-
-const ALL_UNITS = ["Ki", "IC50", "XC50", "EC50", "AC50", "Kd", "Potency", "ED50"];
-
 export default function CompoundGetter() {
   const [unit, setUnit] = useState("Ki");
-  const [binding, setBinding] = useState<keyof typeof VALID_COMBOS | string>("B");
+  const [binding, setBinding] = useState("B");
   const { target, setTarget } = useContext(TargetContext);
   const { ligand, setLigand } = useContext(LigandContext);
   const [ligandSearch, setLigandSearch] = useState(false);
@@ -29,7 +18,7 @@ export default function CompoundGetter() {
     setLoading(true);
 
     const chembl_url = "https://www.ebi.ac.uk";
-    const results: any[] = [{}];
+    const results = [{}];
 
     let nextUrl = chembl_url + url;
 
@@ -51,19 +40,17 @@ export default function CompoundGetter() {
     getFullActivityData(
       `/chembl/api/data/activity?format=json&target_chembl_id=${target.target_id}&type=${unit}&target_organism=Homo%20sapiens&assay_type=${binding}&relation==`,
     ).then((data) => {
-      data.map((x: any) => {
+      data.map((x) => {
         x[unit] = x["standard_value"];
         x["id"] = x["molecule_chembl_id"];
         delete x["standard_value"];
         return x;
       });
       setLigand(data);
     });
-    setTarget({ ...target, activity_columns: [unit], data_source: "chembl" });
+    setTarget({ ...target, activity_columns: [unit], data_source: "chembl"});
   }
 
-  const allowedUnits = VALID_COMBOS[binding] ?? [];
-
   return (
     <div
       style={{
@@ -87,20 +74,11 @@ export default function CompoundGetter() {
         . Since, Binding Assays are more prevalent with Ki being the preferred
         unit, they are the default.
       </FAQComp>
-
       <label htmlFor="input-assay-type">Assay Type</label>
       <select
-        id="input-assay-type"
         className="input"
-        value={binding}
         onChange={(e) => {
-          const newBinding = e.target.value;
-          setBinding(newBinding);
-
-          const newAllowed = VALID_COMBOS[newBinding] ?? [];
-          if (!newAllowed.includes(unit) && newAllowed.length > 0) {
-            setUnit(newAllowed[0]);
-          }
+          setBinding(e.target.value);
         }}
       >
         <option value="B">B (Binding)</option>
@@ -110,29 +88,25 @@ export default function CompoundGetter() {
         <option value="P">P (Physiochemical)</option>
         <option value="U">U (Unclassified)</option>
       </select>
-
       <label htmlFor="input-unit">Unit Type</label>
       <select
-        id="input-unit"
         className="input"
-        value={unit}
-        onChange={(e) => setUnit(e.target.value)}
+        onChange={(e) => {
+          setUnit(e.target.value);
+        }}
       >
-        {ALL_UNITS.map((u) => (
-          <option
-            key={u}
-            value={u}
-            disabled={!allowedUnits.includes(u)}
-          >
-            {u}
-          </option>
-        ))}
+        <option value="Ki">Ki</option>
+        <option value="IC50">IC50</option>
+        <option value="XC50">XC50</option>
+        <option value="EC50">EC50</option>
+        <option value="AC50">AC50</option>
+        <option value="Kd">Kd</option>
+        <option value="Potency">Potency</option>
+        <option value="ED50">ED50</option>
       </select>
-
       <button className="button" onClick={hehe}>
         Fetch Data
       </button>
-
       {loading && (
         <div>
           <progress
@@ -141,13 +115,13 @@ export default function CompoundGetter() {
             max={100}
             style={{ width: "100%", marginBottom: "10px" }}
           ></progress>
-          <br />
+          <br></br>
           <span style={{ textAlign: "center" }}>
             {Math.min(progress, 100).toFixed(2)} %
           </span>
         </div>
       )}
-      <br />
+      <br></br>
       {ligand.length > 0 ? (
         <div style={{ marginTop: "10px" }}>
           <Link className="button" href="/tools/preprocess/">
@@ -162,4 +136,4 @@ export default function CompoundGetter() {
       )}
     </div>
   );
-}
+}