Update app.py

app.py

@@ -31,8 +31,11 @@ def get_available_formats(info_dict):
             if filesize:
                 filesize_mb = round(filesize / (1024 * 1024), 2)
                 resolution = f"{fmt['height']}p"
-                formats[resolution] = f"{resolution} - {filesize_mb} MB"
-    return list(formats.values())
+                formats[resolution] = {
+                    'id': fmt['format_id'],
+                    'info': f"{resolution} - {filesize_mb} MB"
+                }
+    return formats
 
 # "확인" 버튼을 누르면 처리할 로직
 if st.button('확인'):
@@ -60,7 +63,7 @@ if 'confirmed' in st.session_state and st.session_state.confirmed:
     st.image(st.session_state.thumbnail_url, caption='Video Thumbnail', use_column_width=True)
 
     # 해상도 선택
-    format_selection = st.selectbox('Select resolution', st.session_state.formats)
+    format_selection = st.selectbox('Select resolution', list(st.session_state.formats.values()))
 
     def download_video(youtube_url, format_id):
         ydl_opts = {
@@ -75,7 +78,7 @@ if 'confirmed' in st.session_state and st.session_state.confirmed:
 
     # "추출" 버튼 추가
     if st.button('추출'):
-        format_id = format_selection.split('p')[0]
+        format_id = next(key for key, value in st.session_state.formats.items() if value['info'] == format_selection)
         try:
             # 세션 상태의 youtube_url 사용
             download_video(st.session_state.youtube_url, format_id)