aiwonewsletter / app.py
dindizz's picture
Update app.py
82f9412 verified
raw
history blame
2.12 kB
import requests
from bs4 import BeautifulSoup
import openai
import gradio as gr
import os
from dotenv import load_dotenv
# Load environment variables from .env file
load_dotenv()
openai_api_key = os.getenv("OPENAI_API_KEY")
# Function to scrape content from a URL
def scrape_content(url):
response = requests.get(url)
soup = BeautifulSoup(response.content, 'html.parser')
# Example of extracting title and body content - modify based on actual structure of the websites
title = soup.find('title').get_text()
paragraphs = soup.find_all('p')
content = '\n'.join([para.get_text() for para in paragraphs])
return title, content
# Function to create newsletter using OpenAI
def create_newsletter(contents):
openai.api_key = openai_api_key
prompt = "Create a newsletter with the following content:\n\n"
for content in contents:
title, body, url = content
prompt += f"Title: {title}\nURL: {url}\n\n{body}\n\n"
response = openai.ChatCompletion.create(
model="gpt-4",
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": prompt}
],
max_tokens=1500
)
newsletter = response.choices[0].message['content'].strip()
return newsletter
# Function to process URLs and generate the newsletter
def process_urls(url1, url2, url3, url4, url5):
urls = [url for url in [url1, url2, url3, url4, url5] if url]
if not urls:
return "No URLs provided."
contents = []
for url in urls:
title, content = scrape_content(url)
contents.append((title, content, url))
newsletter = create_newsletter(contents)
return newsletter
# Gradio interface
iface = gr.Interface(
fn=process_urls,
inputs=[
gr.Textbox(label="URL 1"),
gr.Textbox(label="URL 2"),
gr.Textbox(label="URL 3"),
gr.Textbox(label="URL 4"),
gr.Textbox(label="URL 5")
],
outputs="html",
title="Newsletter Generator",
description="Enter up to 5 URLs to generate a newsletter."
)
iface.launch()