Add automatic scheduling and notification features

New Features:
- setup-schedule.sh: Automatic cron/LaunchAgent setup
- notifications.py: System and email notification support
- --test-notify: Test notification setup
- --setup-schedule: One-command scheduler setup

Notification Features:
- Desktop/system notifications (macOS/Linux/Windows)
- Email notifications via SMTP
- Automatic notification on collection success/failure
- Test notification command for verification

Configuration:
- Added notifications section to config.yaml
- system_notification: true by default
- email_enabled: false (requires SMTP setup)

Usage:
  python main.py --setup-schedule  # Set up automatic daily execution
  python main.py --test-notify      # Send test notification

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: STOKES-DOT

config.yaml

@@ -68,3 +68,18 @@ logging:
 max_papers: 50             # Maximum number of papers to include in report
 abstract_max_length: 1000  # Maximum abstract length in report
 
+# ============================================
+# NOTIFICATION SETTINGS
+# ============================================
+notifications:
+  # Desktop/system notification (requires compatible OS)
+  system_notification: true
+
+  # Email notification (optional)
+  email_enabled: false
+  smtp_server: "smtp.gmail.com"
+  smtp_port: 587
+  email_from: "your-email@gmail.com"
+  email_to: "your-email@gmail.com"
+  email_password: "your-app-password"  # Use app-specific password, not regular password
+

main.py

@@ -22,6 +22,7 @@
     PdfCompiler,
     PaperScheduler
 )
+from modules.notifications import NotificationManager, send_test_notification
 
 
 def setup_logging(config: dict) -> logging.Logger:
@@ -219,9 +220,28 @@ def run_collector(config: dict, logger: logging.Logger) -> bool:
             logger.info("=" * 60)
             logger.info("Collection completed successfully!")
             logger.info("=" * 60)
+
+            # Send notification
+            try:
+                notification_config = config.get("notifications", {})
+                notifier = NotificationManager(notification_config)
+                total_papers = sum(len(paper_list) for paper_list in grouped_papers.values())
+                notifier.send_notification(total_papers, pdf_path)
+            except Exception as e:
+                logger.warning(f"Failed to send notification: {e}")
+
             return True
         else:
             logger.error("PDF compilation failed")
+
+            # Send failure notification
+            try:
+                notification_config = config.get("notifications", {})
+                notifier = NotificationManager(notification_config)
+                notifier.send_notification(0, "", error="PDF compilation failed")
+            except Exception:
+                pass  # Don't fail on notification errors
+
             return False
 
     except Exception as e:
@@ -280,6 +300,18 @@ def main():
         help="Open config file in default editor for keyword editing"
     )
 
+    parser.add_argument(
+        "--test-notify",
+        action="store_true",
+        help="Send a test notification to verify notification setup"
+    )
+
+    parser.add_argument(
+        "--setup-schedule",
+        action="store_true",
+        help="Set up automatic scheduled execution (cron/LaunchAgent)"
+    )
+
     args = parser.parse_args()
 
     # Load configuration
@@ -293,6 +325,30 @@ def main():
         os.system(f"{editor} {args.config}")
         return
 
+    # Handle --test-notify
+    if args.test_notify:
+        print("Sending test notification...")
+        notification_config = config.get("notifications", {})
+        success = send_test_notification(notification_config)
+        if success:
+            print("✓ Test notification sent!")
+            print("  Check your desktop/system for the notification.")
+        else:
+            print("⚠ Test notification failed.")
+            print("  For system notifications, ensure your OS supports desktop notifications.")
+            print("  For email notifications, configure email settings in config.yaml.")
+        return
+
+    # Handle --setup-schedule
+    if args.setup_schedule:
+        print("Setting up scheduled execution...")
+        if os.path.exists("setup-schedule.sh"):
+            os.execv("/bin/bash", ["bash", "setup-schedule.sh"])
+        else:
+            print("Error: setup-schedule.sh not found")
+            sys.exit(1)
+        return
+
     # Handle --status
     if args.status:
         schedule_config = config.get("schedule", {})

modules/__init__.py

@@ -10,11 +10,16 @@
 from .latex_generator import LatexGenerator
 from .pdf_compiler import PdfCompiler
 from .scheduler import PaperScheduler
+from .notifications import NotificationManager, send_test_notification
+from .config_loader import ConfigLoader
 
 __all__ = [
     "ArxivFetcher",
     "PaperFilter",
     "LatexGenerator",
     "PdfCompiler",
     "PaperScheduler",
+    "NotificationManager",
+    "send_test_notification",
+    "ConfigLoader",
 ]

modules/notifications.py

@@ -0,0 +1,213 @@
+"""
+Notification Module
+Supports email and system notifications for completed collections
+"""
+
+import os
+import smtplib
+import subprocess
+import platform
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+from typing import Optional, List
+import logging
+
+
+class NotificationManager:
+    """Manages notifications for paper collection completion"""
+
+    def __init__(self, config: dict = None):
+        """
+        Initialize notification manager
+
+        Args:
+            config: Configuration dictionary with notification settings
+        """
+        self.config = config or {}
+        self.logger = logging.getLogger(__name__)
+
+    def send_notification(
+        self,
+        num_papers: int,
+        pdf_path: str,
+        error: Optional[str] = None
+    ) -> bool:
+        """
+        Send notification based on configured methods
+
+        Args:
+            num_papers: Number of papers collected
+            pdf_path: Path to generated PDF
+            error: Error message if collection failed
+
+        Returns:
+            True if notification was sent successfully
+        """
+        success = False
+
+        # Determine message content
+        if error:
+            title = "ArXiv Paper Collection - Failed"
+            message = f"Paper collection failed with error:\n\n{error}"
+        else:
+            title = "ArXiv Paper Collection - Completed"
+            message = f"Successfully collected {num_papers} papers\n\nPDF: {pdf_path}"
+
+        # Send system notification
+        if self.config.get("system_notification", True):
+            success |= self._send_system_notification(title, message)
+
+        # Send email notification
+        if self.config.get("email_enabled", False):
+            success |= self._send_email(title, message)
+
+        return success
+
+    def _send_system_notification(self, title: str, message: str) -> bool:
+        """
+        Send desktop system notification
+
+        Args:
+            title: Notification title
+            message: Notification message
+
+        Returns:
+            True if successful
+        """
+        try:
+            system = platform.system()
+
+            if system == "Darwin":  # macOS
+                # Use osascript for macOS notifications
+                cmd = [
+                    "osascript", "-e",
+                    f'display notification "{message}" with title "{title}" sound name "default"'
+                ]
+                subprocess.run(cmd, check=True, capture_output=True)
+                return True
+
+            elif system == "Linux":
+                # Check for notify-send
+                if self._command_exists("notify-send"):
+                    cmd = [
+                        "notify-send",
+                        title,
+                        message,
+                        "-i", "dialog-information"
+                    ]
+                    subprocess.run(cmd, check=True, capture_output=True)
+                    return True
+                else:
+                    self.logger.warning("notify-send not found. Install: sudo apt-get install libnotify-bin")
+                    return False
+
+            elif system == "Windows":
+                # Use Windows toast notification
+                try:
+                    from win10toast import ToastNotifier
+                    toaster = ToastNotifier()
+                    toaster.show_toast(
+                        title,
+                        message,
+                        duration=10,
+                        threaded=True
+                    )
+                    return True
+                except ImportError:
+                    self.logger.warning("win10toast not installed. Install: pip install win10toast")
+                    return False
+
+        except Exception as e:
+            self.logger.error(f"System notification failed: {e}")
+
+        return False
+
+    def _send_email(self, subject: str, body: str) -> bool:
+        """
+        Send email notification
+
+        Args:
+            subject: Email subject
+            body: Email body
+
+        Returns:
+            True if successful
+        """
+        try:
+            # Get email settings from config
+            smtp_server = self.config.get("smtp_server", "smtp.gmail.com")
+            smtp_port = self.config.get("smtp_port", 587)
+            email_from = self.config.get("email_from")
+            email_to = self.config.get("email_to")
+            email_password = self.config.get("email_password")
+
+            if not all([email_from, email_to, email_password]):
+                self.logger.warning("Email configuration incomplete. Skipping email notification.")
+                return False
+
+            # Create message
+            msg = MIMEMultipart()
+            msg['From'] = email_from
+            msg['To'] = email_to
+            msg['Subject'] = subject
+
+            msg.attach(MIMEText(body, 'plain'))
+
+            # Send email
+            with smtplib.SMTP(smtp_server, smtp_port) as server:
+                server.starttls()
+                server.login(email_from, email_password)
+                server.send_message(msg)
+
+            self.logger.info(f"Email notification sent to {email_to}")
+            return True
+
+        except Exception as e:
+            self.logger.error(f"Email notification failed: {e}")
+            return False
+
+    @staticmethod
+    def _command_exists(command: str) -> bool:
+        """Check if a command exists on the system"""
+        try:
+            subprocess.run(
+                ["which", command],
+                check=True,
+                capture_output=True
+            )
+            return True
+        except subprocess.CalledProcessError:
+            return False
+
+    @staticmethod
+    def get_email_config_template() -> dict:
+        """Get template for email configuration"""
+        return {
+            "notifications": {
+                "system_notification": True,
+                "email_enabled": False,
+                "smtp_server": "smtp.gmail.com",
+                "smtp_port": 587,
+                "email_from": "your-email@gmail.com",
+                "email_to": "your-email@gmail.com",
+                "email_password": "your-app-password"  # Use app-specific password
+            }
+        }
+
+
+def send_test_notification(config: dict = None) -> bool:
+    """
+    Send a test notification to verify setup
+
+    Args:
+        config: Notification configuration
+
+    Returns:
+        True if test notification was sent
+    """
+    manager = NotificationManager(config)
+    return manager.send_notification(
+        num_papers=0,
+        pdf_path="test.pdf",
+        error=None
+    )