Update app.py

app.py

@@ -554,12 +554,11 @@ with tab1:
             st.subheader("Calculated Molecular Properties")
             st.dataframe(res1.get('props_df', pd.DataFrame()), use_container_width=True)
         with p1_tabs[2]:
-                    st.subheader("Molecular Properties Dashboard")
-                    if res1.get('props_plot'):
-                        # Display the plot and then close it to prevent memory leaks
-                        st.pyplot(res1['props_plot'], clear_figure=True)
-                        # Explicitly close the figure after displaying
-                        plt.close(res1['props_plot'])
+            st.subheader("Molecular Properties Dashboard")
+            if res1.get('props_plot'):
+                # Display the plot using clear_figure to manage memory without causing reruns
+                st.pyplot(res1['props_plot'], clear_figure=True)
+                # FIX: Removed the explicit plt.close() call which caused the flickering
 
 # ===== TAB 2: LEAD GENERATION & OPTIMIZATION =====
 with tab2:
@@ -702,12 +701,11 @@ with tab4:
         res4 = st.session_state.results_p4
         p4_tabs = st.tabs(["Pharmacovigilance Analysis", "Regulatory & Ethical Frameworks"])
         with p4_tabs[0]:
-                    st.subheader("Simulated Adverse Event Analysis")
-                    if 'plot_bar' in res4 and res4['plot_bar'] is not None:
-                        st.pyplot(res4['plot_bar'], clear_figure=True)
-                        # Explicitly close the figure after displaying
-                        plt.close(res4['plot_bar'])
-                    st.dataframe(res4['rwd_df'], use_container_width=True)
+            st.subheader("Simulated Adverse Event Analysis")
+            if 'plot_bar' in res4 and res4['plot_bar'] is not None:
+                st.pyplot(res4['plot_bar'], clear_figure=True)
+                # FIX: Removed the explicit plt.close() call which caused the flickering
+            st.dataframe(res4['rwd_df'], use_container_width=True)
             
         with p4_tabs[1]:
             col1, col2 = st.columns(2)
@@ -716,4 +714,4 @@ with tab4:
                 st.dataframe(res4.get('reg_df', pd.DataFrame()), use_container_width=True)
             with col2:
                 st.subheader("Ethical Framework for AI in Healthcare")
-                st.dataframe(res4.get('eth_df', pd.DataFrame()), use_container_width=True)
+                st.dataframe(res4.get('eth_df', pd.DataFrame()), use_container_width=True)