Update app.py

app.py

@@ -6,7 +6,7 @@ from annotations import analyze_pdf
 
 def display_pdf(pdf_bytes):
     """Displays the PDF in the browser using an embed tag."""
-    if pdf_bytes:
+    if pdf_bytes and len(pdf_bytes) > 0:
         # Encode the PDF to base64
         base64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
         pdf_display = f"""
@@ -40,6 +40,12 @@ def main():
         else:
             st.success("Analysis complete!")
 
+            # Debugging: Show the size of annotated_pdf
+            if annotated_pdf and len(annotated_pdf) > 0:
+                st.write(f"Annotated PDF size: {len(annotated_pdf)} bytes")
+            else:
+                st.write("Annotated PDF is empty.")
+
             # Display the annotated PDF
             st.subheader("📄 Annotated PDF")
             display_pdf(annotated_pdf)
@@ -58,13 +64,15 @@ def main():
                 st.sidebar.success("No language issues found!")
 
             # Option to download the annotated PDF
-            if annotated_pdf:
+            if annotated_pdf and len(annotated_pdf) > 0:
                 st.download_button(
                     label="📥 Download Annotated PDF",
                     data=annotated_pdf,
                     file_name="annotated.pdf",
                     mime="application/pdf",
                 )
+            else:
+                st.info("No annotated PDF available for download.")
 
 if __name__ == "__main__":
     main()