prthm11 commited on
Commit
599eea2
·
verified ·
1 Parent(s): 0628b7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -19
app.py CHANGED
@@ -68,26 +68,34 @@ backdrop_images_path = r"blocks\Backdrops"
68
  sprite_images_path = r"blocks\sprites"
69
 
70
  count = 0
71
- #BASE_DIR = Path(__file__).parent
72
- BASE_DIR = Path(__file__).parent.resolve()
73
- BLOCKS_DIR = BASE_DIR / "blocks"
74
- STATIC_DIR = BASE_DIR / "static"
75
- GEN_PROJECT_DIR = BASE_DIR / "generated_projects"
76
- BACKDROP_DIR = BLOCKS_DIR / "Backdrops"
77
- SPRITE_DIR = BLOCKS_DIR / "sprites"
78
- # OUTPUT_DIR = BASE_DIR / "OUTPUTS"
79
- # OUTPUT_DIR2 = BASE_DIR / "outputs"
80
-
81
- # DETECTED_IMAGE_DIR = OUTPUT_DIR / "DETECTED_IMAGE"
82
- DETECTED_IMAGE_DIR = os.path.join("outputs", "DETECTED_IMAGE")
83
- # IMAGE_DIR = OUTPUT_DIR / "SCANNED_IMAGE"
84
- IMAGE_DIR = os.path.join("outputs", "SCANNED_IMAGE")
85
- # JSON_DIR = OUTPUT_DIR / "EXTRACTED_JSON"
86
- JSON_DIR = os.path.join("outputs", "EXTRACTED_JSON")
87
-
88
- for d in (BLOCKS_DIR, STATIC_DIR, GEN_PROJECT_DIR, BACKDROP_DIR, SPRITE_DIR):
 
 
 
 
 
 
 
 
 
89
  d.mkdir(parents=True, exist_ok=True)
90
-
91
  # def classify_image_type(description_or_name: str) -> str:
92
  # desc = description_or_name.lower()
93
 
 
68
  sprite_images_path = r"blocks\sprites"
69
 
70
  count = 0
71
+ rom pathlib import Path
72
+
73
+ BASE_DIR = Path(__file__).parent.resolve()
74
+ BLOCKS_DIR = BASE_DIR / "blocks"
75
+ STATIC_DIR = BASE_DIR / "static"
76
+ GEN_PROJECT_DIR = BASE_DIR / "generated_projects"
77
+ BACKDROP_DIR = BLOCKS_DIR / "Backdrops"
78
+ SPRITE_DIR = BLOCKS_DIR / "sprites"
79
+
80
+ # === new: outputs rooted under BASE_DIR ===
81
+ OUTPUT_DIR = BASE_DIR / "outputs"
82
+ DETECTED_IMAGE_DIR = OUTPUT_DIR / "DETECTED_IMAGE"
83
+ SCANNED_IMAGE_DIR = OUTPUT_DIR / "SCANNED_IMAGE"
84
+ JSON_DIR = OUTPUT_DIR / "EXTRACTED_JSON"
85
+
86
+ # make all of them in one go
87
+ for d in (
88
+ BLOCKS_DIR,
89
+ STATIC_DIR,
90
+ GEN_PROJECT_DIR,
91
+ BACKDROP_DIR,
92
+ SPRITE_DIR,
93
+ OUTPUT_DIR,
94
+ DETECTED_IMAGE_DIR,
95
+ SCANNED_IMAGE_DIR,
96
+ JSON_DIR,
97
+ ):
98
  d.mkdir(parents=True, exist_ok=True)
 
99
  # def classify_image_type(description_or_name: str) -> str:
100
  # desc = description_or_name.lower()
101