Update app.py

app.py

@@ -194,28 +194,39 @@ st.markdown("github : https://github.com/ldh-Hoon/ko_deepfake-whisper-features")
 
 tab1, tab2, tab3 = st.tabs(["YouTube URL", "파일 업로드", "실시간 오디오 입력"])
 
-with tab1:
-    youtube_url = st.text_input("YouTube URL")
-    st.markdown("""example 
->fake: 
-
-https://youtu.be/ha3gfD7S0_E
+example_urls_fake = [
+    "https://youtu.be/ha3gfD7S0_E",
+    "https://youtu.be/5lmJ0Rhr-ec",
+    "https://youtu.be/q6ra0KDgVbg",
+    "https://youtu.be/hfmm1Oo6SSY?feature=shared"
+]
 
-https://youtu.be/5lmJ0Rhr-ec
+example_urls_real = [
+    "https://youtu.be/54y1sYLZjqs",
+    "https://youtu.be/7qT0Stb3QNY",
+]
 
-https://youtu.be/q6ra0KDgVbg
+if 'youtube_url' not in st.session_state:
+    st.session_state['youtube_url'] = ''
 
-https://youtu.be/hfmm1Oo6SSY?feature=shared
-
->real:
+with tab1:
+    st.markdown("""example
+>fake:
+""")
+    for url in example_urls_fake:
+        if st.button(url, key=url):
+            st.session_state.youtube_url = url
 
-https://youtu.be/54y1sYLZjqs
+    st.markdown(""">real:
+""")
+    for url in example_urls_real:
+        if st.button(url, key=url):
+            st.session_state.youtube_url = url
 
-https://youtu.be/7qT0Stb3QNY
+    youtube_url = st.text_input("YouTube URL", value=st.session_state.youtube_url)
 
-""")
     if youtube_url:
-        result = pred_from_url(youtube_url)
+        result = pred_from_url(youtube_url)  # 여기에 pred_from_url 함수 정의가 필요합니다.
         st.text_area("결과", value=result, height=150)
         st.video(youtube_url)
 
@@ -238,4 +249,4 @@ with tab3:
                 f.write(wav_audio_data)
         result = pred_from_file(file)
         st.text_area("결과", value=result, height=150)
-        os.remove(file)  # 임시 파일 삭제
+        os.remove(file)