Spaces:
Sleeping
Sleeping
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) | |