Introduced UserInput class to take care of userinput

utilities/py/ui_elements.py

@@ -0,0 +1,75 @@
+import pandas as pd
+
+import streamlit as st
+from utilities.py.data_management import CompData
+
+class UserInputData:
+    def __init__(self, company_list, opt_method, start_date, init_invest, ef_parameter = None):
+        self.company_list = company_list
+        self.opt_method = opt_method
+        self.ef_parameter = ef_parameter
+        self.start_date = start_date
+        self.init_invest = init_invest
+
+
+class UserInput:
+
+    def __init__(self, company_data: CompData):
+        """
+        Class that renders the user selection (company, optimization technique, etc.)
+        """
+        self.comp_data = company_data
+        self.ef_parameter_input = None
+        self.company_list_input = None
+        self.opt_method_input = None
+        self.start_date_input = None
+        self.initial_investment_input = None
+
+    def company_selection(self):
+        self.company_list_input = st.multiselect(
+            "Select Multiple Companies", self.comp_data.company_names, default=None
+        )
+
+    def opt_method_selection(self):
+        self.opt_method_input = st.selectbox(
+            "Choose an optimization method accordingly",
+            (
+                "Efficient Frontier",
+                "Hierarchical Risk Parity",
+            ),
+        )
+        
+        if self.opt_method_input == "Efficient Frontier":
+            self.ef_parameter_input = st.selectbox(
+                "Choose an optimization parameter accordingly",
+                (
+                    "Maximum Sharpe Ratio",
+                    "Efficient Risk",
+                    "Minimum Volatility",
+                    "Efficient Return",
+                ),
+            )
+
+    def start_date(self):
+        self.start_date_input = st.date_input(
+            "Start Date",
+            format="YYYY-MM-DD",
+            value=pd.Timestamp("1947-08-15"),
+            max_value=pd.Timestamp.now(),
+        )
+
+    def initial_investment(self):
+        self.innit_invest_input = st.number_input("How much would you want to invest?", value=45000)
+
+    def get_selected_comp_ids(self):
+        if self.company_list_input is not None:
+            return self.comp_data.comp_name_to_id(self.company_list_input)
+        print("WARINING: Selected company ids accessed, eventhough company not yet rendered in UI")
+        return None
+
+    def get_user_input_data(self) -> UserInputData:
+        return UserInputData(self.company_list_input,
+                             self.opt_method_input,
+                             self.start_date_input,
+                             self.innit_invest_input,
+                             self.ef_parameter_input)