Update app.py

app.py

@@ -94,7 +94,6 @@ def page_article_to_social_post():
     st.title("Article to Social Media Post")
     
     niche = st.text_input('Enter your niche', 'Technology')
-    article_text = ""  # Initialize article_text with an empty string
     if niche:
         google_news = GNews(language='en', country='US')
         news_list = google_news.get_news(niche)
@@ -103,29 +102,43 @@ def page_article_to_social_post():
             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_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
             
+            # Use a session state to store the full article text
+            if 'full_article_text' not in st.session_state:
+                st.session_state.full_article_text = ""
+
+            if st.button('Fetch Full Article'):
+                try:
+                    # Fetch the full article text and store it in the session state
+                    st.session_state.full_article_text = fetch_full_article(selected_url)
+                    # Display the first 500 characters of the article
+                    st.write(st.session_state.full_article_text[:500], "...")
+                except Exception as e:
+                    st.error(f"Failed to fetch the article. Error: {e}")
+
             if st.button('Generate Social Media Post'):
                 with st.spinner('Generating...'):
-                    post_content = summarizer(article_text, max_length=150, min_length=30, do_sample=False)[0]['summary_text']
-                    st.success('Generated Content:')
-                    st.write(post_content)
+                    try:
+                        # Ensure there's text to summarize
+                        if st.session_state.full_article_text.strip():
+                            # Generate a summary from the full article text
+                            post_content = summarizer(st.session_state.full_article_text, max_length=150, min_length=30, do_sample=False)[0]['summary_text']
+                            st.success('Generated Content:')
+                            st.write(post_content)
+                        else:
+                            st.error("Article text is empty. Please fetch the article before generating a post.")
+                    except Exception as e:
+                        st.error(f"Failed to generate social media post. Error: {e}")
         else:
             st.error("Failed to find the selected article.")
 
 
-
 def page_test():
     st.title('Test Page')
     st.write('This is a test page with a test name.')