Update src/streamlit_app.py

src/streamlit_app.py

@@ -73,7 +73,9 @@ Results are presented in easy-to-read tables, visualized in an interactive tree
 
 **Supported Languages:** English
 
-**Technical issues:** If your connection times out, please refresh the page or reopen the app's URL. For any errors or inquiries, please contact us at [email protected]""")
+**Technical issues:** If your connection times out, please refresh the page or reopen the app's URL. 
+
+For any errors or inquiries, please contact us at [email protected]""")
 
 with st.sidebar:
     st.write("Use the following code to embed the StoryCraft web app on your website. Feel free to adjust the width and height values to fit your page.")
@@ -290,13 +292,21 @@ if st.session_state.show_results:
                 mime="application/zip",
             )
         
+        # Moved the Comet ML logging block to the correct place
         if comet_initialized:
-            experiment.log_figure(figure=fig_treemap, figure_name="entity_treemap_categories")
-            experiment.end()
+            # Check if df is not empty before trying to log
+            if not df.empty:
+                experiment = Experiment(
+                    api_key=COMET_API_KEY,
+                    workspace=COMET_WORKSPACE,
+                    project_name=COMET_PROJECT_NAME,
+                )
+                experiment.log_figure(figure=fig_treemap, figure_name="entity_treemap_categories")
+                experiment.end()
             
     else:  # If df is empty
         st.warning("No entities were found in the provided text.")
-    
+        
     st.text("")
     st.text("")
     st.info(f"Results processed in **{st.session_state.elapsed_time:.2f} seconds**.")