Update app.py

app.py

@@ -60,10 +60,9 @@ def fetch_full_article(url):
 # Initialize the summarization pipeline with BART
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
-# Function to generate social media post based on the article description
-def generate_social_media_post(description):
-    description = description[:500]  # Adjust length as needed
-    summary = summarizer(description, max_length=130, min_length=30, do_sample=False)[0]['summary_text']
+def generate_social_media_post(article_text):
+    # Summarize the article text
+    summary = summarizer(article_text, max_length=130, min_length=30, do_sample=False)[0]['summary_text']
     return summary
 
 
@@ -93,50 +92,41 @@ def page_social_media_generator():
 def page_article_to_social_post():
     st.title("Article to Social Media Post")
     
+    # User input for niche
     niche = st.text_input('Enter your niche', 'Technology')
+    
     if niche:
+        # Fetch news articles
         google_news = GNews(language='en', country='US')
         news_list = google_news.get_news(niche)
         
         if not news_list:
             st.write("No news found for the given niche.")
             return
-
+        
+        # Display article titles in a selectbox
         article_titles = [news['title'] for news in news_list[:5]]
         selected_title = st.selectbox("Select an article:", article_titles)
-
+        
         selected_article = next((item for item in news_list if item['title'] == selected_title), None)
         if selected_article:
             selected_url = selected_article['url']
             
-            # 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'):
+                # Fetch the full article text
+                article_text = fetch_full_article(selected_url)
+                # Show a snippet of the article to the user
+                st.write(article_text[:500], "...")
+                
+                # Store the full article text in session state for later use
+                st.session_state.full_article_text = article_text
+                
+            if st.button('Generate Social Media Post') and 'full_article_text' in st.session_state:
                 with st.spinner('Generating...'):
-                    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.")
+                    # Generate a summary based on the full article text
+                    post_content = generate_social_media_post(st.session_state.full_article_text)
+                    st.success('Generated Content:')
+                    st.write(post_content)
 
 
 def page_test():