Update app.py

app.py

@@ -21,7 +21,7 @@ ARTICLE_LIMIT = 5  # Limit to the last 5 articles
 
 # News sources
 NEWS_SOURCES = {
-      "Technology": {
+    "Technology": {
         "TheNewYorkTimes": "https://rss.nytimes.com/services/xml/rss/nyt/Technology.xml",
         "reutersagency": "https://www.reutersagency.com/feed/?best-topics=tech&post_type=best"
     },
@@ -118,8 +118,8 @@ def summarize_text(text):
         logging.error(f"Summarization failed: {e}")
         return "Summary unavailable."
 
-def summarize_articles(articles):
-    """Summarize the last 5 fetched articles."""
+def summarize_articles(articles, template_choice):
+    """Summarize the last 5 fetched articles using the selected template."""
     summaries = []
     for article in articles:
         try:
@@ -134,37 +134,42 @@ def summarize_articles(articles):
             summary = summarize_text(content)
 
             if summary:
-              summaries.append(f"""
-               📰 {title}
-
-              - 📁 Category:  {category}  
-              - 💡 Source: {source}  
-              - ⏰ Published:  {published}  
-
-                📃 Summary: 
-                {summary}  
-            🔗 Read the full article here ({link})  
-
----
-""")
+                if template_choice == "Default":
+                    summaries.append(f"""
+                    📰 {title}
+
+                    - 📁 Category:  {category}  
+                    - 💡 Source: {source}  
+                    - ⏰ Published:  {published}  
+
+                    📃 Summary: 
+                    {summary}  
+                    🔗 Read the full article here ({link})  
+
+                    ---
+                    """)
+                elif template_choice == "Free":
+                    summaries.append(f"""
+                    **{title}**
+
+                    *Category:* {category}  
+                    *Source:* {source}  
+                    *Published:* {published}  
+
+                    *Summary:* 
+                    {summary}  
+                    *Link:* [{link}]({link})  
+
+                    ---
+                    """)
         except Exception as e:
             logging.error(f"Error summarizing article: {e}")
             continue
     return summaries
 
-def generate_user_summary(name):
-    """Generate a personalized news summary based on user preferences."""
-    # Load preferences
-    try:
-        with open(f"user_preferences/preferences_{name}.json") as f:
-            preferences = json.load(f)
-    except FileNotFoundError:
-        return "Preferences not found. Please set your preferences first."
-    except Exception as e:
-        logging.error(f"Error loading preferences: {e}")
-        return "Failed to load preferences."
-
-    categories = preferences.get("interests", [])
+def generate_summary(selected_interests, template_choice):
+    """Generate a news summary based on user preferences."""
+    categories = selected_interests
     if not categories:
         return "No categories selected. Please update your preferences."
 
@@ -174,7 +179,7 @@ def generate_user_summary(name):
         return "No recent news found in your selected categories."
 
     # Summarize all articles
-    summaries = summarize_articles(articles)
+    summaries = summarize_articles(articles, template_choice)
 
     # Combine and return summaries
     return "\n\n".join(summaries) if summaries else "No summaries available."
@@ -185,36 +190,21 @@ demo = gr.Blocks()
 with demo:
     gr.Markdown("# 📰 AI News Summarizer")
     
-    with gr.Tab("Set Preferences"):
-        name_input = gr.Textbox(label="Your Name")
+    with gr.Row():
         interests = gr.CheckboxGroup(
             choices=list(NEWS_SOURCES.keys()),
             label="Select Your Interests"
         )
-        save_button = gr.Button("Save Preferences")
-        save_status = gr.Textbox(label="Status")
-        
-        def save_preferences(name, selected_interests):
-            if not name or not selected_interests:
-                return "Name and interests are required!"
-            preferences = {"name": name, "interests": selected_interests}
-            try:
-                os.makedirs("user_preferences", exist_ok=True)
-                with open(f"user_preferences/preferences_{name}.json", "w") as f:
-                    json.dump(preferences, f)
-                return "Preferences saved successfully!"
-            except Exception as e:
-                logging.error(f"Failed to save preferences: {e}")
-                return "Failed to save preferences."
-
-        save_button.click(save_preferences, inputs=[name_input, interests], outputs=save_status)
-
-    with gr.Tab("Get News Summary"):
-        name_input_summary = gr.Textbox(label="Your Name")
-        fetch_button = gr.Button("Get Summary")
-        summary_output = gr.Textbox(label="News Summary", lines=20)
-        
-        fetch_button.click(generate_user_summary, inputs=[name_input_summary], outputs=summary_output)
+        template_choice = gr.Radio(
+            choices=["Default", "Free"],
+            label="Choose Summary Template",
+            value="Default"
+        )
+    
+    fetch_button = gr.Button("Get Summary")
+    summary_output = gr.Textbox(label="News Summary", lines=20)
+    
+    fetch_button.click(generate_summary, inputs=[interests, template_choice], outputs=summary_output)
 
 if __name__ == "__main__":
     demo.launch()