Update app.py

app.py

@@ -144,7 +144,7 @@ def get_model(weights: str) -> YOLO | None:
 #                   QUALITY‑EVALUATION (UNCHANGED from v3)
 # ---------------------------------------------------------------------------
 # --‑‑  <Functions qc_integrity / qc_class_balance / qc_image_quality ...>
-#  **(unchanged – omitted here for brevity; same as your previous v3 script)**
+#  **(unchanged – omitted for brevity; same as your previous v3 script)**
 # ---------------------------------------------------------------------------
 
 # ════════════════════════════════════════════════════════════════════════════
@@ -152,7 +152,7 @@ def get_model(weights: str) -> YOLO | None:
 # ════════════════════════════════════════════════════════════════════════════
 
 # -------------------- Roboflow helpers --------------------
-RF_RE = re.compile(r"https?://universe\.roboflow\.com/([^/]+)/([^/]+)/?([^/]*)")
+RF_RE = re.compile(r"https?://universe\.roboflow\.com/([^/]+)/([^/]+)/(.*)")
 
 def parse_roboflow_url(url: str) -> tuple[str, str, int | None]:
     """
@@ -432,14 +432,13 @@ with gr.Blocks(css="#classdf td{min-width:120px}") as demo:
         zips_in  = gr.Files(label="One or more dataset ZIPs")
         load_btn = gr.Button("Load datasets")
         load_log = gr.Markdown()
-        ds_state = gr.State([])  # List[(dloc, class_names, splits, name)]
+        ds_state = gr.State([])
 
         def _load_cb(rf_key, rf_urls_file, zip_files):
             global autoinc
             info_list = []
             log_lines = []
 
-            # Roboflow URLs via txt
             if rf_urls_file is not None:
                 for url in Path(rf_urls_file.name).read_text().splitlines():
                     if not url.strip():
@@ -451,7 +450,6 @@ with gr.Blocks(css="#classdf td{min-width:120px}") as demo:
                     except Exception as e:
                         log_lines.append(f"⚠️ RF load failed for {url!r}: {e}")
 
-            # ZIPs
             for f in zip_files or []:
                 autoinc += 1
                 tmp = TMP_ROOT / f"zip_{autoinc}"
@@ -469,7 +467,6 @@ with gr.Blocks(css="#classdf td{min-width:120px}") as demo:
 
         load_btn.click(_load_cb, [rf_key, rf_urls, zips_in], [ds_state, load_log])
 
-        # ------------- Class map editable table --------------------------
         gr.Markdown("### 2️⃣ Edit class mapping / limits / removal")
         class_df = gr.Dataframe(
             headers=["original_class", "new_name", "max_images", "remove"],
@@ -483,17 +480,15 @@ with gr.Blocks(css="#classdf td{min-width:120px}") as demo:
             for _dloc, names, _spl, _ in ds_info:
                 class_names_all.extend(names)
             class_names_all = sorted(set(class_names_all))
-            df = pd.DataFrame({
+            return pd.DataFrame({
                 "original_class": class_names_all,
                 "new_name":       class_names_all,
                 "max_images":     [99999] * len(class_names_all),
                 "remove":         [False]  * len(class_names_all),
             })
-            return df
 
         refresh_btn.click(_build_class_df, [ds_state], [class_df])
 
-        # ------------- Merge button & download ---------------------------
         merge_btn = gr.Button("Merge datasets ✨")
         zip_out   = gr.File(label="Download merged ZIP")
         merge_log = gr.Markdown()