File size: 2,806 Bytes
3fa9c8a f7667b8 3fb39a9 3fa9c8a f7667b8 3fb39a9 f7667b8 3fa9c8a f7667b8 3fb39a9 f7667b8 3fb39a9 f7667b8 82649d1 3fb39a9 f7667b8 3fb39a9 f7667b8 3fa9c8a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
import re
import requests
import gradio as gr
from pathlib import Path
import random
import string
def _random_name(ext=".m3u", length=3):
# Make a very short random filename like a9z.m3u
return "".join(random.choices(string.ascii_lowercase + string.digits, k=length)) + ext
def _to_raw_github(url: str) -> str:
if "github.com" in url and "/blob/" in url:
return url.replace("github.com", "raw.githubusercontent.com").replace("/blob/", "/")
return url
def _clean_parenthetical(text: str) -> str:
m = re.search(r"\(([^)]*)\)", text)
if not m:
return text
inside = m.group(1)
parts = [p.strip() for p in inside.split(" / ")]
mt_seg = None
for p in parts:
if re.search(r"\bMT\b", p):
mt_seg = p
break
if not mt_seg:
return text
mt_seg_clean = mt_seg.replace(",", "")
start, end = m.span()
return text[:start] + f"({mt_seg_clean})" + text[end:]
def _filter_text(content: str) -> str:
out_lines = []
for line in content.splitlines():
if "(" in line and " / " in line and "MT" in line:
out_lines.append(_clean_parenthetical(line))
else:
out_lines.append(line)
return "\n".join(out_lines)
def filter_from_url(url: str):
if not url or not url.strip():
return "Please paste a URL."
url = _to_raw_github(url.strip())
try:
r = requests.get(url, timeout=20)
except Exception as e:
return f"Error fetching file: {e}"
if r.status_code != 200:
return f"Error fetching file: HTTP {r.status_code}"
filtered = _filter_text(r.text)
out_path = Path("/tmp") / _random_name()
out_path.write_text(filtered, encoding="utf-8")
return str(out_path)
def filter_from_file(file_obj):
if file_obj is None:
return "Please upload a file."
text = Path(file_obj.name).read_text(encoding="utf-8", errors="ignore")
filtered = _filter_text(text)
out_path = Path("/tmp") / _random_name()
out_path.write_text(filtered, encoding="utf-8")
return str(out_path)
with gr.Blocks(title="M3U Cleaner") as demo:
gr.Markdown("### removes commas in short random name like `a9z`.")
with gr.Tab("From URL"):
url_in = gr.Textbox(label="M3U URL", placeholder="https://…/file.m3u8")
url_btn = gr.Button("Clean")
url_out = gr.File(label="Download filtered M3U")
url_btn.click(filter_from_url, inputs=url_in, outputs=url_out)
with gr.Tab("From File"):
file_in = gr.File(label="Upload .m3u or .m3u8")
file_btn = gr.Button("Clean")
file_out = gr.File(label="Download filtered M3U")
file_btn.click(filter_from_file, inputs=file_in, outputs=file_out)
demo.queue()
if __name__ == "__main__":
demo.launch()
|