Update app.py

app.py

@@ -23,7 +23,7 @@ def stop_generation():
     is_stopped = True
     return "Generation stopped."
 
-def CTXGen(X0, X1, X2, τ, g_num, model_name):
+def CTXGen(X0, X3, X1, X2, τ, g_num, model_name):
     global is_stopped
     is_stopped = False
 
@@ -36,7 +36,6 @@ def CTXGen(X0, X1, X2, τ, g_num, model_name):
     model = torch.load(save_path, map_location=torch.device('cpu'))
     model = model.to(device)
 
-    X3 = X0
     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}")]
@@ -174,6 +173,7 @@ with gr.Blocks() as demo:
     gr.Markdown("# Conotoxin Optimization Generation")
     with gr.Row():
         X0 = gr.Textbox(label="conotoxin")
+        X3 = gr.Textbox(label="Positions that needs optimization")
         X1 = gr.Dropdown(choices=['<α7>','<AChBP>','<α4β2>','<α3β4>','<Ca22>','<α3β2>', '<Na12>','<α9α10>','<K16>', '<α1β1γδ>', 
                                   '<K13>', '<α1BAR>', '<α1β1ε>', '<α1AAR>', '<GluN3A>', '<GluN2B>', '<α75HT3>', '<Na14>', 
                                   '<GluN2C>', '<NET>', '<NavBh>', '<α6β3β4>', '<Na11>', '<Ca13>', '<Ca12>', '<Na16>', '<α6α3β2>', 
@@ -192,7 +192,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         output_file = gr.File(label="Download generated conotoxins")
     
-    start_button.click(CTXGen, inputs=[X0, X1, X2, τ, g_num, model_name], outputs=[output_df, output_file])
+    start_button.click(CTXGen, inputs=[X0, X3, X1, X2, τ, g_num, model_name], outputs=[output_df, output_file])
     stop_button.click(stop_generation, outputs=None)
 
 demo.launch()