feat: improve UI

src/utils/ui.py

@@ -5,6 +5,8 @@ UI components and helper functions for the Defensor application.
 import gradio as gr
 import time
 import traceback
+import uuid
+import re
 
 def create_fallback_ui(custom_css):
     """Create a fallback UI for when data loading fails."""
@@ -243,24 +245,70 @@ def format_progress_stages(stage, is_error=False):
     return ""
 
 def format_answer_html(answer, sources=None, reasoning_steps=None, debug=False):
-    """Format the answer into well-structured HTML."""
+    """Format the answer into well-structured HTML with improved Markdown rendering."""
+    
+    # Process Markdown more comprehensively
     answer_html = ""
     
-    # Convert answer text to HTML paragraphs
+    # Process paragraphs and add proper formatting
     for paragraph in answer.split("\n\n"):
-        if paragraph.strip():
-            # Check if it's a heading
-            if paragraph.startswith("# "):
-                heading = paragraph[2:].strip()
-                answer_html += f"<h2>{heading}</h2>"
-            elif paragraph.startswith("## "):
-                heading = paragraph[3:].strip()
-                answer_html += f"<h3>{heading}</h3>"
-            elif paragraph.startswith("### "):
-                heading = paragraph[4:].strip()
-                answer_html += f"<h4>{heading}</h4>"
-            else:
-                answer_html += f"<p>{paragraph}</p>"
+        if not paragraph.strip():
+            continue
+            
+        # Process headers
+        if paragraph.startswith("# "):
+            heading = paragraph[2:].strip()
+            answer_html += f"<h2>{heading}</h2>"
+        elif paragraph.startswith("## "):
+            heading = paragraph[3:].strip()
+            answer_html += f"<h3>{heading}</h3>"
+        elif paragraph.startswith("### "):
+            heading = paragraph[4:].strip()
+            answer_html += f"<h4>{heading}</h4>"
+        # Process lists
+        elif re.match(r'^\d+\.\s', paragraph):
+            # Ordered list
+            items = paragraph.split("\n")
+            answer_html += "<ol>"
+            for item in items:
+                if re.match(r'^\d+\.\s', item):
+                    item_content = re.sub(r'^\d+\.\s', '', item)
+                    answer_html += f"<li>{item_content}</li>"
+            answer_html += "</ol>"
+        elif re.match(r'^[*-]\s', paragraph):
+            # Unordered list
+            items = paragraph.split("\n")
+            answer_html += "<ul>"
+            for item in items:
+                if re.match(r'^[*-]\s', item):
+                    item_content = re.sub(r'^[*-]\s', '', item)
+                    answer_html += f"<li>{item_content}</li>"
+            answer_html += "</ul>"
+        # Process code blocks
+        elif paragraph.startswith("```") and paragraph.endswith("```"):
+            code = paragraph[3:-3].strip()
+            language = ""
+            if "\n" in code:
+                first_line = code.split("\n")[0].strip()
+                if first_line and not first_line.startswith("```"):
+                    language = first_line
+                    code = "\n".join(code.split("\n")[1:])
+            
+            answer_html += f'<pre class="code-block"><code class="language-{language}">{code}</code></pre>'
+        # Process blockquotes
+        elif paragraph.startswith(">"):
+            content = paragraph.replace("\n>", "\n").replace(">", "")
+            answer_html += f'<blockquote>{content}</blockquote>'
+        # Standard paragraph
+        else:
+            # Process inline code
+            paragraph = re.sub(r'`([^`]+)`', r'<code>\1</code>', paragraph)
+            # Process bold text
+            paragraph = re.sub(r'\*\*([^*]+)\*\*', r'<strong>\1</strong>', paragraph)
+            # Process italic text
+            paragraph = re.sub(r'\*([^*]+)\*', r'<em>\1</em>', paragraph)
+            
+            answer_html += f"<p>{paragraph}</p>"
     
     # Add sources if available
     if sources:
@@ -286,7 +334,7 @@ def format_answer_html(answer, sources=None, reasoning_steps=None, debug=False):
             """
         answer_html += reasoning_html
     
-    # Return the final formatted HTML within the appropriate container divs
+    # Return the final formatted HTML without any copy functionality
     return f"""
     <div class="response-wrapper">
         <div class="answer-container">