Update ui/insights_ui_generator.py

ui/insights_ui_generator.py

@@ -6,13 +6,23 @@ import pandas as pd
 logger = logging.getLogger(__name__)
 
 
-def format_report_for_display(report_data: Optional[pd.Series]) -> str:
+def format_report_for_display(report_data: Optional[pd.Series]) -> Dict[str, str]:
     """
-    Enhanced report formatting with Medium-style typography and layout.
-    This function replaces the original format_report_for_display.
+    Enhanced report formatting. Returns a dictionary with separate HTML for the header
+    and Markdown for the report body, allowing flexible rendering.
+
+    Args:
+        report_data: A pandas Series representing a single row from the agentic analysis DataFrame.
+                     It should contain 'report_text', 'report_type', and 'Created Date'.
+
+    Returns:
+        A dictionary with 'header_html' and 'body_markdown' keys.
+        'header_html' contains the HTML for the report title and subtitle.
+        'body_markdown' contains the raw Markdown text for the report body.
+        If data is invalid, it returns empty state HTML for both.
     """
     if report_data is None or report_data.empty:
-        return """
+        empty_state_html = """
         <div class="empty-state">
             <div class="empty-state-icon">📄</div>
             <div class="empty-state-title">Report Not Available</div>
@@ -22,8 +32,10 @@ def format_report_for_display(report_data: Optional[pd.Series]) -> str:
             </div>
         </div>
         """
+        return {'header_html': '', 'body_markdown': empty_state_html}
 
     # Extract report data
+    # Ensure 'report_text' is treated as raw text that will be interpreted as Markdown
     report_text = report_data.get('report_text', '*Report content not found.*')
     report_type = report_data.get('report_type')
     created_date_str = report_data.get('Created Date')
@@ -46,44 +58,32 @@ def format_report_for_display(report_data: Optional[pd.Series]) -> str:
         logger.error(f"Error generating dynamic report title: {e}")
         # In case of an error, default title and no subtitle will be used.
 
-    # Format the report content with the new HTML structure
-    formatted_content = f"""
-    <div class="report-header">
+    # Format the report header HTML
+    header_html = f"""
+    <div class="report-header-content">
         <h1>{title}</h1>
         {f'<p style="font-size: 1.1rem; color: #6b6b6b; margin-bottom: 0;">{subtitle}</p>' if subtitle else ''}
     </div>
-    
-    <div class="report-body">
-        {report_text.strip()}
-    </div>
     """
     
-    # Wrap the entire content in a div with the 'markdown' class, which Gradio Markdown component expects
-    # and which is styled by the provided CSS.
-    return formatted_content
+    # The report_text itself should be the raw Markdown for the body
+    body_markdown = report_text.strip()
+    
+    return {'header_html': header_html, 'body_markdown': body_markdown}
 
 
 def format_report_to_markdown(report_string: Optional[str]) -> str:
     """
-    Formats the comprehensive analysis report string into a displayable Markdown format.
-    This can be enhanced to add more structure if the report has implicit sections.
-
-    Args:
-        report_string: The raw text report from the orchestrator.
-
-    Returns:
-        A Markdown formatted string.
+    This function was previously for general Markdown formatting.
+    Given the new structure where 'format_report_for_display' handles the split,
+    this function might become redundant or repurposed.
+    Keeping it for now but noting its potential redundancy depending on upstream calls.
     """
     if not report_string or not report_string.strip():
         return "## Comprehensive Analysis Report\n\n*No analysis report was generated, or an error occurred during its generation.*"
     
     # Simple formatting for now. Could be enhanced (e.g., looking for patterns like "Section X:" to make them H3)
-    # Ensure paragraphs are separated. Replace multiple newlines with double newlines for Markdown paragraphs.
-    # report_string_cleaned = re.sub(r'\n\s*\n', '\n\n', report_string.strip())
-    
     formatted_report = f"## Comprehensive Analysis Report\n\n{report_string.strip()}"
-    # You might add more sophisticated parsing here if your LLM output for the report
-    # has a consistent structure that can be converted to richer Markdown.
     return formatted_report
 
 def extract_key_results_for_selection(