feat: add mock statistics

Author: yuvrajsinh5252

src/lib/problem_ui.py

@@ -7,7 +7,7 @@
 from rich.style import Style
 from rich.text import Text
 from rich.align import Align
-from bs4 import BeautifulSoup
+from bs4 import BeautifulSoup, Tag
 import json
 from dataclasses import dataclass
 from typing import List, Dict, Optional
@@ -36,7 +36,6 @@ def __init__(self, problem_data: dict):
         )
         self.code_snippets: List[Dict] = problem_data['codeSnippets']
 
-        # Parse metadata
         self.function_metadata: Optional[FunctionMetadata] = None
         try:
             metadata = json.loads(problem_data['metaData'])
@@ -48,6 +47,10 @@ def __init__(self, problem_data: dict):
         except Exception:
             pass
 
+        self.total_accepted: int = problem_data.get('totalAccepted', 0)
+        self.total_submissions: int = problem_data.get('totalSubmissions', 0)
+        self.acceptance_rate: float = problem_data.get('acRate', 0)
+
         self._parse_content()
         self.console_width = console.width
 
@@ -82,16 +85,6 @@ def _parse_content(self):
             else:
                 current_section.append(text)
 
-    @property
-    def formatted_description(self) -> str:
-        """Get the formatted problem description"""
-        sections = [
-            ' '.join(self.description),
-            *self.examples,
-            *self.constraints
-        ]
-        return '\n'.join(line for line in sections if line.strip())
-
     @property
     def available_languages(self) -> List[str]:
         """Get list of available programming languages"""
@@ -138,17 +131,16 @@ def _create_header(self):
         }
         difficulty_color = difficulty_colors.get(self.difficulty, 'white')
 
-        header = Text()
-        header.append(f"{self.question_id}. {self.title}", style="bold cyan")
-        header.append("  ", style="dim")
-        header.append(self.difficulty, style=difficulty_color)
+        header = (
+            f"[bold cyan]{self.question_id}. {self.title}[/]  "
+            f"[{difficulty_color}]{self.difficulty}[/]"
+        )
 
-        return Align.center(header)
+        return header
 
     def _format_description(self, html_content: str) -> str:
         """Format the problem description, removing example test cases"""
         soup = BeautifulSoup(html_content, 'html.parser')
-        typer.echo(soup.prettify())
 
         # Remove all pre tags and example sections
         for pre in soup.find_all('pre'):
@@ -163,9 +155,16 @@ def _format_description(self, html_content: str) -> str:
         for code in soup.find_all('code'):
             code.replace_with(BeautifulSoup(f'`{code.text}`', 'html.parser'))
 
-        # Format strong and em tags
-        for strong in soup.find_all('strong'):
-            strong.replace_with(BeautifulSoup(f'**{strong.text}**', 'html.parser'))
+        # Format HTML formatting tags
+        for tag in soup.find_all(['strong', 'em', 'b', 'i']):
+            if not isinstance(tag, Tag):
+                continue
+            text = tag.text.strip()
+            if tag.name in ['strong', 'b']:
+                replacement = f"**{text}**"
+            elif tag.name in ['em', 'i']:
+                replacement = f"_{text}_"
+            tag.replace_with(BeautifulSoup(replacement, 'html.parser'))
 
         for em in soup.find_all('em'):
             em.replace_with(BeautifulSoup(f'_{em.text}_', 'html.parser'))
@@ -184,49 +183,87 @@ def _format_test_cases(self):
 
         # Extract example explanations from HTML content
         soup = BeautifulSoup(self.content, 'html.parser')
-        explanations = []
-        for strong in soup.find_all('strong'):
-            if 'Example' in strong.text:
-                parent = strong.find_parent(['p', 'div'])
-                if parent:
-                    explanations.append(parent.get_text())
-
-        # Combine test cases with their explanations
-        for i, (test, explanation) in enumerate(zip(examples, explanations)):
-            test_cases.append(
-                f"[bold blue]Example {i + 1}:[/]\n"
-                f"[cyan]Input:[/] {test}\n"
-                f"{explanation.replace('Example ' + str(i + 1) + ':', '[yellow]Explanation:[/]')}"
+
+        examples_data = []
+
+        # Find all pre tags containing example data
+        for pre in soup.find_all('pre'):
+            # Extract input, output, and explanation
+            if isinstance(pre, Tag):
+                input_tag = pre.find('strong', text='Input:')
+                output_tag = pre.find('strong', text='Output:')
+                explanation_tag = pre.find('strong', text='Explanation:')
+
+            if input_tag and output_tag:
+                input_text = str(input_tag.next_sibling).strip() if input_tag and input_tag.next_sibling else ""
+                output_text = str(output_tag.next_sibling).strip() if output_tag and output_tag.next_sibling else ""
+                explanation_text = str(explanation_tag.next_sibling).strip() if explanation_tag and explanation_tag.next_sibling else ""
+
+            examples_data.append({
+                'input': input_text,
+                'output': output_text,
+                'explanation': explanation_text
+            })
+
+        # Format the examples
+        for i, example in enumerate(examples_data, 1):
+            formatted_example = (
+            f"[bold blue]Example {i}: [/]\n\n"
+            f"[cyan]Input: [/]{example['input']}\n"
+            f"[cyan]Output: [/]{example['output']}"
             )
+            if example['explanation']:
+                formatted_example += f"\n[yellow]Explanation: [/]{example['explanation']}"
+            test_cases.append(formatted_example)
 
         return "\n\n".join(test_cases)
 
+    def _format_stats(self) -> str:
+        """Format problem statistics"""
+        acceptance_rate = f"{self.acceptance_rate:.1f}%" if self.acceptance_rate else "N/A"
+        total_accepted = f"{self.total_accepted:,}" if self.total_accepted else "N/A"
+        total_submissions = f"{self.total_submissions:,}" if self.total_submissions else "N/A"
+
+        return (
+            "[bold blue]Problem Stats[/]\n\n"
+            f"[cyan]Acceptance Rate:[/] {acceptance_rate}\n"
+            f"[cyan]Total Accepted:[/] {total_accepted}\n"
+            f"[cyan]Total Submissions:[/] {total_submissions}"
+        )
+
     def display(self):
         """Display the problem details"""
-        # Clear screen and show header
         console.clear()
-        console.print(self._create_header())
         console.print()
 
+        test_cases = self._format_test_cases()
+        stats = self._format_stats()
+
         # Create main layout
         layout = Layout()
+
+        # Split into left and right sections
         layout.split_row(
             Layout(name="description", ratio=2),
-            Layout(name="examples", ratio=1)
+            Layout(name="right_panel", ratio=1)
+        )
+
+        # Split right panel into examples and stats
+        layout["right_panel"].split_column(
+            Layout(name="examples"),
+            Layout(name="stats", size=10)
         )
 
         # Description panel
-        description = self._format_description(self.content)
         layout["description"].update(Panel(
-            Markdown(description),
+            Markdown(self._format_description(self.content)),
             box=ROUNDED,
-            title="[bold blue]Description",
+            title=str(self._create_header()),
             border_style="blue",
             padding=(1, 2)
         ))
 
         # Example test cases panel
-        test_cases = self._format_test_cases()
         layout["examples"].update(Panel(
             test_cases,
             box=ROUNDED,
@@ -235,5 +272,13 @@ def display(self):
             padding=(1, 2)
         ))
 
-        # Display the layout
+        # Stats panel
+        layout["stats"].update(Panel(
+            stats,
+            box=ROUNDED,
+            title="[bold blue]Statistics",
+            border_style="blue",
+            padding=(1, 2)
+        ))
+
         console.print(layout)