Update app.py

app.py

@@ -280,6 +280,105 @@ def get_document_checks(doc_type, template_type):
     # Add other document types as needed
     return {"required_headings": []}
 
+def process_file(file_obj, doc_type, template_type):
+    """
+    Process the uploaded file and return results with error handling
+    """
+    if file_obj is None:
+        return "Please upload a document first."
+    
+    try:
+        # Convert bytes to BytesIO object that Document can read
+        if isinstance(file_obj, bytes):
+            doc_bytes = io.BytesIO(file_obj)
+        else:
+            doc_bytes = io.BytesIO(file_obj.read())
+            
+        # Process the document and get results
+        results = process_document(doc_bytes, doc_type, template_type)
+        return results
+        
+    except Exception as e:
+        error_trace = traceback.format_exc()
+        print(f"Error processing file: {str(e)}")
+        print(f"Full traceback: {error_trace}")
+        
+        error_message = f"""An error occurred while processing the document:
+        
+Error: {str(e)}
+
+Please ensure:
+1. The file is a valid Word document (.docx)
+2. The file is not corrupted
+3. The file is not password protected
+
+Technical details: {str(e)}"""
+        
+        return error_message
+
+def process_document(file_obj, doc_type, template_type):
+    """Process the document and perform checks."""
+    try:
+        # Read the Word document
+        doc = Document(file_obj)
+        print("Document read successfully.")
+        
+        # Get required headings based on document type
+        required_headings = get_document_checks(doc_type, template_type).get("required_headings", [])
+        
+        # Perform checks
+        heading_valid, headings_found = heading_title_check(doc, required_headings)
+        acronyms_valid, undefined_acronyms = acronym_check(doc)
+        
+        # Format results
+        results = format_results_for_gradio(
+            heading_valid=heading_valid,
+            headings_found=headings_found,
+            acronyms_valid=acronyms_valid,
+            undefined_acronyms=undefined_acronyms,
+            legal_valid=True,  # Placeholder
+            incorrect_legal_references=[],
+            table_valid=True,  # Placeholder
+            incorrect_captions=[],
+            figure_valid=True,  # Placeholder
+            incorrect_fig_captions=[],
+            references_valid=True,  # Placeholder
+            incorrect_table_figure_references=[],
+            title_style_valid=True,  # Placeholder
+            incorrect_titles=[],
+            required_headings=required_headings,
+            doc_type=doc_type,
+            double_period_valid=True,  # Placeholder
+            incorrect_sentences=[],
+            spacing_valid=True,  # Placeholder
+            incorrect_spacing=[],
+            abbreviation_issues=[],
+            date_issues=[],
+            placeholder_issues=[]
+        )
+        
+        return results
+    
+    except Exception as e:
+        print(f"Error in process_document: {str(e)}")
+        raise
+
+def get_document_checks(doc_type, template_type):
+    """Return the required headings and other checks based on document type."""
+    if doc_type == "Advisory Circular":
+        if template_type == "Short AC template AC":
+            return {
+                "required_headings": ["Purpose", "Applicability", "Related Reading Material", 
+                                    "Background", "Discussion"]
+            }
+        else:  # Long AC template
+            return {
+                "required_headings": ["Purpose", "Applicability", "Audience", "Related Reading Material",
+                                    "Background", "Discussion", "Conclusion"]
+            }
+    # Add other document types as needed
+    return {"required_headings": []}
+
 def format_results_for_gradio(**kwargs):
     """Format the results for display in Gradio."""
     results = []
@@ -422,132 +521,6 @@ def format_results_for_gradio(**kwargs):
     
     return "\n".join(results)
 
-def process_file(file_obj, doc_type, template_type):
-    """
-    Process the uploaded file and return results with error handling
-    """
-    if file_obj is None:
-        return "Please upload a document first."
-    
-    try:
-        # Convert bytes to BytesIO object that Document can read
-        if isinstance(file_obj, bytes):
-            doc_bytes = io.BytesIO(file_obj)
-        else:
-            doc_bytes = io.BytesIO(file_obj.read())
-            
-        # Process the document and get results
-        results = process_document(doc_bytes, doc_type, template_type)
-        return results
-        
-    except Exception as e:
-        error_trace = traceback.format_exc()
-        print(f"Error processing file: {str(e)}")
-        print(f"Full traceback: {error_trace}")
-        
-        error_message = f"""An error occurred while processing the document:
-        
-Error: {str(e)}
-
-Please ensure:
-1. The file is a valid Word document (.docx)
-2. The file is not corrupted
-3. The file is not password protected
-
-Technical details: {str(e)}"""
-        
-        return error_message
-
-def process_document(file_obj, doc_type, template_type):
-    """Process the document and perform checks."""
-    try:
-        # Read the Word document
-        doc = Document(file_obj)
-        print("Document read successfully.")
-        
-        # Get required headings based on document type
-        required_headings = get_document_checks(doc_type, template_type).get("required_headings", [])
-        
-        # Perform checks
-        heading_valid, headings_found = heading_title_check(doc, required_headings)
-        acronyms_valid, undefined_acronyms = acronym_check(doc)
-        
-        # Format results
-        results = format_results_for_gradio(
-            heading_valid=heading_valid,
-            headings_found=headings_found,
-            acronyms_valid=acronyms_valid,
-            undefined_acronyms=undefined_acronyms,
-            legal_valid=True,  # Placeholder
-            incorrect_legal_references=[],
-            table_valid=True,  # Placeholder
-            incorrect_captions=[],
-            figure_valid=True,  # Placeholder
-            incorrect_fig_captions=[],
-            references_valid=True,  # Placeholder
-            incorrect_table_figure_references=[],
-            title_style_valid=True,  # Placeholder
-            incorrect_titles=[],
-            required_headings=required_headings,
-            doc_type=doc_type,
-            double_period_valid=True,  # Placeholder
-            incorrect_sentences=[],
-            spacing_valid=True,  # Placeholder
-            incorrect_spacing=[],
-            abbreviation_issues=[],
-            date_issues=[],
-            placeholder_issues=[]
-        )
-        
-        return results
-    
-    except Exception as e:
-        print(f"Error in process_document: {str(e)}")
-        raise
-
-def get_document_checks(doc_type, template_type):
-    """Return the required headings and other checks based on document type."""
-    if doc_type == "Advisory Circular":
-        if template_type == "Short AC template AC":
-            return {
-                "required_headings": ["Purpose", "Applicability", "Related Reading Material", 
-                                    "Background", "Discussion"]
-            }
-        else:  # Long AC template
-            return {
-                "required_headings": ["Purpose", "Applicability", "Audience", "Related Reading Material",
-                                    "Background", "Discussion", "Conclusion"]
-            }
-    # Add other document types as needed
-    return {"required_headings": []}
-
-def format_results_for_gradio(**kwargs):
-    """Format the results for display."""
-    results = []
-    results.append("# Document Check Results\n")
-    
-    # Required Headings Check
-    results.append("## Required Headings Check")
-    if kwargs['heading_valid']:
-        results.append("✅ All required headings are present.\n")
-    else:
-        missing_headings = set(kwargs['required_headings']) - set(kwargs['headings_found'])
-        results.append("❌ Missing Required Headings:")
-        for heading in missing_headings:
-            results.append(f"- {heading}")
-    results.append("")
-    
-    # Acronym Check
-    results.append("## Acronym Check")
-    if kwargs['acronyms_valid']:
-        results.append("✅ All acronyms are properly defined.\n")
-    else:
-        results.append("❌ The following acronyms need to be defined at first use:")
-        for acronym in kwargs['undefined_acronyms']:
-            results.append(f"- {acronym}")
-    
-    return "\n".join(results)
-
 def process_file(file_obj, doc_type, template_type):
     """Process the uploaded file and return results with error handling."""
     if file_obj is None: