Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import requests
+from bs4 import BeautifulSoup
 import json
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
@@ -25,6 +26,21 @@ data = load_json_from_url(json_url)
 # Extract unique brands
 brands = sorted(list(set([item["brand"] for item in data])))
 
+# Web Scraping function to get product description from Ethos website
+def scrape_ethos_description(product_link):
+    try:
+        response = requests.get(product_link)
+        soup = BeautifulSoup(response.text, 'html.parser')
+
+        # Assuming product description is within a specific class or ID, find it
+        description = soup.find('div', class_='product-description')  # Example class, adjust based on actual site structure
+        if description:
+            return description.get_text(strip=True)
+        else:
+            return "No detailed description available on Ethos site."
+    except Exception as e:
+        return f"Error fetching details from Ethos site: {str(e)}"
+
 # Streamlit UI
 st.title("Watch Description Generator")
 
@@ -51,7 +67,19 @@ if selected_brand != "Select":
         if image_url:
             st.image(image_url, caption=f"{watch_data['name']} Image")
 
-        # Attributes without price
+        # Get the Ethos product link for web scraping (assuming 'url' key exists in your JSON)
+        product_link = watch_data.get("url", None)
+        if product_link:
+            st.write(f"Fetching details from: [Product Page]({product_link})")
+
+            # Scrape Ethos product page for more description
+            ethos_description = scrape_ethos_description(product_link)
+            st.write("### Ethos Product Description")
+            st.write(ethos_description)
+        else:
+            st.warning("No Ethos link available for this SKU.")
+
+        # Generate watch description based on attributes and scraped content
         attributes = {
             "brand": watch_data["brand"],
             "name": watch_data.get("name", "Unknown Watch"),
@@ -66,7 +94,7 @@ if selected_brand != "Select":
             "strap_material": watch_data.get("strap_material", "Unknown Strap Material")
         }
 
-        # Create a detailed description prompt following your preferred style
+        # Create a detailed description prompt combining scraped content and attributes
         input_text = f"""Generate a detailed 100-word description for the following watch:
 Brand: {attributes['brand']}
 Name: {attributes['name']}
@@ -80,6 +108,9 @@ Power Reserve: {attributes['power_reserve']}
 Dial Color: {attributes['dial_color']}
 Strap Material: {attributes['strap_material']}
 
+Additional details from Ethos:
+{ethos_description}
+
 Description: Provide a luxurious, detailed description focusing on the craftsmanship, innovation, and design, similar to a high-end editorial style."""
 
         # Tokenize input and generate description
@@ -109,7 +140,7 @@ st.sidebar.title("About")
 st.sidebar.info(
     "This app uses a fine-tuned AI model to generate descriptions for watches. "
     "Select a brand and a watch to get started. The model will generate a unique "
-    "description based on the watch's attributes."
+    "description based on the watch's attributes and additional details."
 )
 
 # Add a footer