Update app.py

app.py

@@ -332,7 +332,8 @@ with tab1:
 
     st.subheader("Top Predictions & Scores")
     if 'results_df' in st.session_state and not st.session_state.results_df.empty:
-        st.dataframe(st.session_state.results__df, use_container_width=True, hide_index=True)
+        # FIX: Corrected the typo from 'results__df' to 'results_df'
+        st.dataframe(st.session_state.results_df, use_container_width=True, hide_index=True)
     else:
         st.info("No valid predictions to display. Try a different input.")
         
@@ -353,13 +354,19 @@ with tab2:
         smiles_input_viewer = st.text_input("SMILES String", value="CC(=O)Oc1ccccc1C(=O)O") # Aspirin
         viewer_submit = st.form_submit_button("👁️ View Molecule", use_container_width=True)
 
+    # Initialize viewer_html in session state if it doesn't exist
+    if 'viewer_html' not in st.session_state:
+        st.session_state.viewer_html = ""
+        st.session_state.viewer_log = "Enter a SMILES string and click 'View Molecule'."
+
+
     if viewer_submit:
         with st.spinner("Generating visualization..."):
             html_view, log = visualize_molecule_2d_3d(smiles_input_viewer, "Molecule")
             st.session_state.viewer_html = html_view
             st.session_state.viewer_log = log
     
-    if 'viewer_html' in st.session_state:
+    if st.session_state.viewer_html:
         st.components.v1.html(st.session_state.viewer_html, height=450)
     
     with st.expander("Show Logs"):