Spaces:
Paused
Paused
| from utils.markdown_analyzer import MarkdownAnalyzer | |
| from utils.logging_config import setup_logging, get_logger | |
| # Initialize logging | |
| setup_logging() | |
| logger = get_logger(__name__) | |
| ### MARKDOWN UTILS ### | |
| def remove_markdown_code_blocks(text: str) -> str: | |
| """ | |
| Remove markdown code block syntax from text. | |
| Args: | |
| text (str): Text that may contain markdown code block syntax | |
| Returns: | |
| str: Text with markdown code block syntax removed | |
| """ | |
| content = text | |
| if content.startswith("```markdown"): | |
| content = content[11:] # Remove ```markdown | |
| if content.endswith("```"): | |
| content = content[:-3] # Remove ``` | |
| return content.strip() | |
| def remove_markdown_list_headers( | |
| merged_tasks: list[tuple[str, str]] | |
| ) -> list[tuple[str, str]]: | |
| """ | |
| Remove list headers (e.g. **element**) from task descriptions. | |
| Args: | |
| merged_tasks (list[tuple[str, str]]): List of (task, duration) tuples | |
| Returns: | |
| list[tuple[str, str]]: List of (task, duration) tuples with headers removed | |
| """ | |
| cleaned_tasks: list[tuple[str, str]] = [] | |
| for task, duration in merged_tasks: | |
| # Use MarkdownAnalyzer to parse and clean the task text | |
| analyzer: MarkdownAnalyzer = MarkdownAnalyzer(task) | |
| # Get the text content without any markdown formatting | |
| cleaned_task: str = analyzer.text.strip() | |
| cleaned_tasks.append((cleaned_task, duration)) | |
| return cleaned_tasks | |
| def remove_markdown_list_elements( | |
| merged_tasks: list[tuple[str, str]] | |
| ) -> list[tuple[str, str]]: | |
| """ | |
| Remove markdown list elements that start and end with ** from task descriptions. | |
| If a task is entirely wrapped in **, remove the entire task. | |
| Args: | |
| merged_tasks (list[tuple[str, str]]): List of (task, duration) tuples | |
| Returns: | |
| list[tuple[str, str]]: List of (task, duration) tuples with markdown list elements removed | |
| """ | |
| cleaned_tasks = [] | |
| for task, duration in merged_tasks: | |
| # Skip tasks that are wrapped in ** | |
| if task.strip().startswith("**") or task.strip().endswith("**"): | |
| continue | |
| cleaned_tasks.append((task, duration)) | |
| return cleaned_tasks | |
| def unwrap_tasks_from_generated(result: list) -> list: | |
| """ | |
| Extract task text from the generated markdown list structure. | |
| Args: | |
| result (list): List containing markdown list structure | |
| Returns: | |
| list: List of task text strings | |
| """ | |
| tasks = [] | |
| # Input validation: check if result is a list | |
| if not isinstance(result, list): | |
| logger.error("Error: 'Unordered list' is not a list!") | |
| return tasks | |
| # We expect result to be a list of lists, with only one entry | |
| if not isinstance(result[0], list): | |
| logger.error("Error: The first element of the result is not a list!") | |
| return tasks | |
| # Unwrap the inner list of dictionaries | |
| for task in result[0]: | |
| if isinstance(task, dict) and "text" in task: | |
| tasks.append(task["text"]) | |
| else: | |
| logger.warning(f"Unexpected task format: {task}") | |
| return tasks | |
| ### LOGGING ### | |
| def log_task_duration_breakdown(merged_tasks: list[tuple[str, str]]) -> None: | |
| """ | |
| Log the duration breakdown for each task. | |
| Args: | |
| merged_tasks (list[tuple[str, str]]): List of (task, duration) tuples | |
| """ | |
| logger.info("Task duration breakdown:") | |
| for task, duration in merged_tasks: | |
| logger.info(f"- {task}: {duration} units") | |
| def safe_int(val): | |
| try: | |
| return int(val) | |
| except (ValueError, TypeError): | |
| return 0 | |
| def log_total_time(merged_tasks: list[tuple[str, str]]) -> None: | |
| """ | |
| Log the total estimated time for all tasks. | |
| Args: | |
| merged_tasks (list[tuple[str, str]]): List of (task, duration) tuples | |
| """ | |
| total_time = sum(safe_int(time) for _, time in merged_tasks) | |
| logger.info("Estimated time:") | |
| logger.info(f"{total_time} units (30 minutes each)") | |