fix: Clean up UI

app.py

@@ -3,29 +3,59 @@ from predict_chromosome import predict_and_write
 
 
 st.title("DeepLoop")
+st.write("Looking for some example data?")
+st.write(
+    "http://hiview.case.edu/ssz20/tmp.HiCorr.ref/HiCorr_test_data/HiCorr_output.tar.gz"
+)
+st.write("Use hg19, HindIII, chr11, CPGZ and LoopDenoise for the Demo data")
 
 
 #########
 # INPUT #
 #########
-
-# TODO A drop down of models for different depths
-# depth = st.selectbox("Select Model", ["Model 1", "Model 2", "Model 3"])
-# HACK
-chromosome = "chr11"
+HiCorr_data = st.file_uploader("Upload HiCorr Data", type=["tar.gz", "zip"])
+training_set = st.selectbox("Select Training Set", ["CPGZ", "H9"], index=0)
+depth = st.selectbox("Select Depth", ["LoopDenoise", "50M", "101K"], index=0)
+# TODO Throw a warning that h9 only has LoopDenoise and 100M
+chromosome = st.selectbox("Select Chromosome", ["chr11"])
 prefix = "results"
+genome = st.selectbox(
+    "Reference Genome",
+    [
+        "hg19",
+        "hg38",
+        "mm10",
+    ],
+)
+digestion_enzyme = st.selectbox(
+    "Digestion Enzyme",
+    [
+        "HindIII",
+        "Arima",
+        # FIXME "microC_5kb_bin"
+        "DPNII",
+    ],
+    index=0,
+)
+# TODO Add the other toggles
+# small_matrix_size
+# step_size
+# max_dist
+# dummy
+# val_cols
+# keep_zeros
+
 
 # Load the model from hugging face
 from huggingface_hub import from_pretrained_keras
 
-model = from_pretrained_keras("funlab/DeepLoop-CPGZ-LoopDenoise")
+model = from_pretrained_keras(f"funlab/DeepLoop-{training_set}-{depth}")
 
 from huggingface_hub import snapshot_download
 
 anchors = snapshot_download(
-    repo_id="funlab/hg19_HindIII_anchor_bed", repo_type="dataset"
+    repo_id=f"funlab/{genome}_{digestion_enzyme}_anchor_bed", repo_type="dataset"
 )
-HiCorr_data = snapshot_download("funlab/HiCorr_test_data", repo_type="dataset")
 
 import os