Update app.py

app.py

@@ -4,125 +4,119 @@ import feedparser
 import streamlit as st
 from functions import *
 
-st.title("PodcastGPT Dashboard")
-
-st.markdown(
-"""
-This app assists busy professionals with transcribing and summarizing Podcasts from [Listen Notes](https://www.listennotes.com/) website by following the below steps:
-- Search for your desired podcast from the Listen Notes website and click on the "RSS" tab to generate a unique link to the podcast. Example shown below: 
-"""
-)
-
-col1, col2 = st.columns(2)
-
-with col1:
-    
-    st.image('RSS.png', caption = 'Click RSS')
-
-with col2:
-
-    st.image('RSS_copy.png',caption='Copy RSS Link')
-
-st.markdown(
-"""
-- Copy the generated link and paste in the sidebar on the left: 
-"""
-)
-
-pods = {}
-
-# Left section - Input fields
-st.sidebar.header("Podcast RSS Feeds")
-
-# Input Box
-podcast_url = st.sidebar.text_input('Please paste the podcast RSS feed link here')
-
-latest_ep_button = st.sidebar.button("Get latest 5 Episodes")
-
-# st.sidebar.markdown("**Note**: Podcast processing can take upto 5 mins, please be patient.")
-
-if latest_ep_button is not False:
-    
-    #Extract the list of episodes for the given podcast
-    podcast_feed = feedparser.parse(podcast_url)
-
-    for pod in podcast_feed.entries[:5]:
-        podcast_title = pod['title']
-    
-        try:
-            podcast_image = pod['image']['href']
-    
-        except:
-            podcast_image = ''
-    
-        for i in pod['links']:
-            if i['type'] == 'audio/mpeg':
-                podcast_url = i['href']
-
-        pods.update({pod['title']:[podcast_url,podcast_image]})
-        
-    #Get the most recent 5 episodes
-    podcast_five_titles = list(pods.keys())
-
-    # Dropdown box
-    st.sidebar.subheader("Available Podcasts Feeds")
-    
-    selected_podcast = st.sidebar.selectbox("Select Podcast", options=podcast_five_titles,index=None,key='podcast_selection')
-
-    print(selected_podcast)
-
-    if selected_podcast is not None:
-    
-        if 'podcast_selection' not in st.session_state:
-            st.session_state.podcast_selection = selected_podcast
-    
-        print(selected_podcast)
-        
-        st.sidebar.markdown("**Note**: Podcast processing can take upto 5 mins, please be patient.")
-        
-        podcast_link = pods[st.session_state.podcast_selection][0]
-        podcast_image = pods[st.session_state.podcast_selection][1] 
+def main():
     
-        # Right section - Newsletter content
-        st.header("Newsletter Content")
+    st.title("PodcastGPT Dashboard")
     
-        # Display the podcast title
-        st.subheader("Episode Title")
-        st.write(selected_podcast)
+    st.markdown(
+    """
+    This app assists busy professionals with transcribing and summarizing Podcasts from [Listen Notes](https://www.listennotes.com/) website by following the below steps:
+    - Search for your desired podcast from the Listen Notes website and click on the "RSS" tab to generate a unique link to the podcast. Example shown below: 
+    """
+    )
     
-        # Display the podcast summary and the cover image in a side-by-side layout
-        col1, col2 = st.columns([7, 3])
+    col1, col2 = st.columns(2)
     
-        # Get podcast transcription and info
-        podcast_info = process_podcast(podcast_link)
+    with col1:
         
-        with col1:
-            # Display the podcast episode summary
-            st.subheader("Podcast Episode Summary")
-            st.write(podcast_info['podcast_summary'])
-    
-        if podcast_image:
-            with col2:
-                st.image(podcast_image, caption="Podcast Cover", width=300, use_column_width=True)
-    
-        # Display the podcast guest and their details in a side-by-side layout
-        col3, col4 = st.columns([3, 7])
-    
-        with col3:
-            st.subheader("Podcast Guest")
-            st.write(podcast_info['podcast_guest'])
-    
-        with col4:
-            st.subheader("Podcast Guest Details")
-            st.write(podcast_info["podcast_guest_title"])
-            st.write(podcast_info["podcast_guest_org"])
-    
-        # Display the five key moments
-        st.subheader("Key Moments")
-        key_moments = podcast_info['podcast_highlights']
-        for moment in key_moments.split('\n'):
-            st.markdown(
-                f"<p style='margin-bottom: 5px;'>{moment}</p>", unsafe_allow_html=True)
+        st.image('RSS.png', caption = 'Click RSS')
+    
+    with col2:
+    
+        st.image('RSS_copy.png',caption='Copy RSS Link')
+    
+    st.markdown(
+    """
+    - Copy the generated link and paste in the sidebar on the left: 
+    """
+    )
+    
+    pods = {}
+    
+    # Left section - Input fields
+    st.sidebar.header("Podcast RSS Feeds")
+    
+    # Input Box
+    podcast_url = st.sidebar.text_input('Please paste the podcast RSS feed link here')
+    
+    latest_ep_button = st.sidebar.button("Get latest 5 Episodes")
+    
+    # st.sidebar.markdown("**Note**: Podcast processing can take upto 5 mins, please be patient.")
+    
+    if latest_ep_button:
+        # Extract the list of episodes for the given podcast
+        podcast_feed = feedparser.parse(podcast_url)
+    
+        # Ensure you don't exceed the feed's length
+        num_episodes = min(len(podcast_feed.entries), 5)
+    
+        for pod in podcast_feed.entries[:num_episodes]:
+            podcast_title = pod['title']
+            podcast_image = pod.get('image', {}).get('href', '')
+    
+            episode_url = next((i['href'] for i in pod['links'] if i['type'] == 'audio/mpeg'), None)
+    
+            pods[podcast_title] = [episode_url, podcast_image]
+    
+        if pods:
+            # Dropdown box
+            st.sidebar.subheader("Available Podcasts Feeds")
+            podcast_five_titles = list(pods.keys())
+            selected_podcast = st.sidebar.selectbox("Select Podcast", options=podcast_five_titles)
+    
+            if selected_podcast:
+                # Process selected podcast
+                podcast_link, podcast_image = pods[selected_podcast]
+                
+                st.sidebar.markdown("**Note**: Podcast processing can take upto 5 mins, please be patient.")
+                
+                podcast_link = pods[st.session_state.podcast_selection][0]
+                podcast_image = pods[st.session_state.podcast_selection][1] 
+            
+                # Right section - Newsletter content
+                st.header("Newsletter Content")
+            
+                # Display the podcast title
+                st.subheader("Episode Title")
+                st.write(selected_podcast)
+            
+                # Display the podcast summary and the cover image in a side-by-side layout
+                col1, col2 = st.columns([7, 3])
+            
+                # Get podcast transcription and info
+                podcast_info = process_podcast(podcast_link)
+                
+                with col1:
+                    # Display the podcast episode summary
+                    st.subheader("Podcast Episode Summary")
+                    st.write(podcast_info['podcast_summary'])
+            
+                if podcast_image:
+                    with col2:
+                        st.image(podcast_image, caption="Podcast Cover", width=300, use_column_width=True)
+            
+                # Display the podcast guest and their details in a side-by-side layout
+                col3, col4 = st.columns([3, 7])
+            
+                with col3:
+                    st.subheader("Podcast Guest")
+                    st.write(podcast_info['podcast_guest'])
+            
+                with col4:
+                    st.subheader("Podcast Guest Details")
+                    st.write(podcast_info["podcast_guest_title"])
+                    st.write(podcast_info["podcast_guest_org"])
+            
+                # Display the five key moments
+                st.subheader("Key Moments")
+                key_moments = podcast_info['podcast_highlights']
+                for moment in key_moments.split('\n'):
+                    st.markdown(
+                        f"<p style='margin-bottom: 5px;'>{moment}</p>", unsafe_allow_html=True)
+
+
+if __name__ == '__main__':
+    main()