Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,3 +1,4 @@
+
 import os
 os.environ['HF_HOME'] = '/tmp'
 import time
@@ -395,14 +396,17 @@ def generate_html_report(df, text_input, elapsed_time, df_topic_data):
            <p><strong>Generated At:</strong> {time.strftime('%Y-%m-%d %H:%M:%S')}</p>
            <p><strong>Processing Time:</strong> {elapsed_time:.2f} seconds</p>
        </div>
+       <!-- Section 1: Original Text & Highlighted Entities -->
        <h2>1. Analyzed Text & Extracted Entities</h2>
        <h3>Original Text with Highlighted Entities</h3>
        <div class="highlighted-text-container">
            {highlighted_text_html}
        </div>
        
+       <!-- Section 2: Full Extracted Entities Table -->
        <h2>2. Full Extracted Entities Table</h2>
        {entity_table_html}
+       <!-- Section 3: Visualizations (Treemap, Pie, Bar Charts) -->
        <h2>3. Data Visualizations</h2>
        
        <h3>3.1 Entity Distribution Treemap</h3>
@@ -416,6 +420,7 @@ def generate_html_report(df, text_input, elapsed_time, df_topic_data):
        <h3>3.3 Entity Co-occurrence Network (Edges = Same Sentence)</h3>
        <div class="chart-box">{network_html}</div>
        
+       <!-- Section 4: Topic Modeling -->
        <h2>4. Topic Modeling (LDA on Entities)</h2>
        {topic_charts_html}
        
@@ -577,17 +582,18 @@ if st.button("Results"):
                     st.session_state.topic_results = perform_topic_modeling(
                         df, 
                         num_topics=2, 
-                       num_top_words=N_TOP_WORDS_TO_USE
+                        num_top_words=N_TOP_WORDS_TO_USE
                     )
                     
                     if comet_initialized:
+                        # FIX APPLIED HERE: Corrected indentation for the following lines
                         experiment = Experiment(api_key=COMET_API_KEY, workspace=COMET_WORKSPACE, project_name=COMET_PROJECT_NAME)
                         experiment.log_parameter("input_text", text)
                         experiment.log_table("predicted_entities", df)
                         experiment.end()
                 else:
                     st.session_state.results_df = pd.DataFrame()
-                   st.session_state.topic_results = None
+                    st.session_state.topic_results = None
                 
                 end_time = time.time()
                 st.session_state.elapsed_time = end_time - start_time
@@ -740,6 +746,3 @@ if st.session_state.show_results:
                 file_name=f"entity_topic_report_{time.strftime('%Y%m%d_%H%M%S')}.html",
                 mime="text/html",
             )
- 
-
-