Update app.py

app.py

@@ -1,11 +1,13 @@
 import gradio as gr
 from annotations import analyze_pdf
-import base64
 import io
 
 def process_pdf(file):
+    MAX_SIZE = 10 * 1024 * 1024  # 10MB
     if file is None:
         return "❌ No file uploaded.", None
+    if len(file) > MAX_SIZE:
+        return "❌ File size exceeds the 10MB limit.", None
 
     try:
         # Wrap the binary data in BytesIO to make it file-like
@@ -18,27 +20,33 @@ def process_pdf(file):
             return f"❌ Error: {issues['error']}", None
 
         # Prepare issues for display
-        issues_display = f"**Total Issues Found:** {issues['total_issues']}\n\n"
-        for idx, issue in enumerate(issues['issues'], start=1):
-            issues_display += f"**Issue {idx}:**\n"
-            issues_display += f"- **Message:** {issue['message']}\n"
-            issues_display += f"- **Context:** {issue['context']}\n"
-            issues_display += f"- **Suggestions:** {', '.join(issue['suggestions']) if issue['suggestions'] else 'None'}\n"
-            issues_display += f"- **Category:** {issue['category']}\n"
-            issues_display += f"- **Rule ID:** {issue['rule_id']}\n"
-            issues_display += f"- **Offset:** {issue['offset']}\n"
-            issues_display += f"- **Length:** {issue['length']}\n\n"
+        if issues['total_issues'] == 0:
+            issues_display = "✅ No language issues found. Great job!"
+        else:
+            issues_display = f"**Total Issues Found:** {issues['total_issues']}\n\n"
+            for idx, issue in enumerate(issues['issues'], start=1):
+                issues_display += f"**Issue {idx}:**\n"
+                issues_display += f"- **Message:** {issue['message']}\n"
+                issues_display += f"- **Context:** {issue['context']}\n"
+                issues_display += f"- **Suggestions:** {', '.join(issue['suggestions']) if issue['suggestions'] else 'None'}\n"
+                issues_display += f"- **Category:** {issue['category']}\n"
+                issues_display += f"- **Rule ID:** {issue['rule_id']}\n"
+                issues_display += f"- **Offset:** {issue['offset']}\n"
+                issues_display += f"- **Length:** {issue['length']}\n\n"
 
         # Prepare annotated PDF for download
         if annotated_pdf:
-            # Save the annotated PDF to a BytesIO object
-            annotated_pdf_io = io.BytesIO(annotated_pdf)
-            annotated_pdf_io.name = "annotated_document.pdf"  # Set a default filename
-            annotated_pdf_io.seek(0)
-            return issues_display, annotated_pdf_io
+            # Return a dictionary with 'name' and 'data' keys
+            annotated_pdf_dict = {
+                "name": "annotated_document.pdf",
+                "data": annotated_pdf
+            }
+            return issues_display, annotated_pdf_dict
         else:
             return issues_display, None
 
+    except language_tool_python.LanguageToolError as e:
+        return f"❌ LanguageTool Error: {str(e)}", None
     except Exception as e:
         return f"❌ An unexpected error occurred: {str(e)}", None