Update app.py

app.py

@@ -66,7 +66,6 @@ st.set_page_config(
     page_title="YouTube Talks",
     page_icon='📽️'
 )
-col1, col2 = st.columns(2)
 st.header('Query YouTube Videos!')
 thumbnail_placeholder = st.empty()
 st.sidebar.header("URL details:")
@@ -83,8 +82,8 @@ if submit_button:
     # get video transcription
     yt_obj = YouTube(video_url)
     # display video thumbnail
-    with col1:
-        thumbnail_placeholder.image(yt_obj.thumbnail_url, caption="Video Thumbnail")
+    st.session_state.thumbnail_url = yt_obj.thumbnail_url
+    thumbnail_placeholder.image(st.session_state.thumbnail_url)
     # get transcription
     get_yt_trans(yt_obj)
     st.sidebar.subheader("Transcription:")
@@ -113,12 +112,10 @@ Question: {question}'''
 
 if video_query_button:
     # display video thumbnail
-    with col1:
-        thumbnail_placeholder.image(yt_obj.thumbnail_url, caption="Video Thumbnail")
-    
+    thumbnail_placeholder.image(st.session_state.thumbnail_url)
+    # if video not transcribed display error
     if st.session_state.chain is None:
         st.error("Please transcribe a video first by submitting a URL.")
-        
     else:
         main_ph.text("Fetching Results...")
         # print results