Update app.py

app.py

@@ -877,21 +877,20 @@ with ui.navset_card_tab(id="tab"):
             
             # plot the comparison
             fig = None
-            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
+            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()