Update app.py

app.py

@@ -328,42 +328,62 @@ def create_gradio_interface() -> gr.Blocks:
 
     def format_markdown_report(report_text: str) -> str:
         """Format the report text with proper Markdown and styling"""
-        lines = report_text.split('\n')
-        formatted_lines = []
-        
-        for line in lines:
-            # Handle headings
-            if '=============================' in line:
-                continue
-            if '-----------------------------' in line:
-                formatted_lines.append('---')
-                continue
-                
-            # Handle star ratings
-            if '[★★★★☆]' in line:
-                line = line.replace('[★★★★☆]', '`★★★★☆`')
+        # Extract just the report content using markers
+        try:
+            start_marker = "📊 E-COMMERCE ANALYSIS REPORT"
+            end_marker = "TECHNICAL INSIGHTS"
             
-            # Handle section titles with emojis
-            if any(emoji in line for emoji in ['📊', '🔍', '📝', '🔄', '💰', '📈', '🔧']):
-                formatted_lines.append('\n**' + line.strip() + '**\n')
-                continue
-                
-            # Handle bullet points
-            if line.strip().startswith('•'):
-                line = line.replace('•', '-')
-                
-            formatted_lines.append(line)
-        
-        # Join the lines and add final formatting
-        formatted_text = '\n'.join(formatted_lines)
-        
-        # Add spacing after section breaks
-        formatted_text = formatted_text.replace('---\n', '---\n\n')
-        
-        # Ensure bullet points are properly spaced
-        formatted_text = formatted_text.replace('\n-', '\n\n-')
+            # Find the report content
+            start_idx = report_text.find(start_marker)
+            if start_idx == -1:
+                return "Error: Could not find report content"
+            
+            # Extract and clean the report
+            report_lines = []
+            in_report = False
+            for line in report_text.split('\n'):
+                if start_marker in line:
+                    in_report = True
+                    report_lines.append("# " + line.strip())
+                    continue
+                    
+                if in_report:
+                    # Skip empty lines
+                    if not line.strip():
+                        continue
+                        
+                    # Format section headers
+                    if any(emoji in line for emoji in ['🔍', '📝', '🔄', '💰', '📈', '🔧']):
+                        if ":" in line:
+                            title, score = line.split(":", 1)
+                            report_lines.append(f"\n## {title.strip()}")
+                            if score.strip():
+                                report_lines.append(f"**Score: {score.strip()}**\n")
+                        else:
+                            report_lines.append(f"\n## {line.strip()}\n")
+                        continue
+                    
+                    # Format bullet points
+                    if line.strip().startswith('•'):
+                        report_lines.append(line.replace('•', '-'))
+                        continue
+                        
+                    # Add other lines as is
+                    report_lines.append(line.strip())
         
-        return formatted_text
+            # Join the lines and clean up the formatting
+            report_text = '\n'.join(report_lines)
+            
+            # Clean up multiple blank lines
+            report_text = '\n'.join(line for line, _ in itertools.groupby(report_text.split('\n')))
+            
+            # Ensure proper spacing around headers and bullet points
+            report_text = re.sub(r'\n#{1,2} ', r'\n\n# ', report_text)
+            report_text = re.sub(r'\n- ', r'\n\n- ', report_text)
+            
+            return report_text
+        except Exception as e:
+            return f"Error formatting report: {str(e)}"
 
     async def run_analysis(api_key: str,
                          website_url: str,
@@ -437,10 +457,8 @@ def create_gradio_interface() -> gr.Blocks:
                     value="Results will appear here...",
                     label="Analysis Results",
                     elem_classes="analysis-output",
-                    sanitize_html=False,  # Allow HTML for custom formatting
-                    line_breaks=True,     # Preserve line breaks
-                    header_links=False,   # Disable header links for cleaner look
-                    show_copy_button=True # Allow users to copy the report
+                    show_copy_button=True,
+                    line_breaks=True
                 )
         
         analyze_button.click(