PrajwalW's picture
Update app.py
68708fb verified
import streamlit as st
import requests
import os
import json
import boto3
from dotenv import load_dotenv
# API Key (Ensure security; do not share publicly)
# API_KEY = "67bd857847938bebf6b0ff82a26ba45f251986f3"
# Define the API endpoint
# API_URL = "https://google.serper.dev/search"
# Load API keys from .env file
load_dotenv()
NEWS_API_KEY = "3d542da6aaa94a79ad7d2b852e944dac"
SERPER_API_KEY = "67bd857847938bebf6b0ff82a26ba45f251986f3"
# Claude 3 via AWS Bedrock
bedrock = boto3.client("bedrock-runtime", region_name=os.getenv("AWS_REGION"))
# Streamlit app UI
st.title("πŸ“° News Summarizer Agent")
topic = st.text_input("Enter a topic, company name, or keyword:")
# Fetch news from NewsAPI
def fetch_newsapi_news(topic):
url = f"https://newsapi.org/v2/everything?q={topic}&pageSize=5&sortBy=publishedAt&apiKey={NEWS_API_KEY}"
response = requests.get(url)
articles = response.json().get("articles", [])
return [{"title": a["title"], "description": a["description"], "url": a["url"]} for a in articles]
# Fetch news from Serper.dev
def fetch_serper_news(topic):
url = "https://google.serper.dev/news"
headers = {"X-API-KEY": SERPER_API_KEY, "Content-Type": "application/json"}
data = {"q": topic}
response = requests.post(url, headers=headers, data=json.dumps(data))
results = response.json().get("news", [])[:5]
return [{"title": r["title"], "description": r.get("snippet", ""), "url": r["link"]} for r in results]
# Summarize with Claude 3
def summarize_with_claude(article):
messages = [
{
"role": "user",
"content": f"""You are a helpful news summarizer agent.
Given the article title and description, generate:
1. A more relevant headline based on the topic
2. A 3-5 line clear and concise summary.
Title: {article['title']}
Description: {article['description']}
Respond as:
Headline: ...
Summary: ...
"""
}
]
body = {
"messages": messages,
"max_tokens": 300,
"temperature": 0.7,
"anthropic_version": "bedrock-2023-05-31" # REQUIRED
}
response = bedrock.invoke_model(
modelId="anthropic.claude-3-sonnet-20240229-v1:0",
body=json.dumps(body),
contentType="application/json",
accept="application/json"
)
result = json.loads(response["body"].read())
return result["content"][0]["text"].strip()
# Button and main logic
if st.button("Get News"):
if not topic:
st.warning("Please enter a topic.")
else:
st.subheader("Fetching top news...")
news_items = fetch_newsapi_news(topic) + fetch_serper_news(topic)
for idx, article in enumerate(news_items[:5]):
st.markdown(f"### πŸ—žοΈ News #{idx+1}")
st.markdown(f"[Original Article]({article['url']})")
with st.spinner("Generating summary..."):
summary = summarize_with_claude(article)
st.markdown(summary)