Update app.py

app.py

@@ -4,6 +4,7 @@ import re
 from docx import Document
 import tempfile
 import os
+import traceback
 
 def setup_logging():
     """Initialize logging configuration."""
@@ -642,63 +643,41 @@ def format_results_for_gradio(heading_valid, headings_found, acronyms_valid, und
     
     return "\n".join(results)
 
-# Modify the process_document function to return formatted results instead of writing to file
-def process_document(file_obj, doc_type, template_type):
-    """Process the document and perform checks based on document type and template type."""
-    logger = logging.getLogger(__name__)
+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."
     
-    # Create a temporary file to save the uploaded file
-    with tempfile.NamedTemporaryFile(delete=False, suffix='.docx') as tmp_file:
-        tmp_file.write(file_obj.read())
-        tmp_path = tmp_file.name
+    if not hasattr(file_obj, 'name') or not file_obj.name.endswith('.docx'):
+        return "Please upload a valid .docx file."
     
     try:
-        # Step 1: Read the Word document
-        doc = read_word_document(tmp_path)
-        logger.info("Document read successfully.")
-        
-        # Get required headings for document type and template type
-        checks = get_document_checks(doc_type, template_type)
-        required_headings = checks.get("required_headings", [])
+        # Process the document and get results
+        results = process_document(file_obj, doc_type, template_type)
+        return results
+    except Exception as e:
+        error_trace = traceback.format_exc()
+        error_message = f"""An error occurred while processing the document:
         
-        # Step 2: Perform all checks
-        heading_valid, headings_found = heading_title_check(doc, required_headings)
-        acronyms_valid, undefined_acronyms = acronym_check(doc)
-        legal_valid, incorrect_legal_references = legal_check(doc)
-        table_valid, incorrect_captions = table_caption_check(doc, doc_type)
-        figure_valid, incorrect_fig_captions = figure_caption_check(doc, doc_type)
-        references_valid, incorrect_table_figure_references = table_figure_reference_check(doc, doc_type)
-        title_style_valid, incorrect_titles = document_title_check(tmp_path, doc_type)
-        double_period_valid, incorrect_sentences = double_period_check(doc)
-        spacing_valid, incorrect_spacing = spacing_check(doc)
-        abbreviation_issues = check_abbreviation_usage(doc)
-        date_issues = check_date_formats(doc)
-        placeholder_issues = check_placeholders(doc)
-
-        # Format results for Gradio
-        results = format_results_for_gradio(
-            heading_valid, headings_found,
-            acronyms_valid, undefined_acronyms,
-            legal_valid, incorrect_legal_references,
-            table_valid, incorrect_captions,
-            figure_valid, incorrect_fig_captions,
-            references_valid, incorrect_table_figure_references,
-            title_style_valid, incorrect_titles,
-            required_headings, doc_type,
-            double_period_valid, incorrect_sentences,
-            spacing_valid, incorrect_spacing,
-            abbreviation_issues, date_issues,
-            placeholder_issues
-        )
+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"""
         
-        return results
-    
-    finally:
-        # Clean up the temporary file
-        os.unlink(tmp_path)
+        logging.error(f"Error processing document: {error_trace}")
+        return error_message
+
+# Create the interface with simplified layout for Hugging Face Spaces
+demo = gr.Blocks(title="FAA Document Checker")
 
-# Create the Gradio interface
-def create_gradio_interface():
+with demo:
+    gr.Markdown("# FAA Document Checker")
+    gr.Markdown("Upload a Word document to check for compliance with FAA documentation standards.")
+    
     document_types = [
         "Advisory Circular", "Airworthiness Criteria", "Deviation Memo", "Exemption", 
         "Federal Register Notice", "Handbook/Manual", "Order", "Policy Statement", 
@@ -707,30 +686,66 @@ def create_gradio_interface():
     
     template_types = ["Short AC template AC", "Long AC template AC"]
     
-    def process_file(file_obj, doc_type, template_type):
-        if doc_type != "Advisory Circular":
-            template_type = "N/A"
-        return process_document(file_obj, doc_type, template_type)
+    with gr.Row():
+        with gr.Column(scale=1):
+            file_input = gr.File(
+                label="Upload Word Document (.docx)",
+                file_types=[".docx"],
+                type="binary"
+            )
+            doc_type = gr.Dropdown(
+                choices=document_types,
+                label="Document Type",
+                value="Advisory Circular"
+            )
+            template_type = gr.Radio(
+                choices=template_types,
+                label="Template Type (Only for Advisory Circular)",
+                visible=True,
+                value="Short AC template AC"
+            )
+            submit_btn = gr.Button("Check Document", variant="primary")
+        
+        with gr.Column(scale=2):
+            output = gr.Markdown(
+                label="Check Results",
+                value="Results will appear here after processing..."
+            )
+    
+    # Update template type visibility based on document type
+    def update_template_visibility(doc_type):
+        return gr.update(visible=doc_type == "Advisory Circular")
+    
+    doc_type.change(
+        fn=update_template_visibility,
+        inputs=[doc_type],
+        outputs=[template_type]
+    )
     
-    # Create the interface
-    iface = gr.Interface(
+    # Process file when submit button is clicked
+    submit_btn.click(
         fn=process_file,
-        inputs=[
-            gr.File(label="Upload Word Document (.docx)", type="binary"),
-            gr.Dropdown(choices=document_types, label="Document Type"),
-            gr.Radio(choices=template_types, label="Template Type (Only for Advisory Circular)", visible=True)
-        ],
-        outputs=gr.Markdown(label="Check Results"),
-        title="FAA Document Checker",
-        description="Upload a Word document to check for compliance with FAA documentation standards.",
-        article="This tool checks document formatting, headings, acronyms, legal references, and more.",
-        theme="default"
+        inputs=[file_input, doc_type, template_type],
+        outputs=[output]
     )
     
-    return iface
-
-# Launch the Gradio interface
-if __name__ == "__main__":
-    setup_logging()
-    iface = create_gradio_interface()
-    iface.launch()
+    gr.Markdown("""
+    ## Instructions
+    1. Upload your Word document (.docx format)
+    2. Select the document type
+    3. If it's an Advisory Circular, select the template type
+    4. Click 'Check Document' to process
+    5. Review the results in the output panel
+    
+    The checker will verify:
+    - Required headings
+    - Acronym definitions
+    - Legal terminology
+    - Table and figure captions
+    - Document title styling
+    - Spacing and formatting
+    - And more...
+    """)
+
+# For Hugging Face Spaces, we just need to expose the 'demo'
+demo.launch()