spacr 0.1.0__py3-none-any.whl → 0.1.6__py3-none-any.whl

spacr/annotate_app_v2.py

@@ -1,16 +1,18 @@
+import sqlite3
 from queue import Queue
-from tkinter import Label, Entry, Button
+from tkinter import Label
 import tkinter as tk
+from tkinter import ttk
 import os, threading, time, sqlite3
 import numpy as np
 from PIL import Image, ImageOps
 from concurrent.futures import ThreadPoolExecutor
 from PIL import ImageTk
-import pandas as pd
 from skimage.exposure import rescale_intensity
-import cv2
-import matplotlib.pyplot as plt
 from IPython.display import display, HTML
+from tkinter import font as tkFont
+
+from .gui_utils import ScrollableFrame, CustomButton, set_dark_style, set_default_font, style_text_boxes, create_menu_bar
 
 class ImageApp:
     def __init__(self, root, db_path, src, image_type=None, channels=None, grid_rows=None, grid_cols=None, image_size=(200, 200), annotation_column='annotate', normalize=False, percentiles=(1,99), measurement=None, threshold=None):
@@ -509,3 +511,160 @@ def annotate(settings):
     
     app.load_images()
     root.mainloop()
+
+# Global list to keep references to PhotoImage objects
+global_image_refs = []
+
+def initiate_annotation_app_root(parent_frame):
+    style = ttk.Style(parent_frame)
+    set_dark_style(style)
+    style_text_boxes(style)
+    set_default_font(parent_frame, font_name="Arial", size=8)
+
+    parent_frame.configure(bg='black')
+    
+    container = tk.PanedWindow(parent_frame, orient=tk.HORIZONTAL, bg='black')
+    container.pack(fill=tk.BOTH, expand=True)
+
+    scrollable_frame = ScrollableFrame(container, bg='black')
+    container.add(scrollable_frame, stretch="always")
+
+    # Setup input fields
+    vars_dict = {
+        'src': ttk.Entry(scrollable_frame.scrollable_frame),
+        'image_type': ttk.Entry(scrollable_frame.scrollable_frame),
+        'channels': ttk.Entry(scrollable_frame.scrollable_frame),
+        'geom': ttk.Entry(scrollable_frame.scrollable_frame),
+        'img_size': ttk.Entry(scrollable_frame.scrollable_frame),
+        'rows': ttk.Entry(scrollable_frame.scrollable_frame),
+        'columns': ttk.Entry(scrollable_frame.scrollable_frame),
+        'annotation_column': ttk.Entry(scrollable_frame.scrollable_frame),
+        'normalize': ttk.Entry(scrollable_frame.scrollable_frame),
+        'percentiles': ttk.Entry(scrollable_frame.scrollable_frame),
+        'measurement': ttk.Entry(scrollable_frame.scrollable_frame),
+        'threshold': ttk.Entry(scrollable_frame.scrollable_frame),
+    }
+
+    default_settings = {
+        'src': 'path',
+        'image_type': 'cell_png',
+        'channels': 'r,g,b',
+        'geom': "3200x2000",
+        'img_size': '(200, 200)',
+        'rows': '10',
+        'columns': '18',
+        'annotation_column': 'recruited_test',
+        'normalize': 'False',
+        'percentiles': '(2,98)',
+        'measurement': 'None',
+        'threshold': 'None'
+    }
+
+    # Arrange input fields and labels
+    row = 0
+    for name, entry in vars_dict.items():
+        ttk.Label(scrollable_frame.scrollable_frame, text=f"{name.replace('_', ' ').capitalize()}:",
+                  background="black", foreground="white").grid(row=row, column=0)
+        entry.insert(0, default_settings[name])
+        entry.grid(row=row, column=1)
+        row += 1
+
+    # Function to be called when "Run" button is clicked
+    def run_app():
+        settings = {key: entry.get() for key, entry in vars_dict.items()}
+        settings['channels'] = settings['channels'].split(',')
+        settings['img_size'] = tuple(map(int, settings['img_size'][1:-1].split(',')))
+        settings['percentiles'] = tuple(map(int, settings['percentiles'][1:-1].split(',')))
+        settings['normalize'] = settings['normalize'].lower() == 'true'
+        settings['rows'] = int(settings['rows'])
+        settings['columns'] = int(settings['columns'])
+        if settings['measurement'].lower() == 'none':
+            settings['measurement'] = None
+        if settings['threshold'].lower() == 'none':
+            settings['threshold'] = None
+        
+        # Clear previous content instead of destroying the root
+        if hasattr(parent_frame, 'winfo_children'):
+            for widget in parent_frame.winfo_children():
+                widget.destroy()
+        
+        # Start the annotate application in the same root window
+        annotate_app(parent_frame, settings)
+        
+    run_button = CustomButton(scrollable_frame.scrollable_frame, text="Run", command=run_app,
+                              font=tkFont.Font(family="Arial", size=12, weight=tkFont.NORMAL))
+    run_button.grid(row=row, column=0, columnspan=2, pady=10, padx=10)
+
+    return parent_frame
+
+def annotate_app(parent_frame, settings):
+    global global_image_refs
+    global_image_refs.clear()
+    root = parent_frame.winfo_toplevel()
+    annotate_with_image_refs(settings, root, lambda: load_next_app(root))
+
+def annotate_with_image_refs(settings, root, shutdown_callback):
+    settings = get_annotate_default_settings(settings)
+    src = settings['src']
+
+    db = os.path.join(src, 'measurements/measurements.db')
+    conn = sqlite3.connect(db)
+    c = conn.cursor()
+    c.execute('PRAGMA table_info(png_list)')
+    cols = c.fetchall()
+    if settings['annotation_column'] not in [col[1] for col in cols]:
+        c.execute(f"ALTER TABLE png_list ADD COLUMN {settings['annotation_column']} integer")
+    conn.commit()
+    conn.close()
+
+    app = ImageApp(root, db, src, image_type=settings['image_type'], channels=settings['channels'], image_size=settings['img_size'], grid_rows=settings['rows'], grid_cols=settings['columns'], annotation_column=settings['annotation_column'], normalize=settings['normalize'], percentiles=settings['percentiles'], measurement=settings['measurement'], threshold=settings['threshold'])
+
+    # Set the canvas background to black
+    root.configure(bg='black')
+
+    next_button = tk.Button(root, text="Next", command=app.next_page, background='black', foreground='white')
+    next_button.grid(row=app.grid_rows, column=app.grid_cols - 1)
+    back_button = tk.Button(root, text="Back", command=app.previous_page, background='black', foreground='white')
+    back_button.grid(row=app.grid_rows, column=app.grid_cols - 2)
+    exit_button = tk.Button(root, text="Exit", command=lambda: [app.shutdown(), shutdown_callback()], background='black', foreground='white')
+    exit_button.grid(row=app.grid_rows, column=app.grid_cols - 3)
+
+    app.load_images()
+
+    # Store the shutdown function and next app details in the root
+    root.current_app_exit_func = app.shutdown
+    root.next_app_func = None
+    root.next_app_args = ()
+
+def load_next_app(root):
+    # Get the next app function and arguments
+    next_app_func = root.next_app_func
+    next_app_args = root.next_app_args
+
+    if next_app_func:
+        next_app_func(*next_app_args)
+
+def gui_annotate():
+    root = tk.Tk()
+    width = root.winfo_screenwidth()
+    height = root.winfo_screenheight()
+    root.geometry(f"{width}x{height}")
+    root.title("Annotate Application")
+    
+    # Clear previous content if any
+    if hasattr(root, 'content_frame'):
+        for widget in root.content_frame.winfo_children():
+            widget.destroy()
+        root.content_frame.grid_forget()
+    else:
+        root.content_frame = tk.Frame(root)
+        root.content_frame.grid(row=1, column=0, sticky="nsew")
+        root.grid_rowconfigure(1, weight=1)
+        root.grid_columnconfigure(0, weight=1)
+    
+    initiate_annotation_app_root(root.content_frame)
+    create_menu_bar(root)
+    root.mainloop()
+
+if __name__ == "__main__":
+    gui_annotate()