Initial commit: IYmtg Master project

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-02-24 09:12:46 -05:00
commit b993ef4020
16 changed files with 3303 additions and 0 deletions
--- a/IYmtg_Automation/fetch_set_symbols.py
+++ b/IYmtg_Automation/fetch_set_symbols.py
@@ -0,0 +1,91 @@
+# INSTALL: pip install requests pillow
+import os, requests, time, concurrent.futures
+from PIL import Image
+from io import BytesIO
+
+HEADERS = { 'User-Agent': 'IYmtg/1.0 (contact@yourdomain.com)', 'Accept': 'application/json' }
+OUTPUT_DIR = "Set_Symbol_Training"
+
+def main():
+    if not os.path.exists(OUTPUT_DIR): os.makedirs(OUTPUT_DIR)
+    print("--- IYmtg Symbol Harvester ---")
+    
+    session = requests.Session()
+    session.headers.update(HEADERS)
+    
+    try:
+        response = session.get("https://api.scryfall.com/sets", timeout=15)
+        response.raise_for_status()
+        all_sets = response.json().get('data', [])
+    except Exception as e:
+        print(f"Error fetching sets: {e}")
+        return
+    
+    valid_types = ['core', 'expansion', 'masters', 'draft_innovation']
+    target_sets = [s for s in all_sets if s.get('set_type') in valid_types]
+    
+    print(f"Found {len(target_sets)} valid sets.")
+    
+    # OPTIMIZATION: Process sets in parallel to speed up dataset creation
+    with concurrent.futures.ThreadPoolExecutor(max_workers=5) as executor:
+        executor.map(lambda s: process_set(s, session), target_sets)
+
+    print("DONE. Drag 'Set_Symbol_Training' into Create ML -> Image Classification.")
+
+def process_set(set_obj, session):
+    set_code = set_obj['code']
+    print(f"Processing {set_code}...")
+    
+    set_dir = os.path.join(OUTPUT_DIR, set_code.upper())
+    os.makedirs(set_dir, exist_ok=True)
+    
+    try:
+        url = f"https://api.scryfall.com/cards/search?q=set:{set_code}+unique:prints&per_page=5"
+        resp = session.get(url, timeout=10)
+        
+        if resp.status_code != 200:
+            print(f"Skipping {set_code}: HTTP {resp.status_code}")
+            return
+        
+        cards_resp = resp.json()
+        
+        for i, card in enumerate(cards_resp.get('data', [])):
+            image_url = None
+            uris = {}
+            
+            if 'image_uris' in card:
+                uris = card['image_uris']
+            elif 'card_faces' in card and len(card['card_faces']) > 0 and 'image_uris' in card['card_faces'][0]:
+                uris = card['card_faces'][0]['image_uris']
+            
+            if 'large' in uris: image_url = uris['large']
+            elif 'normal' in uris: image_url = uris['normal']
+            
+            if image_url:
+                try:
+                    img_resp = session.get(image_url, timeout=10)
+                    if img_resp.status_code != 200: continue
+                    
+                    try:
+                        img = Image.open(BytesIO(img_resp.content))
+                        img.verify() # Verify integrity
+                        img = Image.open(BytesIO(img_resp.content)) # Re-open after verify
+                    except Exception:
+                        print(f"Skipping corrupt image in {set_code}")
+                        continue
+                    
+                    width, height = img.size
+                    # WARNING: This crop area is tuned for modern card frames (M15+).
+                    # Older sets or special frames (Planeswalkers, Sagas) may require different coordinates.
+                    crop_area = (width * 0.85, height * 0.58, width * 0.95, height * 0.65)
+                    
+                    symbol = img.crop(crop_area)
+                    symbol = symbol.convert("RGB")
+                    symbol.save(os.path.join(set_dir, f"sample_{i}.jpg"))
+                    
+                except Exception as e:
+                    print(f"Error downloading image for {set_code}: {e}")
+    except Exception as e:
+        print(f"Error searching cards for {set_code}: {e}")
+
+if __name__ == "__main__": main()