Update app.py

app.py

@@ -280,28 +280,17 @@ def get_document_checks(doc_type, template_type):
     # Add other document types as needed
     return {"required_headings": []}
 
-def 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):
-    """Format the results for Gradio display."""
+def format_results_for_gradio(**kwargs):
+    """Format the results for display in Gradio."""
     results = []
     results.append("# Document Check Results\n")
     
     # Required Headings Check
     results.append("## Required Headings Check")
-    if heading_valid:
+    if kwargs['heading_valid']:
         results.append("✅ All required headings are present.\n")
     else:
-        missing_headings = set(required_headings) - set(headings_found)
+        missing_headings = set(kwargs['required_headings']) - set(kwargs['headings_found'])
         results.append("❌ Missing Required Headings:")
         for heading in missing_headings:
             results.append(f"- {heading}")
@@ -309,61 +298,61 @@ def format_results_for_gradio(heading_valid, headings_found,
     
     # Acronym Check
     results.append("## Acronym Check")
-    if acronyms_valid:
+    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 undefined_acronyms:
+        for acronym in kwargs['undefined_acronyms']:
             results.append(f"- {acronym}")
     results.append("")
-    
+
     # Legal Check
     results.append("## Legal Terminology Check")
-    if legal_valid:
+    if kwargs['legal_valid']:
         results.append("✅ All legal references are properly formatted.\n")
     else:
         results.append("❌ Incorrect Legal Terminology:")
-        for incorrect_term, correct_term in incorrect_legal_references:
+        for incorrect_term, correct_term in kwargs['incorrect_legal_references']:
             results.append(f"- Use '{correct_term}' instead of '{incorrect_term}'")
     results.append("")
     
     # Table Caption Check
     results.append("## Table Caption Check")
-    if table_valid:
+    if kwargs['table_valid']:
         results.append("✅ All table captions are correctly formatted.\n")
     else:
         results.append("❌ Incorrect Table Captions:")
-        for caption in incorrect_captions:
+        for caption in kwargs['incorrect_captions']:
             results.append(f"- {caption}")
     results.append("")
 
     # Figure Caption Check
     results.append("## Figure Caption Check")
-    if figure_valid:
+    if kwargs['figure_valid']:
         results.append("✅ All figure captions are correctly formatted.\n")
     else:
         results.append("❌ Incorrect Figure Captions:")
-        for caption in incorrect_fig_captions:
+        for caption in kwargs['incorrect_fig_captions']:
             results.append(f"- {caption}")
     results.append("")
-    
+
     # Table and Figure References Check
     results.append("## Table and Figure References Check")
-    if references_valid:
+    if kwargs['references_valid']:
         results.append("✅ All table and figure references are correctly formatted.\n")
     else:
         results.append("❌ Incorrect Table/Figure References:")
-        for ref in incorrect_table_figure_references:
+        for ref in kwargs['incorrect_table_figure_references']:
             results.append(f"- {ref}")
     results.append("")
-    
+
     # Document Title Style Check
     results.append("## Document Title Style Check")
-    if title_style_valid:
+    if kwargs['title_style_valid']:
         results.append("✅ All document title references are properly styled.\n")
     else:
         results.append("❌ Incorrect Document Title Styling:")
-        for title in incorrect_titles:
+        for title in kwargs['incorrect_titles']:
             results.append(f"- {title['text']}")
             results.append(f"  - Issue: {title['issue']}")
         
@@ -375,6 +364,7 @@ def format_results_for_gradio(heading_valid, headings_found,
             "Policy Statement": "Document titles should not have any special formatting (no italics, no quotation marks)."
         }
         
+        doc_type = kwargs.get('doc_type', 'Unknown')
         if doc_type in formatting_notes:
             results.append(f"\nNote: {formatting_notes[doc_type]}")
         else:
@@ -383,51 +373,51 @@ def format_results_for_gradio(heading_valid, headings_found,
 
     # Double Period Check
     results.append("## Double Period Check")
-    if double_period_valid:
+    if kwargs['double_period_valid']:
         results.append("✅ No double periods found.\n")
     else:
         results.append("❌ Sentences found with double periods:")
-        for sentence in incorrect_sentences:
+        for sentence in kwargs['incorrect_sentences']:
             results.append(f"- {sentence}")
     results.append("")
 
     # Spacing Check
     results.append("## Spacing Check")
-    if spacing_valid:
+    if kwargs['spacing_valid']:
         results.append("✅ All spacing is correct.\n")
     else:
         results.append("❌ Incorrect spacing found in:")
-        for spacing in incorrect_spacing:
+        for spacing in kwargs['incorrect_spacing']:
             results.append(f"- {spacing}")
     results.append("")
 
     # Abbreviation Consistency
     results.append("## Abbreviation Consistency")
-    if not abbreviation_issues:
+    if not kwargs['abbreviation_issues']:
         results.append("✅ All abbreviations are used consistently after definition.\n")
     else:
         results.append("❌ Abbreviation Issues:")
-        for full_term, acronym, paragraph in abbreviation_issues:
+        for full_term, acronym, paragraph in kwargs['abbreviation_issues']:
             results.append(f"- Use '{acronym}' instead of '{full_term}' in: {paragraph}")
     results.append("")
-    
+
     # Date Format Consistency
     results.append("## Date Format Consistency")
-    if not date_issues:
+    if not kwargs['date_issues']:
         results.append("✅ All dates are in the correct format.\n")
     else:
         results.append("❌ Date Format Issues:")
-        for date, paragraph in date_issues:
+        for date, paragraph in kwargs['date_issues']:
             results.append(f"- Incorrect date format '{date}' in: {paragraph}")
     results.append("")
-    
+
     # Placeholder Check
     results.append("## Placeholder Check")
-    if not placeholder_issues:
+    if not kwargs['placeholder_issues']:
         results.append("✅ No placeholders found.\n")
     else:
         results.append("❌ Placeholders Found:")
-        for phrase, paragraph in placeholder_issues:
+        for phrase, paragraph in kwargs['placeholder_issues']:
             results.append(f"- Placeholder '{phrase}' in: {paragraph}")
     
     return "\n".join(results)