| """ |
| Output handling for document summaries in multiple formats. |
| """ |
| import os |
| from app.config.settings import SUMMARIES_OUTPUT_DIR |
|
|
|
|
| class SummaryOutputManager: |
| """ |
| Manager for saving and retrieving document summaries in multiple formats. |
| """ |
|
|
| def __init__(self, output_dir=SUMMARIES_OUTPUT_DIR): |
| """ |
| Initialize output manager with output directory. |
| |
| Args: |
| output_dir (str): Directory to save summaries |
| """ |
| self.output_dir = output_dir |
| self._ensure_output_dir() |
|
|
| def _ensure_output_dir(self): |
| """Create output directory if it doesn't exist.""" |
| if not os.path.exists(self.output_dir): |
| os.makedirs(self.output_dir) |
| print(f"Created output directory: {self.output_dir}") |
|
|
| def save_summary(self, filename, summary, formats=None): |
| """ |
| Save a document summary to files in specified formats. |
| |
| Args: |
| filename (str): Name of the original document |
| summary (str): Summary text |
| formats (list): List of formats to save. Defaults to ['markdown', 'html'] |
| |
| Returns: |
| dict: Paths to the saved summary files by format |
| """ |
| if formats is None: |
| formats = ['markdown', 'html'] |
|
|
| output_paths = {} |
|
|
| |
| for fmt in formats: |
| if fmt == 'markdown': |
| output_paths['markdown'] = self._save_markdown(filename, summary) |
| elif fmt == 'html': |
| output_paths['html'] = self._save_html(filename, summary) |
| else: |
| print(f"Warning: Unsupported format '{fmt}' requested") |
|
|
| return output_paths |
|
|
| def _save_markdown(self, filename, summary): |
| """ |
| Save a document summary to a markdown file. |
| |
| Args: |
| filename (str): Name of the original document |
| summary (str): Summary text |
| |
| Returns: |
| str: Path to the saved markdown file |
| """ |
| |
| markdown_content = f"" |
| markdown_content += summary |
| markdown_content += "\n\n---\n" |
|
|
| |
| output_path = os.path.join(self.output_dir, f"{filename}.md") |
| with open(output_path, "w") as f: |
| f.write(markdown_content) |
|
|
| print(f"Saved markdown summary to: {output_path}") |
| return output_path |
|
|
| def _save_html(self, filename, summary): |
| """ |
| Save a document summary to an HTML file. |
| |
| Args: |
| filename (str): Name of the original document |
| summary (str): Summary text |
| |
| Returns: |
| str: Path to the saved HTML file |
| """ |
| |
| paragraphs = summary.split('\n\n') |
| html_paragraphs = ''.join([f"<p>{p}</p>" for p in paragraphs if p.strip()]) |
|
|
| |
| html_content = f"""<!DOCTYPE html> |
| <html> |
| <head> |
| <meta charset="UTF-8"> |
| <meta name="viewport" content="width=device-width, initial-scale=1.0"> |
| <title>Summary for {filename}</title> |
| <style> |
| body {{ |
| font-family: Arial, sans-serif; |
| line-height: 1.6; |
| margin: 0; |
| padding: 20px; |
| max-width: 800px; |
| margin: 0 auto; |
| color: #333; |
| }} |
| h1 {{ |
| color: #2c3e50; |
| border-bottom: 1px solid #eee; |
| padding-bottom: 10px; |
| }} |
| p {{ |
| margin-bottom: 16px; |
| }} |
| .footer {{ |
| margin-top: 30px; |
| padding-top: 10px; |
| border-top: 1px solid #eee; |
| font-size: 0.9em; |
| color: #7f8c8d; |
| }} |
| </style> |
| </head> |
| <body> |
| <h1>Summary for {filename}</h1> |
| <div class="content"> |
| {html_paragraphs} |
| </div> |
| <div class="footer"> |
| <p>Generated summary</p> |
| </div> |
| </body> |
| </html> |
| """ |
| |
| output_path = os.path.join(self.output_dir, f"{filename}.html") |
| with open(output_path, "w") as f: |
| f.write(html_content) |
|
|
| print(f"Saved HTML summary to: {output_path}") |
| return output_path |
|
|
| def get_available_formats(self, filename): |
| """ |
| Check which formats are available for a given file. |
| |
| Args: |
| filename (str): Base filename to check |
| |
| Returns: |
| list: Available formats for this file |
| """ |
| available_formats = [] |
| base_name = os.path.splitext(filename)[0] |
|
|
| if os.path.exists(os.path.join(self.output_dir, f"{base_name}.md")): |
| available_formats.append('markdown') |
| if os.path.exists(os.path.join(self.output_dir, f"{base_name}.html")): |
| available_formats.append('html') |
|
|
| return available_formats |
