working

app.py

@@ -37,7 +37,6 @@ parameters_section = sidebar.expander("Parameters", expanded=False)
 tree_width = parameters_section.number_input("Tree Width", min_value=1, max_value=5, value=1)
 tree_depth = parameters_section.number_input("Tree Depth", min_value=1, max_value=8, value=3)
 iterations = parameters_section.number_input("Iterations", min_value=1, max_value=4, value=2)
-key = st.sidebar.text_input("Enter your OpenAI Api Key:", type="password")
 sidebar.markdown('<hr style="margin-top: 0.5rem; margin-bottom: 0.5rem;">', unsafe_allow_html=True)
 
 with sidebar:
@@ -51,7 +50,6 @@ def make_args(instruction, tree_depth, tree_width, iterations):
 
     parser.add_argument("--strategy", default="mcts", help="Strategy to use")
     parser.add_argument("--language", default="py", help="Programming language")
-    parser.add_argument("--model", default="samba", help="Model type")
     parser.add_argument("--max_iters", default=iterations, help="Maximum iterations")
     parser.add_argument("--instruction", default=instruction, help="Instruction text")
     parser.add_argument("--verbose", action="store_true", help="Verbose output")
@@ -66,7 +64,6 @@ def make_args(instruction, tree_depth, tree_width, iterations):
 
 def run_querry():
     if user_input:
-        
         # Create a new container for each subsequent message
         runtime_container.write("Initiating process...")
 
@@ -78,6 +75,7 @@ def run_querry():
 
             with st.spinner('Running...'):
                 args = make_args(user_input, tree_depth, tree_width, iterations)
+                setattr(args, 'model', 'samba')
                 # main call
                 response = lats_main(args)
 
@@ -101,6 +99,5 @@ with chat_col:
             fail = True
         
         if (not fail):
-            openai.api_key = key
             run_querry()