Update app.py

app.py

@@ -45,21 +45,6 @@ class UxState(str, Enum):
     CAPTCHA = "captcha"
     TRAIN = "train"
 
-# Command-line arguments to control some stuff for easier local testing.
-# Eventually may want to move everything into functions and have a
-# if __name__ == "main" setup instead of everything inline.
-parser = argparse.ArgumentParser()
-parser.add_argument(
-    "--dry-run", action="store_true",
-    help="Skip sending train request to backend server.",
-)
-parser.add_argument(
-    "--train-endpoint-url", default=None,
-    help="URL of backend server to send train request to. If None, use hardcoded banana setup.",
-)
-cli_args = parser.parse_args()
-
-
 def setup_session_state():
     if "key" not in st.session_state:
         st.session_state["key"] = uuid.uuid4().hex
@@ -257,21 +242,10 @@ def verify_captcha(captcha_id, captcha_response):
     return {"error": "Failed to verify captcha"}
 
 def train_model(model_inputs):
-    if cli_args.dry_run:
-        logger.info("Skipping model training since --dry-run is enabled.")
-        logger.info(f"model_inputs: {model_inputs}")
-        return
-
-    if cli_args.train_endpoint_url is None:
-        # Use banana backend
-        api_key = "03cdd72e-5c04-4207-bd6a-fd5712c1740e"
-        model_key = "1a3b4ce5-164f-4efb-9f4a-c2ad3a930d0b"
-        st.markdown(str(model_inputs))
-        _ = banana.run(api_key, model_key, model_inputs)
-    else:
-        # Send request directly to specified url
-        _ = requests.post(cli_args.train_endpoint_url, json=model_inputs)
-
+    api_key = os.environ.get('API_KEY')
+    model_key = os.environ.get('MODEL_KEY')
+    st.markdown(str(model_inputs))
+    _ = banana.run(api_key, model_key, model_inputs)
 
 def run_login():
     user_email_input = st.empty()