File size: 4,161 Bytes
092aaf8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 | """
Interactive runner for Agent 1 - Change Detection.
Walks you through:
1. Choosing the item type from a list
2. For each required angle, uploading the BASELINE (pickup) photo and the
RETURN photo, one pair at a time, via a native file-picker dialog
3. Optionally entering the student's complaint text
4. Running the pipeline and printing a clear summary
Run: python run_interactive.py
"""
import os
import json
from change_detection import run_agent1, ITEM_TYPE_ANGLES
def pick_file(window_title):
"""Open a native file-picker dialog. Falls back to typed path entry if
tkinter isn't available (e.g. headless environments)."""
try:
import tkinter as tk
from tkinter import filedialog
root = tk.Tk()
root.withdraw()
root.attributes("-topmost", True)
path = filedialog.askopenfilename(
title=window_title,
filetypes=[("Image files", "*.jpg *.jpeg *.png *.bmp *.webp"), ("All files", "*.*")],
)
root.destroy()
if path:
return path
print(" No file selected, please try again.")
return pick_file(window_title)
except Exception:
# fallback: manual path entry
while True:
path = input(f"{window_title}\n Enter the full file path: ").strip().strip('"')
if os.path.isfile(path):
return path
print(f" File not found: {path}")
def choose_item_type():
"""Print a numbered menu of all known item types (from ITEM_TYPE_ANGLES) and return the chosen key."""
types = list(ITEM_TYPE_ANGLES.keys())
print("\nSelect item type:")
for i, t in enumerate(types, 1):
print(f" {i}. {t}")
while True:
choice = input("Enter number: ").strip()
if choice.isdigit() and 1 <= int(choice) <= len(types):
return types[int(choice) - 1]
print("Invalid choice, try again.")
def main():
"""Full interactive flow: item type -> per-angle photo pairs -> optional complaint -> run_agent1() -> print results."""
print("=== Agent 1 - Change Detection ===")
item_id = input("Item ID (e.g. item_laptop_014, press Enter for a default): ").strip() or "item_test_001"
item_type = choose_item_type()
angles = ITEM_TYPE_ANGLES[item_type]
print(f"\nItem type '{item_type}' requires {len(angles)} angle(s): {', '.join(angles)}")
print("For each angle you'll upload the BASELINE (pickup) photo, then the RETURN photo.\n")
baseline_photos = []
return_photos = []
for angle in angles:
print(f"--- Angle: {angle} ---")
print(f"[{angle}] Select the BASELINE (pickup) photo...")
base_path = pick_file(f"Baseline photo - angle: {angle}")
print(f" Baseline: {base_path}")
print(f"[{angle}] Select the RETURN photo...")
return_path = pick_file(f"Return photo - angle: {angle}")
print(f" Return: {return_path}\n")
baseline_photos.append({"angle": angle, "path": base_path})
return_photos.append({"angle": angle, "path": return_path})
complaint_text = input("Optional: student complaint text (press Enter to skip): ").strip() or None
print("\nRunning change detection...\n")
result = run_agent1(
item_id=item_id,
item_type=item_type,
baseline_photos=baseline_photos,
return_photos=return_photos,
complaint_text=complaint_text,
crop_output_dir="agent1_crops",
)
print("=" * 50)
print("RESULT")
print("=" * 50)
print(f"Overall change detected : {result['overall_change_detected']}")
print(f"Overall change score : {result['overall_change_score']}")
print(f"Routing hint : {result['routing_hint']}")
print()
for a in result["angles"]:
print(f"- {a['angle']}: score={a['change_score']}, aligned_ok={a['aligned_ok']}, regions={len(a['regions'])}")
for r in a["regions"]:
print(f" region: bbox={r['bbox']} area={r['area']} crop={r['crop_path']}")
print("\nFull JSON output:")
print(json.dumps(result, indent=2))
if __name__ == "__main__":
main()
|