Update app.py

app.py

@@ -16,16 +16,19 @@ def fetch_news(topic):
         })
     return articles
 
+from gnews import GNews
+
 def page_trending_niche():
-    col1, col2 = st.columns([3, 1])
-    with col1:
-        st.title("What is trending in my niche?")
-    with col2:
-        st.image('Robot.png', use_column_width=True)
-    
+    st.title("What is trending in my niche?")
     niche = st.text_input('Enter your niche', 'Technology')
     if niche:
         news_items = fetch_news(niche)
+        article_titles = [item['title'] for item in news_items]
+        selected_article = st.selectbox("Select an article to generate a social media post about:", article_titles)
+        selected_article_url = next((item['url'] for item in news_items if item['title'] == selected_article), None)
+        
+        # Save the selected article's URL in the session state to use in another page
+        st.session_state['selected_article_url'] = selected_article_url
         for item in news_items:
             st.write(f"**Title:** {item['title']}")
             st.write(f"**Published Date:** {item['published_date']}")
@@ -36,31 +39,36 @@ def page_trending_niche():
 
 
 
-# Define the pages
-def page_social_media_generator():
     
+from transformers import pipeline, set_seed
+generator = pipeline('text-generation', model='gpt2')
+set_seed(42)
+
+def page_social_media_generator():
+    st.title("Social Media Content Generator")
+
     # Using st.columns to create a two-column layout
     col1, col2 = st.columns([3, 1])
     with col1:
         st.title("German Medical Content Manager")
     with col2:
         st.image('Content_Creation_Pic.png', use_column_width=True)
-
     
-    input_topic = st.text_input('Enter a medical topic', 'Type 1 Diabetes')
-    st.write(f"Creating social media content for: {input_topic}")
-
-    def generate_content(topic):
-        generated_text = generator(f"Letzte Nachrichten über {topic}:", max_length=50, num_return_sequences=1)
-        return generated_text[0]['generated_text']
-
-    if st.button('Generate Social Media Post'):
-        with st.spinner('Generating...'):
-            post_content = generate_content(input_topic)
-            st.success('Generated Content:')
-            st.write(post_content)
+    # Retrieve the URL saved in the session state on the trending niche page
+    article_url = st.session_state.get('selected_article_url', '')
+    
+    if article_url:
+        st.write(f"Selected Article URL: {article_url}")
+        input_text = st.text_area("Enter additional context or leave empty to use only the article URL:", value=article_url)
+        
+        if st.button('Generate Social Media Post'):
+            with st.spinner('Generating...'):
+                post_content = generator(input_text, max_length=100, num_return_sequences=1)[0]['generated_text']
+                st.success('Generated Content:')
+                st.write(post_content)
+    else:
+        st.write("Please select an article from the 'What is trending in my niche?' page.")
 
-    st.write('Generated social media posts will appear here after clicking the "Generate" button.')
 
 def page_test():
     st.title('Test Page')