Update app.py

app.py

@@ -37,13 +37,18 @@ def CTXGen(X1, X2, τ, g_num, length_range):
     is_stopped = False
     start, end = length_range
 
-    msa_data = pd.read_csv('conoData_C0.csv') 
-    msa = msa_data['Sequences'].tolist()
-    msa = [x for x in msa if x.startswith(f"{X1}|{X2}")]
-    msa = random.choice(msa)
-    X4 = msa.split("|")[3]
-    X5 = msa.split("|")[4]
-    X6 = msa.split("|")[5]
+    if X1!='X' and X2!='X':
+        msa_data = pd.read_csv('conoData_C0.csv') 
+        msa = msa_data['Sequences'].tolist()
+        msa = [x for x in msa if x.startswith(f"{X1}|{X2}")]
+        msa = random.choice(msa)
+        X4 = msa.split("|")[3]
+        X5 = msa.split("|")[4]
+        X6 = msa.split("|")[5]
+    else:
+        X4 = ''
+        X5 = ''
+        X6 = ''
 
     model.eval()
     with torch.no_grad():