Update app.py

app.py

@@ -877,20 +877,21 @@ with ui.navset_card_tab(id="tab"):
             
             # plot the comparison
             fig = None
-            if input.plot_type_macro() == "2D Line":
-                fig = plot_2d_comparison(grouped, grouped.index)
-            if input.plot_type_macro() == "ColorSquare":
-                filtered_df = df.groupby('Organism_Name').apply(filter_and_select).reset_index(drop=True)
-                fig = plot_color_square(filtered_df['Sequence'], filtered_df['Organism_Name'].unique())
-            if input.plot_type_macro() == "Wens Method":
-                fig = wens_method_heatmap(df, df['Organism_Name'].unique())
-            if input.plot_type_macro() == "Chaos Game Representation":
-                filtered_df = df.groupby('Organism_Name').apply(filter_and_select).reset_index(drop=True)
-                fig = plot_fcgr(filtered_df['Sequence'], df['Organism_Name'].unique())
-            if input.plot_type_macro() == "Persistant Homology":
-                filtered_df = df.groupby('Organism_Name').apply(filter_and_select).reset_index(drop=True)
-                fig = plot_persistence_homology(filtered_df['Sequence'], filtered_df['Organism_Name'])
-            return fig
+            if input.tab() =='Viral Macrostructure':
+                if input.plot_type_macro() == "2D Line":
+                    fig = plot_2d_comparison(grouped, grouped.index)
+                if input.plot_type_macro() == "ColorSquare":
+                    filtered_df = df.groupby('Organism_Name').apply(filter_and_select).reset_index(drop=True)
+                    fig = plot_color_square(filtered_df['Sequence'], filtered_df['Organism_Name'].unique())
+                if input.plot_type_macro() == "Wens Method":
+                    fig = wens_method_heatmap(df, df['Organism_Name'].unique())
+                if input.plot_type_macro() == "Chaos Game Representation":
+                    filtered_df = df.groupby('Organism_Name').apply(filter_and_select).reset_index(drop=True)
+                    fig = plot_fcgr(filtered_df['Sequence'], df['Organism_Name'].unique())
+                if input.plot_type_macro() == "Persistant Homology":
+                    filtered_df = df.groupby('Organism_Name').apply(filter_and_select).reset_index(drop=True)
+                    fig = plot_persistence_homology(filtered_df['Sequence'], filtered_df['Organism_Name'])
+                return fig
    # with ui.nav_panel("Viral Model"):
   #      gr.load("models/Hack90/virus_pythia_31_1024").launch()