Update app.py

app.py

@@ -173,15 +173,26 @@ with ui.navset_card_tab(id="tab"):
             import seaborn as sns 
             plot_type = input.plot_type_distro_sub()
             if plot_type == "2D sub-specie - nominal":
-                df = pd.read_parquet('virus_ds.parquet')
+                df = pd.read_parquet('new_viral_dataset.parquet')
                 df = process_data_sub_specie(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")
+                fig, ax = plt.subplots()
+                # Get unique groups and assign colors
+                groups = df['group'].unique()
+                colors = plt.cm.rainbow(np.linspace(0, 1, len(groups)))
+                color_dict = dict(zip(groups, colors))
+            
+                # Iterate through rows and plot
+                for _, row in df.iterrows():
+                    x, y = zip(*row['two_d'])
+                    ax.plot(x, y, c=color_dict[row['group']])
+            
+                # Remove duplicate labels
+                handles, labels = ax.get_legend_handles_labels()
+                by_label = dict(zip(labels, handles))
+                #ax.legend(by_label.values(), by_label.keys())
+            
+                ax.set_title("Sub-specie"))
+    
                 return ax