Update app.py

app.py

@@ -11,12 +11,8 @@ import io
 import os
 import traceback
 from datetime import datetime
-import gc
 
-# Import your document checker classes
-from main import FAADocumentChecker, DocumentCheckResult
-
-# Core data structures and utilities
+# Core data structures
 @dataclass
 class DocumentCheckResult:
     """Structured result for document checks."""
@@ -187,7 +183,7 @@ class DocumentChecker:
             return []
 
 class FAADocumentChecker(DocumentChecker):
-    """Main document checker implementation with all check methods."""
+    """Main document checker implementation."""
     def __init__(self, config_path: Optional[str] = None):
         super().__init__(config_path)
 
@@ -869,6 +865,33 @@ class FAADocumentChecker(DocumentChecker):
 
         return results
 
+def process_document(file_obj, doc_type: str, template_type: Optional[str] = None) -> str:
+    """Process document and run all checks."""
+    try:
+        checker = FAADocumentChecker()
+        
+        if isinstance(file_obj, bytes):
+            file_obj = io.BytesIO(file_obj)
+            
+        results = checker.run_all_checks(file_obj, doc_type, template_type)
+        return format_markdown_results(results, doc_type)
+        
+    except Exception as e:
+        logging.error(f"Error processing document: {str(e)}")
+        traceback.print_exc()
+        return f"""
+# ❌ Error Processing Document
+
+**Error Details:** {str(e)}
+
+Please ensure:
+1. The file is a valid .docx document
+2. The file is not corrupted or password protected
+3. The file is properly formatted
+
+Try again after checking these issues. If the problem persists, contact support.
+"""
+    
 def format_markdown_results(results: Dict[str, DocumentCheckResult], doc_type: str) -> str:
     """Format check results into a Markdown string for Gradio display."""
     output = []
@@ -951,35 +974,6 @@ def format_markdown_results(results: Dict[str, DocumentCheckResult], doc_type: s
     return "\n".join(output)
 
 def create_interface():
-
-    def process_document(file_obj, doc_type: str, template_type: Optional[str] = None) -> str:
-        """Process document and run all checks."""
-        try:
-            checker = FAADocumentChecker()
-            
-            if isinstance(file_obj, bytes):
-                file_obj = io.BytesIO(file_obj)
-                
-            results = checker.run_all_checks(file_obj, doc_type, template_type)
-            return format_markdown_results(results, doc_type)
-            
-        except Exception as e:
-            logging.error(f"Error processing document: {str(e)}")
-            traceback.print_exc()
-            return f"""
-        # ❌ Error Processing Document
-
-        **Error Details:** {str(e)}
-
-        Please ensure:
-        1. The file is a valid .docx document
-        2. The file is not corrupted or password protected
-        3. The file is properly formatted
-
-        Try again after checking these issues. If the problem persists, contact support.
-        """
-
-    def create_interface():
         """Create and configure the Gradio interface."""
         document_types = [
             "Advisory Circular",
@@ -1149,100 +1143,19 @@ def create_interface():
         
         return demo
 
-    def format_markdown_results(results: Dict[str, DocumentCheckResult], doc_type: str) -> str:
-        """Format check results into a Markdown string for Gradio display."""
-        output = []
-        
-        current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        output.extend([
-            f"# Document Check Results - {current_time}",
-            f"## Document Type: {doc_type}",
-            "---\n"
-        ])
-
-        total_issues = sum(1 for r in results.values() if not r.success)
-        
-        if total_issues == 0:
-            output.append("✅ **All checks passed successfully!**\n")
-            return "\n".join(output)
-        
-        output.append(f"❗ Found issues in {total_issues} check categories\n")
-
-        check_categories = {
-            'heading_title_check': {'title': '📋 Required Headings', 'priority': 1},
-            'heading_title_period_check': {'title': '🔍 Heading Period Usage', 'priority': 1},
-            'acronym_check': {'title': '📝 Acronym Definitions', 'priority': 2},
-            'terminology_check': {'title': '📖 Terminology Usage', 'priority': 2},
-            'section_symbol_usage_check': {'title': '§ Section Symbol Usage', 'priority': 2},
-            'caption_check_table': {'title': '📊 Table Captions', 'priority': 3},
-            'caption_check_figure': {'title': '🖼️ Figure Captions', 'priority': 3},
-            'table_figure_reference_check': {'title': '🔗 Table/Figure References', 'priority': 3},
-            'document_title_check': {'title': '📑 Document Title Format', 'priority': 1},
-            'double_period_check': {'title': '⚡ Double Periods', 'priority': 4},
-            'spacing_check': {'title': '⌨️ Spacing Issues', 'priority': 4},
-            'abbreviation_usage_check': {'title': '📎 Abbreviation Usage', 'priority': 3},
-            'date_formats_check': {'title': '📅 Date Formats', 'priority': 3},
-            'placeholders_check': {'title': '🚩 Placeholder Content', 'priority': 1}
-        }
-
-        sorted_checks = sorted(
-            [(name, result) for name, result in results.items()],
-            key=lambda x: check_categories.get(x[0], {'priority': 999})['priority']
-        )
-
-        for check_name, result in sorted_checks:
-            if not result.success:
-                category = check_categories.get(check_name, {'title': check_name.replace('_', ' ').title()})
-                
-                output.append(f"### {category['title']}")
-                
-                if isinstance(result.issues, list):
-                    for issue in result.issues[:5]:
-                        if isinstance(issue, dict):
-                            for key, value in issue.items():
-                                if isinstance(value, list):
-                                    output.extend([f"- {item}" for item in value])
-                                else:
-                                    output.append(f"- {key}: {value}")
-                        else:
-                            output.append(f"- {issue}")
-                    
-                    if len(result.issues) > 5:
-                        output.append(f"\n*...and {len(result.issues) - 5} more similar issues*")
-                
-                output.append("")
-
-        output.extend([
-            "## 📋 Summary and Recommendations",
-            "",
-            "### Priority Order for Fixes:",
-            "1. 🔴 Critical: Heading formats, required content, and document structure",
-            "2. 🟡 Important: Terminology, acronyms, and references",
-            "3. 🟢 Standard: Formatting, spacing, and style consistency",
-            "",
-            "### Next Steps:",
-            "1. Address issues in priority order",
-            "2. Use search/replace for consistent fixes",
-            "3. Re-run checker after making changes",
-            "4. Update your document template if needed",
-            ""
-        ])
-
-        return "\n".join(output)
-
-    # Initialize and launch the interface
-    if __name__ == "__main__":
-        # Setup logging
-        logging.basicConfig(
-            level=logging.INFO,
-            format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
-        )
+# Initialize and launch the interface
+if __name__ == "__main__":
+    # Setup logging
+    logging.basicConfig(
+        level=logging.INFO,
+        format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+    )
         
-        # Create and launch the interface
-        demo = create_interface()
-        demo.launch(
-            share=False,  # Set to True if you want to generate a public link
-            server_name="0.0.0.0",  # Allows external access
-            server_port=7860,  # Default Gradio port
-            debug=True
-        )
+    # Create and launch the interface
+    demo = create_interface()
+    demo.launch(
+        share=False,  # Set to True if you want to generate a public link
+        server_name="0.0.0.0",  # Allows external access
+        server_port=7860,  # Default Gradio port
+        debug=True
+    )