Update app.py

app.py

@@ -46,12 +46,17 @@ if selected_brand != "Select":
         watch_data = next((item for item in data if item["sku"] == selected_sku), None)
 
     if watch_data:
+        # Display the image from the JSON
+        image_url = watch_data.get("image", None)  # Assume the "image" key holds the image URL
+        if image_url:
+            st.image(image_url, caption=f"{watch_data['name']} Image")
+
         # Generation parameters
-        max_length = st.slider("Max Length", min_value=100, max_value=500, value=300)  # Increase max length to allow for more text
+        max_length = st.slider("Max Length", min_value=100, max_value=500, value=300)
         temperature = st.slider("Temperature", min_value=0.1, max_value=1.0, value=0.7, step=0.1)
-        repetition_penalty = st.slider("Repetition Penalty", min_value=1.0, max_value=2.0, value=1.2, step=0.1)  # Penalize repetitive text
+        repetition_penalty = st.slider("Repetition Penalty", min_value=1.0, max_value=2.0, value=1.2, step=0.1)
 
-        # Generate description based on attributes
+        # Generate description based on attributes and image
         if st.button("Generate Description"):
             attributes = {
                 "brand": watch_data["brand"],
@@ -61,9 +66,10 @@ if selected_brand != "Select":
                 "casesize": watch_data.get("casesize", "Unknown Case Size"),
                 "movement": watch_data.get("movement", "Unknown Movement"),
                 "gender": watch_data.get("gender", "Unknown Gender"),
+                "color": watch_data.get("color", "Unknown Color"),  # Simulated color field
             }
 
-            # Format input similar to training data (adjust this based on your training data format)
+            # Simulate a description that mentions the color, look, and other aspects
             input_text = f"""Generate a detailed description for the following watch:
 Brand: {attributes['brand']}
 Name: {attributes['name']}
@@ -72,6 +78,8 @@ Features: {attributes['features']}
 Case Size: {attributes['casesize']}
 Movement: {attributes['movement']}
 Gender: {attributes['gender']}
+Color: {attributes['color']}
+Look: Based on its stunning design, as seen in the image, this watch has a {attributes['color']} finish that complements its {attributes['casesize']}mm case size. It exudes elegance and sophistication, making it perfect for {attributes['gender']} audiences. The craftsmanship seen in the movement adds to its luxurious appeal.
 
 Description:"""