CX

app.py

@@ -1,4 +1,5 @@
 import os
+import re
 
 # os.system('pip install jedi')
 # os.system('python -m pip install --upgrade pip setuptools wheel')
@@ -10,18 +11,18 @@ import streamlit as st
 from streamlit_option_menu import option_menu
 from Torrent import torrent_info, torrent_download, torrent_group, simpan_txt, link_redirect, sesi
 
-
+option = ['Magnet Link', 'Index Torrent', 'Link Redirect', 'Kode']
 # Navigasi Sidebar
 with st.sidebar:
-    selected = option_menu("Torrent", ['Magnet Link', 'Index Torrent', 'Link Redirect', 'Kode'], 
+    selected = option_menu("Torrent", option, 
         icons=['search', 'house', 'fire', 'water'], menu_icon="cast", default_index=0)
 
 st.sidebar.image('DanMachi.png')
 
-if selected == 'Magnet Link':
+if selected == option[0]:
     url = 'DanMachi.png'
     st.image(url)
-    st.title("Magnet Link Torrent")
+    st.title(selected)
     link = st.text_input("Masukkan Magnet Link", "")
     search = link = st.text_input("Masukkan Pencarian", "")
     extension = st.selectbox('Pilih Extension', ['.mp4', '.zip'])
@@ -34,8 +35,8 @@ if selected == 'Magnet Link':
             selected_index  = int(file_priority.split('.')[0])
             downloaded_files = torrent_download(link, [selected_index])
 
-elif selected == 'Index Torrent':      
-    st.title("Index Torrent")
+elif selected == option[1]:
+    st.title(selected)
     teks = st.text_area("Masukkan Teks", '', height=(25 * 10))
     tim = st.checkbox("Gak Perlu Dikelompokkan")
     if tim:
@@ -48,8 +49,8 @@ elif selected == 'Index Torrent':
         lokasi_file = simpan_txt('Group Torrent', info)
         sesi(info, lokasi_file)
 
-elif selected == 'Link Redirect':      
-    st.title("Link Redirect")
+elif selected == option[2]:
+    st.title(selected)
     link = st.text_input("Masukkan Link", "https://t.co/c31TBe7s4W")
     
     if st.button("Trace On"):
@@ -58,11 +59,22 @@ elif selected == 'Link Redirect':
         sesi(info, lokasi_file)
 
 else:
-    st.title("Kode")
+    st.title(selected)
     link = st.text_input("Masukkan Link", "https://t.co/8Qes0WO3GV")
     
     if st.button("Trace On"):
         info = link_redirect(link)
-        info = info.split('%')[-10].split('D')[1].replace('000', '-0').upper()
-        lokasi_file = simpan_txt('Group Torrent', info)
+        info = info.split('%')[-10] #.split('D1')[1].split('D')[1].replace('000', '-0').upper()
+
+        # Pola regex untuk mencocokkan pola yang dijelaskan
+        pattern = r'[A-Za-z]+(\d{3})'
+
+        # Mencari kecocokan dengan pola regex
+        matches = re.findall(pattern, info)
+
+        if matches:
+            alfabet, number = matches[0][:3], matches[0][3:]
+            info = f"{alfabet.upper()}-{number.lstrip('0')}"
+
+        lokasi_file = simpan_txt('Group Torrent', info) 
         sesi(info, lokasi_file)