fix lookup

app.py

@@ -36,7 +36,6 @@ estimated_expression_meta = estimated_expression_meta["estimated_expression_meta
 estimated_expression = datasets.load_dataset(
     path = "maomlab/CryptoCEN",
     data_files = {"estimated_expression": "estimated_expression_matrix.parquet"})
-print(f"dir(estimated_expression): {dir(estimated_expression)}")
 estimated_expression = estimated_expression["estimated_expression"].to_pandas()
 
 #DEBUG
@@ -59,18 +58,18 @@ with col2:
 
 # check the user input
 try:
-    cnag_id_1 = h99_transcript_annotations.loc[h99_transcript_annotations["gene_id"] == gene_id_1]["cnag_id"]
+    cnag_id_1 = h99_transcript_annotations.loc[h99_transcript_annotations["gene_id"] == gene_id_1]["cnag_id"].values
 except:
     st.error(f"Unable to locate cnag_id for Gene ID 1: {gene_id_1}, it should be of the form 'CNAG_######'")
 
 try:
-    cnag_id_2 = h99_transcript_annotations.loc[h99_transcript_annotations["gene_id"] == gene_id_2]["cnag_id"]
+    cnag_id_2 = h99_transcript_annotations.loc[h99_transcript_annotations["gene_id"] == gene_id_2]["cnag_id"].values
 except:
     st.error(f"Unable to locate cnag_id for Gene ID 2: {gene_id_2}, it should be of the form 'CNAG_######'")
 
 chart_data = pd.DataFrame({
-    "expression_1": np.log10(estimated_expression.loc[estimated_expression.index == cnag_id_1].to_numpy()[0] + 1),
-    "expression_2": np.log10(estimated_expression.loc[estimated_expression_index == cnag_id_2].to_numpy()[0] + 1),
+    "expression_1": np.log10(estimated_expression.loc[h99_transcript_annotations["gene_id"] == gene_id_1].to_numpy()[0] + 1),
+    "expression_2": np.log10(estimated_expression.loc[h99_transcript_annotations["gene_id"] == gene_id_2].to_numpy()[0] + 1),
     "run_accession": estimated_expression.columns,
     "run_accession_meta": estimated_expression_meta["run_accession"],
     "study_accession": estimated_expression_meta["study_accession"]})