main.py

import tkinter as tk
from tkinter import filedialog, messagebox
import ui
import ocr_engine
import text_processor
import os

def main():
    root = tk.Tk()
    
    # Components
    processor = text_processor.TextProcessor()
    
    # Callback when OCR gets text
    def on_ocr_result(raw_text):
        final_text = processor.process_text(raw_text)
        if final_text:
            # Update UI on main thread
            root.after(0, app_ui.append_text, final_text)

    # Callback to get capture region
    def get_capture_region():
        # If focus box is hidden, maybe capture full screen?
        # For now, let's enforce using the box or default to full main screen if hidden?
        # PRD says "Focus Box". If hidden, maybe we shouldn't capture.
        # But for UX, if hidden, we could assume "Select a region"
        # Let's rely on the module logic: if hidden/small, it skips.
        # But we need to make sure the user knows to open it.
        return app_ui.get_focus_geometry()

    # Callbacks for UI buttons
    def start_capture():
        # Check if focus box is visible?
        # If not, maybe show warning or auto-show it.
        # For now, just start.
        if app_ui.focus_box.state() == 'withdrawn':
            app_ui.toggle_focus()
        
        ocr.start()

    def stop_capture():
        ocr.stop()
        processor.reset() # Optional: reset deduplication state on stop?

    def save_content(text):
        if not text.strip():
            messagebox.showinfo("Info", "Nothing to save.")
            return
            
        file_path = filedialog.asksaveasfilename(
            defaultextension=".txt",
            filetypes=[("Text files", "*.txt"), ("Markdown", "*.md"), ("All files", "*.*")]
        )
        if file_path:
            try:
                with open(file_path, "w", encoding="utf-8") as f:
                    f.write(text)
                messagebox.showinfo("Success", f"Saved to {file_path}")
            except Exception as e:
                messagebox.showerror("Error", f"Failed to save: {e}")

    app_ui = ui.VisionTextUI(root, start_capture, stop_capture, save_content)
    
    ocr = ocr_engine.OCREngine(on_ocr_result, get_capture_region)

    # Handle close
    def on_closing():
        ocr.stop()
        root.destroy()
        
    root.protocol("WM_DELETE_WINDOW", on_closing)

    root.mainloop()

if __name__ == "__main__":
    main()