Update app.py

app.py

@@ -89,30 +89,38 @@ def page_social_media_generator():
         st.write("Please select an article from the 'What is trending in my niche?' page.")
 
         
-def page_article_to_social_post():
+def article_to_social_post():
     st.title("Article to Social Media Post")
     
-    # User input for the niche
     niche = st.text_input('Enter your niche', 'Technology')
     if niche:
         google_news = GNews(language='en', country='US')
         news_list = google_news.get_news(niche)
         
-        # Display articles as options in a selectbox
-        article_urls = [news['url'] for news in news_list[:5]]
+        if not news_list:
+            st.write("No news found for the given niche.")
+            return
+
+        # Use the titles directly for the selectbox
         article_titles = [news['title'] for news in news_list[:5]]
-        selected_url = st.selectbox("Select an article:", article_titles, format_func=lambda x: article_titles[article_urls.index(x)])
-        
-        if st.button('Fetch Full Article'):
-            article_text = fetch_full_article(selected_url)
-            st.write(article_text[:500], "...")  # Display a snippet of the article text
-        
-        if st.button('Generate Social Media Post'):
-            with st.spinner('Generating...'):
-                # Call the function to generate the post using the full article text
-                post_content = summarizer(article_text, max_length=130, min_length=30, do_sample=False)[0]['summary_text']
-                st.success('Generated Content:')
-                st.write(post_content)
+        selected_title = st.selectbox("Select an article:", article_titles)
+
+        # Find the selected article's URL based on the title
+        selected_article = next((item for item in news_list if item['title'] == selected_title), None)
+        if selected_article:
+            selected_url = selected_article['url']
+            # The rest of your code to fetch and display the article or generate the post...
+            if st.button('Fetch Full Article'):
+                article_text = fetch_full_article(selected_url)
+                st.write(article_text[:500], "...")  # Display a snippet of the article text
+            
+            if st.button('Generate Social Media Post'):
+                with st.spinner('Generating...'):
+                    post_content = summarizer(article_text, max_length=130, min_length=30, do_sample=False)[0]['summary_text']
+                    st.success('Generated Content:')
+                    st.write(post_content)
+        else:
+            st.error("Failed to find the selected article.")
 
 
 def page_test():