(feat) add deploy using streamlit

frontend/components/deploy_v2_with_controllers.py

@@ -0,0 +1,110 @@
+import time
+import streamlit as st
+import pandas as pd
+
+from CONFIG import BACKEND_API_HOST, BACKEND_API_PORT
+from backend.services.backend_api_client import BackendAPIClient
+
+
+class LaunchV2WithControllers:
+    DEFAULT_COLUMNS = [
+        "id", "controller_name", "controller_type", "connector_name",
+        "trading_pair", "total_amount_quote", "max_loss_quote", "stop_loss",
+        "take_profit", "trailing_stop", "time_limit", "selected"
+    ]
+
+    def __init__(self):
+        self._backend_api_client = BackendAPIClient.get_instance(host=BACKEND_API_HOST, port=BACKEND_API_PORT)
+        self._controller_configs_available = self._backend_api_client.get_all_controllers_config()
+        self._controller_config_selected = []
+        self._bot_name = None
+        self._image_name = "hummingbot/hummingbot:latest"
+        self._credentials = "master_account"
+
+    def _set_bot_name(self, bot_name):
+        self._bot_name = bot_name
+
+    def _set_image_name(self, image_name):
+        self._image_name = image_name
+
+    def _set_credentials(self, credentials):
+        self._credentials = credentials
+
+    def launch_new_bot(self):
+        if self._bot_name and self._image_name and self._controller_config_selected:
+            start_time_str = time.strftime("%Y.%m.%d_%H.%M")
+            bot_name = f"{self._bot_name}-{start_time_str}"
+            script_config = {
+                "name": bot_name,
+                "content": {
+                    "markets": {},
+                    "candles_config": [],
+                    "controllers_config": self._controller_config_selected,
+                    "config_update_interval": 20,
+                    "script_file_name": "v2_with_controllers.py",
+                    "time_to_cash_out": None,
+                }
+            }
+
+            self._backend_api_client.add_script_config(script_config)
+            deploy_config = {
+                "instance_name": bot_name,
+                "script": "v2_generic_with_cash_out.py",
+                "script_config": bot_name + ".yml",
+                "image": self._image_name,
+                "credentials_profile": self._credentials,
+            }
+            self._backend_api_client.create_hummingbot_instance(deploy_config)
+            with st.spinner('Starting Bot... This process may take a few seconds'):
+                time.sleep(3)
+        else:
+            st.warning("You need to define the bot name and select the controllers configs "
+                       "that you want to deploy.")
+
+    def __call__(self):
+        st.write("#### Select the controllers configs that you want to deploy.")
+        all_controllers_config = self._controller_configs_available
+        data = []
+        for config in all_controllers_config:
+            connector_name = config.get("connector_name", "Unknown")
+            trading_pair = config.get("trading_pair", "Unknown")
+            total_amount_quote = config.get("total_amount_quote", 0)
+            stop_loss = config.get("stop_loss", 0)
+            take_profit = config.get("take_profit", 0)
+            trailing_stop = config.get("trailing_stop", {"activation_price": 0, "trailing_delta": 0})
+            time_limit = config.get("time_limit", 0)
+            data.append({
+                "selected": False,
+                "id": config["id"],
+                "controller_name": config["controller_name"],
+                "controller_type": config["controller_type"],
+                "connector_name": connector_name,
+                "trading_pair": trading_pair,
+                "total_amount_quote": total_amount_quote,
+                "max_loss_quote": total_amount_quote * stop_loss / 2,
+                "stop_loss": f"{stop_loss:.2%}",
+                "take_profit": f"{take_profit:.2%}",
+                "trailing_stop": f"{trailing_stop['activation_price']:.2%} / {trailing_stop['trailing_delta']:.2%}",
+                "time_limit": time_limit,
+            })
+
+        df = pd.DataFrame(data)
+
+        edited_df = st.data_editor(df, hide_index=True)
+
+        self._controller_config_selected = [f"{config}.yml" for config in edited_df[edited_df["selected"]]["id"].tolist()]
+        st.write(self._controller_config_selected)
+        c1, c2, c3, c4 = st.columns([1, 1, 1, 0.3])
+        with c1:
+            self._bot_name = st.text_input("Instance Name")
+        with c2:
+            available_images = self._backend_api_client.get_available_images("hummingbot")
+            self._image_name = st.selectbox("Hummingbot Image", available_images,
+                                            index=available_images.index("hummingbot/hummingbot:latest"))
+        with c3:
+            available_credentials = self._backend_api_client.get_available_credentials()
+            self._credentials = st.selectbox("Credentials", available_credentials, index=0)
+        with c4:
+            deploy_button = st.button("Deploy Bot")
+        if deploy_button:
+            self.launch_new_bot()