088975a70f
Here are some improvements to the og_tag_generator function in opengraph_generator.py: Refactor Code for Better Readability: Group related operations together and add comments for better understanding. Use helper functions to break down larger pieces of functionality into smaller, reusable code blocks. Error Handling: Improve error handling by providing more specific error messages and handling different types of errors separately. User Experience Enhancements: Add informative messages and examples to guide the user through the input process. Provide feedback on the success or failure of the Open Graph tag generation.
130 lines
4.7 KiB
Python
130 lines
4.7 KiB
Python
import streamlit as st
|
|
import requests
|
|
from bs4 import BeautifulSoup
|
|
from ..gpt_providers.text_generation.main_text_generation import llm_text_gen
|
|
|
|
|
|
def generate_og_tags(url, title_hint, description_hint, platform="General"):
|
|
"""
|
|
Generate Open Graph tags based on the provided URL, title hint, description hint, and platform.
|
|
|
|
Args:
|
|
url (str): The URL of the webpage.
|
|
title_hint (str): A hint for the title.
|
|
description_hint (str): A hint for the description.
|
|
platform (str): The platform for which to generate the tags (General, Facebook, or Twitter).
|
|
|
|
Returns:
|
|
str: The generated Open Graph tags or an error message.
|
|
"""
|
|
# Create a prompt for the text generation model
|
|
prompt = (
|
|
f"Generate Open Graph tags for the following page:\nURL: {url}\n"
|
|
f"Title hint: {title_hint}\nDescription hint: {description_hint}"
|
|
)
|
|
if platform == "Facebook":
|
|
prompt += "\nSpecifically for Facebook"
|
|
elif platform == "Twitter":
|
|
prompt += "\nSpecifically for Twitter"
|
|
|
|
try:
|
|
# Generate Open Graph tags using the text generation model
|
|
response = llm_text_gen(prompt)
|
|
return response
|
|
except Exception as err:
|
|
st.error(f"Failed to generate Open Graph tags: {err}")
|
|
return None
|
|
|
|
|
|
def extract_default_og_tags(url):
|
|
"""
|
|
Extract default Open Graph tags from the provided URL.
|
|
|
|
Args:
|
|
url (str): The URL of the webpage.
|
|
|
|
Returns:
|
|
tuple: A tuple containing the title, description, and image URL, or None in case of an error.
|
|
"""
|
|
try:
|
|
# Fetch the HTML content of the URL
|
|
response = requests.get(url)
|
|
response.raise_for_status()
|
|
|
|
# Parse the HTML content using BeautifulSoup
|
|
soup = BeautifulSoup(response.content, 'html.parser')
|
|
|
|
# Extract the title, description, and image URL
|
|
title = soup.find('title').text if soup.find('title') else None
|
|
description = soup.find('meta', attrs={'name': 'description'})['content'] if soup.find('meta', attrs={'name': 'description'}) else None
|
|
image_url = soup.find('meta', attrs={'property': 'og:image'})['content'] if soup.find('meta', attrs={'property': 'og:image'}) else None
|
|
|
|
return title, description, image_url
|
|
|
|
except requests.exceptions.RequestException as req_err:
|
|
st.error(f"Error fetching the URL: {req_err}")
|
|
return None, None, None
|
|
|
|
except Exception as err:
|
|
st.error(f"Error parsing the HTML content: {err}")
|
|
return None, None, None
|
|
|
|
|
|
def og_tag_generator():
|
|
"""Main function to run the Streamlit app."""
|
|
st.title("AI Open Graph Tag Generator")
|
|
|
|
# Platform selection
|
|
platform = st.selectbox(
|
|
"**Select the platform**",
|
|
["General", "Facebook", "Twitter"],
|
|
help="Choose the platform for which you want to generate Open Graph tags."
|
|
)
|
|
|
|
# URL input
|
|
url = st.text_input(
|
|
"**Enter the URL of the page to generate Open Graph tags for:**",
|
|
placeholder="e.g., https://example.com",
|
|
help="Provide the URL of the page you want to generate Open Graph tags for."
|
|
)
|
|
|
|
if url:
|
|
# Extract default Open Graph tags
|
|
title, description, image_url = extract_default_og_tags(url)
|
|
|
|
# Title hint input
|
|
title_hint = st.text_input(
|
|
"**Modify existing title or suggest a new one (optional):**",
|
|
value=title if title else "",
|
|
placeholder="e.g., Amazing Blog Post Title"
|
|
)
|
|
|
|
# Description hint input
|
|
description_hint = st.text_area(
|
|
"**Modify existing description or suggest a new one (optional):**",
|
|
value=description if description else "",
|
|
placeholder="e.g., This is a detailed description of the content."
|
|
)
|
|
|
|
# Image URL hint input
|
|
image_hint = st.text_input(
|
|
"**Use this image or suggest a new URL (optional):**",
|
|
value=image_url if image_url else "",
|
|
placeholder="e.g., https://example.com/image.jpg"
|
|
)
|
|
|
|
# Generate Open Graph tags
|
|
if st.button("Generate Open Graph Tags"):
|
|
with st.spinner("Generating Open Graph tags..."):
|
|
try:
|
|
og_tags = generate_og_tags(url, title_hint, description_hint, platform)
|
|
if og_tags:
|
|
st.success("Open Graph tags generated successfully!")
|
|
st.markdown(og_tags)
|
|
else:
|
|
st.error("Failed to generate Open Graph tags.")
|
|
except Exception as e:
|
|
st.error(f"Failed to generate Open Graph tags: {e}")
|
|
else:
|
|
st.info("Please enter a URL to generate Open Graph tags.")
|