Update app.py

app.py

@@ -2,15 +2,17 @@ import streamlit as st
 import json
 import requests
 from newspaper import Article
+from newsapi import NewsApiClient
 
 st.set_page_config(page_title='Short News App',
-layout = 'wide',
-initial_sidebar_state = 'expanded',
-menu_items={
-    'About':'This is a demo application with One AI',
-    'Get help':'https://studio.oneai.com/docs',
-    'Report a Bug':'https://discord.com/channels/941458663493746698/941458828187287603'
-})
+    layout = 'wide',
+    initial_sidebar_state = 'expanded',
+    menu_items={
+        'About':'This is a demo application with One AI',
+        'Get help':'https://studio.oneai.com/docs',
+        'Report a Bug':'https://discord.com/channels/941458663493746698/941458828187287603'
+    })
+
 st.title('Short Summary News Application Demo with OneAI')
 st.markdown('This application takes an input from the user and displays upto five latest news articles along with their summary. This application uses the free quota api calls.')
 st.sidebar.image('logo.jpg')
@@ -29,47 +31,89 @@ st.sidebar.markdown('[Contact](https://www.oneai.com/contact-us)')
 st.sidebar.markdown('[Community](https://discord.com/channels/941458663493746698/942326235722309642)')
 st.sidebar.markdown('© 2022 Logo rights reserved to One AI')
 
-
-
 def run():
+    # Initialize NewsAPI client
+    newsapi = NewsApiClient(api_key='e07356679fcb40e98d44a37b323e9dd6')
     
     @st.cache()
     def summary(text1):
-        api_key = "1c93487c-695c-4089-adfc-5e4b7623718c"
-        url = "https://api.oneai.com/api/v0/pipeline"
-        text = text1
-        headers = {'api-key':api_key, 'content-type':'application/json'}
-        payload = {'input':text, 'input_type':'article', 'steps':[{'skill':'summarize'}]}
-        r = requests.post(url, json=payload, headers=headers)
-        data = r.json()
-        return data['output'][0]['text']
+        try:
+            api_key = "1c93487c-695c-4089-adfc-5e4b7623718c"
+            url = "https://api.oneai.com/api/v0/pipeline"
+            headers = {'api-key': api_key, 'content-type': 'application/json'}
+            payload = {
+                'input': text1,
+                'input_type': 'article',
+                'steps': [{'skill': 'summarize'}]
+            }
+            r = requests.post(url, json=payload, headers=headers)
+            r.raise_for_status()
+            data = r.json()
+            return data['output'][0]['text']
+        except requests.exceptions.RequestException as e:
+            st.error(f"Error calling One AI API: {str(e)}")
+            return None
+        except (KeyError, IndexError) as e:
+            st.error(f"Error processing One AI response: {str(e)}")
+            return None
+
+    def get_links(query):
+        try:
+            all_articles = newsapi.get_everything(q=query,
+                                                language='en',
+                                                sort_by='relevancy',
+                                                page_size=5,
+                                                page=1)
+            
+            if all_articles['status'] != 'ok':
+                st.error(f"NewsAPI Error: {all_articles.get('message', 'Unknown error')}")
+                return []
+            
+            links = [article['url'] for article in all_articles['articles']]
+            return links
+            
+        except Exception as e:
+            st.error(f"Error fetching news: {str(e)}")
+            return []
 
-    def get_links(text2):
-        url = "https://free-news.p.rapidapi.com/v1/search"
-        querystring = {"q":text2,"lang":"en", "page":1, "page_size":5}
-        headers = {'x-rapidapi-host': "free-news.p.rapidapi.com",'x-rapidapi-key': "375ffbaab0mshb442ffb69d6f025p117ba0jsn01e8146148e3"}
-        response = requests.request("GET", url, headers=headers, params=querystring)
-        response_dict = json.loads(response.text)
-        links = [response_dict['articles'][i]['link'] for i in range(len(response_dict['articles']))]
-        return links
-    
     input_text = st.text_input('Search your favorite topic:')
     submitted = st.button('Submit')
     
-    if submitted:
-        links = get_links(input_text)
-        for link in links:
-            try:
-                news_article = Article(link, language='en')
-                news_article.download()
-                news_article.parse()
-                st.image(news_article.top_image)
-                st.header(news_article.title)
-                st.markdown('*Summary of the Article:*')
-                st.markdown(summary(news_article.text))
-                with st.expander('Full Article'):
-                    st.markdown(news_article.text)
-            except:
-                print('No Results!! Please try with new search!!')
+    if submitted and input_text:
+        with st.spinner('Fetching news articles...'):
+            links = get_links(input_text)
+            
+            if not links:
+                st.warning("No articles found. Please try a different search term.")
+                return
+                
+            for link in links:
+                try:
+                    with st.spinner(f'Processing article from {link}...'):
+                        news_article = Article(link, language='en')
+                        news_article.download()
+                        news_article.parse()
+                        
+                        if news_article.top_image:
+                            st.image(news_article.top_image)
+                        
+                        st.header(news_article.title)
+                        st.markdown('*Summary of the Article:*')
+                        
+                        article_summary = summary(news_article.text)
+                        if article_summary:
+                            st.markdown(article_summary)
+                        else:
+                            st.warning("Could not generate summary for this article")
+                            
+                        with st.expander('Full Article'):
+                            st.markdown(news_article.text)
+                            
+                except Exception as e:
+                    st.error(f"Error processing article from {link}: {str(e)}")
+                    continue
+    elif submitted:
+        st.warning("Please enter a search term")
+
 if __name__ == '__main__':
     run()