Update app.py

app.py

@@ -7,6 +7,18 @@ from difflib import get_close_matches
 
 CHANNELS_JSON = "channels_fixed.json"
 
+# Abbreviations mapping for Events channels
+ABBREVIATIONS = {
+    "BTN": "BIG TEN Network (BTN USA)"
+    # Add more abbreviations here if needed
+}
+
+# Special cases mapping
+SPECIAL_CASES = {
+    "MLB LEAGUE PASS": "MLB.Baseball.Dummy.us",
+    "NHL GAMECENTER": "NHL.Hockey.Dummy.us"
+}
+
 def normalize(name):
     name = name.lower()
     name = re.sub(r'[^a-z0-9]', '', name)
@@ -50,7 +62,6 @@ def process_playlist(m3u_content):
         if current_section == '24_7':
             m3u_24_7.append(line)
         elif current_section == 'events':
-            # Remove 24/7 group-title if mistakenly present
             line = re.sub(r'group-title="24/7 CHANNELS[^"]*"', '', line)
             m3u_events.append(line)
         else:
@@ -66,30 +77,41 @@ def process_playlist(m3u_content):
                 parts = line.split(',')
                 ch_name = parts[-1].strip()
 
+                new_tvg_id = None
+
                 if is_event:
                     match_brackets = re.search(r'\[(.*?)\]', ch_name)
                     ch_name_to_match = match_brackets.group(1) if match_brackets else ch_name
 
-                    ch_name_norm = normalize(ch_name_to_match)
-                    json_keys_norm = {k: normalize(k) for k in channels_ref.keys()}
-
-                    match_key = None
-                    matches = get_close_matches(ch_name_norm, list(json_keys_norm.values()), n=1, cutoff=0.7)
-                    if matches:
-                        for k, v in json_keys_norm.items():
-                            if v == matches[0]:
-                                match_key = k
-                                break
-
-                    new_tvg_id = channels_ref[match_key]['tvg_id'] if match_key else 'Live.Event.us'
-                    # Replace any tvg-id="test" with Live.Event.us
-                    if 'tvg-id="test"' in line:
-                        new_tvg_id = 'Live.Event.us'
+                    if ch_name_to_match in SPECIAL_CASES:
+                        new_tvg_id = SPECIAL_CASES[ch_name_to_match]
+                    elif ch_name_to_match in ABBREVIATIONS:
+                        match_key = ABBREVIATIONS[ch_name_to_match]
+                        new_tvg_id = channels_ref.get(match_key, {}).get('tvg_id', 'Live.Event.us')
+                    else:
+                        if re.search(r'ATP|WTA', ch_name_to_match, re.IGNORECASE):
+                            new_tvg_id = 'Tennis.Channel.us'
+                        else:
+                            ch_name_norm = normalize(ch_name_to_match)
+                            json_keys_norm = {k: normalize(k) for k in channels_ref.keys()}
+                            match_key = None
+                            matches = get_close_matches(ch_name_norm, list(json_keys_norm.values()), n=1, cutoff=0.7)
+                            if matches:
+                                for k, v in json_keys_norm.items():
+                                    if v == matches[0]:
+                                        match_key = k
+                                        break
+                            new_tvg_id = channels_ref[match_key]['tvg_id'] if match_key else 'Live.Event.us'
+
+                    # Ensure tvg-id="test" is replaced with tvg-id="Live.Event.us"
+                    line = re.sub(r'tvg-id="test"', 'tvg-id="Live.Event.us"', line)
                 else:
                     match_key = fuzzy_match(ch_name, channels_ref.keys())
                     new_tvg_id = channels_ref[match_key]['tvg_id'] if match_key else 'Info.Guide.Dummy.us'
-                    if 'tvg-id="test"' in line:
-                        new_tvg_id = 'Info.Guide.Dummy.us'
+                    line = re.sub(r'tvg-id="test"', 'tvg-id="Info.Guide.Dummy.us"', line)
+
+                if is_event:
+                    line = re.sub(r'group-title="[^"]*"', '', line)
 
                 if 'tvg-id=' in line:
                     line = re.sub(r'tvg-id=".*?"', f'tvg-id="{new_tvg_id}"', line)
@@ -126,7 +148,7 @@ demo = gr.Interface(
         gr.File(label="Download Events M3U")
     ],
     title="Project 1 Playlist Updater - Smart Events",
-    description="Paste the M3U8 playlist content. Updates 24/7 and Events channels tvg-id using channels_fixed.json with smart Events matching."
+    description="Paste the M3U8 playlist content. Updates 24/7 and Events channels tvg-id using channels_fixed.json with smart Events matching. Handles special cases like MLB, NHL, ATP/WTA. Replaces tvg-id=\"test\" properly."
 )
 
 demo.launch()