ALwrity Version 0.5.1 (Fastapi + React)

ToBeMigrated/ai_seo_tools/opengraph_generator.py

@@ -0,0 +1,129 @@
+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.")