Update app.py

app.py

@@ -8,13 +8,13 @@ os.system('apt install python3-libtorrent')
 
 import streamlit as st
 from streamlit_option_menu import option_menu
-from Torrent import torrent_info, torrent_download, torrent_group, simpan_txt
+from Torrent import torrent_info, torrent_download, torrent_group, simpan_txt, link_redirect
 
 
 # Navigasi Sidebar
 with st.sidebar:
-    selected = option_menu("Torrent", ['Magnet Link', 'Torrent Info'], 
-        icons=['search', 'house'], menu_icon="cast", default_index=0)
+    selected = option_menu("Torrent", ['Magnet Link', 'Index Torrent', 'Link Redirect'], 
+        icons=['search', 'house', 'fire'], menu_icon="cast", default_index=0)
 
 st.sidebar.image('DanMachi.png')
 
@@ -34,8 +34,8 @@ if selected == 'Magnet Link':
             selected_index  = int(file_priority.split('.')[0])
             downloaded_files = torrent_download(link, [selected_index])
 
-else:      
-    st.title("Magnet Link Torrent")
+elif selected == 'Index Torrent':      
+    st.title("Index Torrent")
     teks = st.text_area("Masukkan Teks", '', height=(25 * 10))
     tim = st.checkbox("Gak Perlu Dikelompokkan")
     if tim:
@@ -61,6 +61,25 @@ else:
             file_name=st.session_state.lokasi_file.replace('/home/user/app/', '').title().replace('Txt', 'txt'),
             mime='text/plain'
         )
+else:
+    st.title("Link Redirect")
+    link = st.text_input("Masukkan Link", "https://t.co/c31TBe7s4W")
+    
+    if st.button("Trace On"):
+        info = link_redirect(link)
+        lokasi_file = simpan_txt('Group Torrent', info)
+        st.session_state.info = info
+        st.session_state.lokasi_file = lokasi_file
 
-
-        
+    if 'info' in st.session_state:
+        num_lines = st.session_state.info.count('\n') + 1
+        text_area_height = 25 * num_lines
+        st.text_area("Hasil Redirect", st.session_state.info, height=text_area_height) 
+        with open(st.session_state.lokasi_file, 'rb') as f:
+            file_contents = f.read()
+        st.download_button(
+            label="Download File TXT",
+            data=file_contents,
+            file_name=st.session_state.lokasi_file.replace('/home/user/app/', '').title().replace('Txt', 'txt'),
+            mime='text/plain'
+        )