Update app.py

app.py

@@ -11,7 +11,8 @@ from utils import (
     wens_method_heatmap,
     plot_fcgr,
     plot_persistence_homology,
-    plot_distrobutions
+    plot_distrobutions, 
+    process_data_sub_specie
 )
 import os
 import seaborn as sns
@@ -148,7 +149,40 @@ with ui.navset_card_tab(id="tab"):
                 ax.set_xlabel("Basepair")
                 ax.set_ylabel("Variance")
                 return ax
+########################################################### 2D Sub-Specie #####################################################################
+    with ui.nav_panel("Virus Sub-Specie"):
+        ui.panel_title("Can we create sub-specie based on 2d representation? How does their 2D representation vary?")
+        with ui.layout_columns():
+            with ui.card():
+                ui.input_selectize("virus_selector_2", "Select your viruses:", virus_new, multiple=False, selected=None)
+    
+                ui.input_selectize(
+                    "plot_type_distro_sub",
+                    "Select plot type:",
+                    ["2D sub-specie - nominal", "2D sub-specie - detrended"],
+                    multiple=False,
+                    selected="2D sub-specie - nominal",
+                )
+                ui.input_slider("variance", "variance from 2d rep", 1, 5, 3)
+                
+                
+
+        @render.plot()
+        def plot_sub_specie():
+            import seaborn as sns 
+            plot_type = input.plot_type_distro_sub()
+            if plot_type == "2D sub-specie - nominal":
+                df = process_data_sub_specie(MASTER_DF, input.virus_selector_2(), input.variance())
+                df = df.explode('two_d').copy()
+                df['x'] = np.array((df['two_d'].tolist()))[:,0]
+                df['y'] = np.array((df['two_d'].tolist()))[:,1]
+                ax = sns.scatterplot(data=df, x='x',y='y', hue='group')  
+                ax.set_title("Sub-specie")
+                # ax.set_xlabel("")
+                # ax.set_ylabel("Density")
+                return ax
 
+    
     with ui.nav_panel("Viral Microstructure"):
         ui.panel_title("Kmer Distribution")
         with ui.layout_columns():