Update app.py

app.py

@@ -154,50 +154,53 @@ def get_my_model(url_data, progress=gr.Progress(track_tqdm=True)):
 # ==================== नया मॉडल स्कैनिंग लॉजिक (फिक्स) ====================
 def scan_models():
     """
-    logs फ़ोल्डर के अंदर किसी भी स्थान पर मौजूद .pth और .index फ़ाइलों को ढूंढता है।
-    हर मॉडल का नाम .pth फ़ाइल के बेसनेम (बिना एक्सटेंशन) से लिया जाता है।
-    संबंधित .index फ़ाइल उसी डायरेक्टरी में या logs फ़ोल्डर में खोजी जाती है।
+    logs फ़ोल्डर के अंदर कहीं भी .pth और .index जोड़ी ढूंढता है।
+    ड्रॉपडाउन के लिए (display_name, pth_path, idx_path) की सूची बनाता है।
     """
     logs_dir = "logs"
     if not os.path.isdir(logs_dir):
         return []
-
+    
     models = []
-    # पूरे logs फ़ोल्डर में .pth फ़ाइलें खोजें
+    # पहले सभी .pth फ़ाइलें ढूंढें
+    pth_files = []
     for root, dirs, files in os.walk(logs_dir):
-        for file in files:
-            if file.endswith(".pth"):
-                pth_path = os.path.join(root, file)
-                model_name = os.path.splitext(file)[0]
-                # संबंधित .index फ़ाइल खोजें (पहले उसी फ़ोल्डर में, फिर पूरे logs में)
-                idx_path = None
-                # उसी फ़ोल्डर में देखें
-                for idx_file in os.listdir(root):
-                    if idx_file.endswith(".index") and os.path.splitext(idx_file)[0] == model_name:
-                        idx_path = os.path.join(root, idx_file)
-                        break
-                # अगर न मिले तो पूरे logs में ढूंढें
-                if idx_path is None:
-                    for r, d, f in os.walk(logs_dir):
-                        for idx_file in f:
-                            if idx_file.endswith(".index") and os.path.splitext(idx_file)[0] == model_name:
-                                idx_path = os.path.join(r, idx_file)
-                                break
-                        if idx_path:
-                            break
-                # डुप्लिकेट नाम से बचने के लिए यूनिक key बनाएँ (फ़ोल्डर+नाम)
-                unique_key = os.path.relpath(pth_path, logs_dir).replace(os.sep, "_")
-                models.append((unique_key, pth_path, idx_path if idx_path else ""))
+        for f in files:
+            if f.endswith(".pth"):
+                pth_files.append(os.path.join(root, f))
+    
+    for pth_path in pth_files:
+        base = os.path.splitext(pth_path)[0]  # बिना .pth
+        # .index फ़ाइल खोजें
+        idx_path = None
+        # पहले उसी फ़ोल्डर में
+        dir_name = os.path.dirname(pth_path)
+        for ext in ['.index', '.added.index']:
+            candidate = base + ext
+            if os.path.isfile(candidate):
+                idx_path = candidate
+                break
+        # अगर न मिले तो पूरे logs में ढूंढें
+        if not idx_path:
+            for ext in ['.index', '.added.index']:
+                candidate = base + ext
+                if os.path.isfile(candidate):
+                    idx_path = candidate
+                    break
+        # अगर .index मिल गया तो ही मॉडल को लिस्ट में डालें
+        if idx_path and os.path.isfile(idx_path):
+            # डिस्प्ले नाम: फ़ोल्डर/फ़ाइलनाम (बिना .pth)
+            rel_path = os.path.relpath(pth_path, logs_dir)
+            display_name = os.path.splitext(rel_path)[0].replace(os.sep, " > ")
+            models.append((display_name, pth_path, idx_path))
+    
     return models
 
-def update_model_paths(selected_key):
-    """
-    ड्रॉपडाउन से चुने गए मॉडल के लिए .pth और .index के पूरे पाथ लौटाता है।
-    """
+def update_model_paths(display_name):
     models = scan_models()
-    for key, pth, idx in models:
-        if key == selected_key:
-            return pth, idx if idx else None
+    for name, pth, idx in models:
+        if name == display_name:
+            return pth, idx
     return None, None
 
 # ========== ऑडियो इफेक्ट्स ==========