Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
-from flask import Flask, render_template, request
-from rss_processor import fetch_rss_feeds, process_and_store_articles, vector_db
+from flask import Flask, render_template, request, jsonify
+from rss_processor import fetch_rss_feeds, process_and_store_articles, vector_db, summarize_article
 import logging
 
 app = Flask(__name__)
@@ -14,24 +14,24 @@ def index():
     logger.info("Starting to fetch RSS feeds")
     articles = fetch_rss_feeds()
     logger.info(f"Fetched {len(articles)} articles")
-    process_and_store_articles(articles)
+    process_and_store_articles(articles, summarize=False)  # Skip summarization initially
     logger.info("Articles processed and stored")
     stored_docs = vector_db.similarity_search("news", k=len(articles))
     enriched_articles = [
         {
             "title": doc.metadata["title"],
             "link": doc.metadata["link"],
-            "summary": doc.page_content,
+            "summary": doc.page_content or doc.metadata["original_description"],  # Use description if no summary
             "category": doc.metadata["category"],
             "sentiment": doc.metadata["sentiment"],
             "published": doc.metadata["published"],
-            "image": doc.metadata.get("image", "svg"),  # Use "svg" as a flag for default
+            "image": doc.metadata.get("image", "svg"),
         }
         for doc in stored_docs
     ]
     logger.info(f"Enriched {len(enriched_articles)} articles for display")
 
-    if request.method == 'POST':
+    if request.method == 'POST' and 'search' in request.form:
         query = request.form.get('search')
         if query:
             logger.info(f"Processing search query: {query}")
@@ -40,7 +40,7 @@ def index():
                 {
                     "title": doc.metadata["title"],
                     "link": doc.metadata["link"],
-                    "summary": doc.page_content,
+                    "summary": doc.page_content or doc.metadata["original_description"],
                     "category": doc.metadata["category"],
                     "sentiment": doc.metadata["sentiment"],
                     "published": doc.metadata["published"],
@@ -59,5 +59,14 @@ def index():
 
     return render_template("index.html", categorized_articles=categorized_articles)
 
+@app.route('/summarize', methods=['POST'])
+def summarize():
+    data = request.get_json()
+    text = data.get('text')
+    logger.info(f"Received summarize request for text: {text[:50]}...")
+    summary = summarize_article(text)
+    logger.info(f"Generated summary: {summary}")
+    return jsonify({"summary": summary})
+
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)