Update app.py

app.py

@@ -3,17 +3,29 @@ from transformers import pipeline
 from bs4 import BeautifulSoup
 import requests
 from googlesearch import search
+import logging
 
-def summarize_news(query):
-    # Initialize summarization pipeline
-    summarizer = pipeline("summarization")
+# Configure logging
+logging.basicConfig(level=logging.DEBUG)
+
+def summarize_news(query, num_results=3):
+    logging.debug(f"Query received: {query}")
+    logging.debug(f"Number of results requested: {num_results}")
+
+    # Initialize summarization pipeline with a specific model
+    logging.debug("Initializing summarization pipeline...")
+    summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
 
     # Search for news articles
-    search_results = search(query, num_results=3)
+    logging.debug("Searching for news articles...")
+    search_results = search(query, num_results=num_results)
     articles = []
     
+    logging.debug(f"Search results: {search_results}")
+
     for url in search_results:
         try:
+            logging.debug(f"Fetching content from URL: {url}")
             # Fetch the content of the news article
             r = requests.get(url)
             soup = BeautifulSoup(r.text, 'html.parser')
@@ -22,6 +34,7 @@ def summarize_news(query):
             ARTICLE = ' '.join(text)
 
             # Chunk the article text
+            logging.debug("Chunking the article text...")
             max_chunk = 500
             ARTICLE = ARTICLE.replace('.', '.<eos>')
             ARTICLE = ARTICLE.replace('?', '?<eos>')
@@ -43,13 +56,20 @@ def summarize_news(query):
             for chunk_id in range(len(chunks)):
                 chunks[chunk_id] = ' '.join(chunks[chunk_id])
 
+            logging.debug(f"Chunks created: {chunks}")
+
             # Summarize the chunks
+            logging.debug("Summarizing the chunks...")
             summaries = summarizer(chunks, max_length=120, min_length=30, do_sample=False)
             summary_text = " ".join([summary['summary_text'] for summary in summaries])
             articles.append((url, summary_text))
+
+            logging.debug(f"Summary for URL {url}: {summary_text}")
         except Exception as e:
+            logging.error(f"Error processing URL {url}: {e}")
             continue
 
+    logging.debug(f"Final summarized articles: {articles}")
     return articles
 
 def format_output(articles):
@@ -59,11 +79,13 @@ def format_output(articles):
     return formatted_text
 
 iface = gr.Interface(
-    fn=lambda query: format_output(summarize_news(query)), 
-    inputs="text", 
+    fn=lambda query, num_results: format_output(summarize_news(query, num_results)), 
+    inputs=[gr.inputs.Textbox(label="Query"), gr.inputs.Slider(minimum=1, maximum=10, default=3, label="Number of Results")], 
     outputs="text",
     title="News Summarizer",
     description="Enter a query to get summarized versions of the top news articles."
 )
 
-iface.launch()
+if __name__ == "__main__":
+    logging.debug("Launching Gradio interface...")
+    iface.launch()