Update app.py

app.py

@@ -47,30 +47,28 @@ if selected_brand != "Select":
 
     if watch_data:
         # Display the image from the JSON
-        image_url = watch_data.get("image", None)  # Assume the "image" key holds the image URL
+        image_url = watch_data.get("image", None)
         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)
-        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)
-
-        # Generate description based on attributes and image
-        if st.button("Generate Description"):
-            attributes = {
-                "brand": watch_data["brand"],
-                "name": watch_data.get("name", "Unknown Watch"),
-                "sku": watch_data.get("sku", "Unknown SKU"),
-                "features": watch_data.get("features", "Unknown Features"),
-                "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
-            }
-
-            # Simulate a description that mentions the color, look, and other aspects
-            input_text = f"""Generate a detailed description for the following watch:
+        # Generate description based on attributes automatically
+        attributes = {
+            "brand": watch_data["brand"],
+            "name": watch_data.get("name", "Unknown Watch"),
+            "sku": watch_data.get("sku", "Unknown SKU"),
+            "features": watch_data.get("features", "Unknown Features"),
+            "casesize": watch_data.get("casesize", "Unknown Case Size"),
+            "movement": watch_data.get("movement", "Unknown Movement"),
+            "gender": watch_data.get("gender", "Unknown Gender"),
+            "water_resistance": watch_data.get("water_resistance", "Unknown Water Resistance"),
+            "power_reserve": watch_data.get("power_reserve", "Unknown Power Reserve"),
+            "dial_color": watch_data.get("dial_color", "Unknown Dial Color"),
+            "strap_material": watch_data.get("strap_material", "Unknown Strap Material"),
+            "price": watch_data.get("price", "Unknown Price"),
+        }
+
+        # Expanded input text with more attributes, using the model as a guide for the output
+        input_text = f"""Generate a detailed 100-word description for the following watch:
 Brand: {attributes['brand']}
 Name: {attributes['name']}
 SKU: {attributes['sku']}
@@ -78,38 +76,33 @@ 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.
+Water Resistance: {attributes['water_resistance']}
+Power Reserve: {attributes['power_reserve']}
+Dial Color: {attributes['dial_color']}
+Strap Material: {attributes['strap_material']}
+Price: {attributes['price']}
 
 Description:"""
 
-            # Tokenize input and generate description
-            inputs = tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True)
-            outputs = model.generate(
-                **inputs,
-                max_length=max_length,
-                num_return_sequences=1,
-                temperature=temperature,
-                top_k=50,
-                top_p=0.95,
-                do_sample=True,
-                repetition_penalty=repetition_penalty  # Prevent repetitive descriptions
-            )
-
-            # Decode generated text
-            description = tokenizer.decode(outputs[0], skip_special_tokens=True)
-
-            # Display the result
-            st.write("### Generated Description")
-            st.write(description)
-
-            # Display watch details
-            st.write("### Watch Details")
-            st.json(json.dumps(watch_data, indent=2))
-
-            # Display input text for debugging
-            st.write("### Input Text (for debugging)")
-            st.text(input_text)
+        # Tokenize input and generate description
+        inputs = tokenizer(input_text, return_tensors="pt", max_length=512, truncation=True)
+        outputs = model.generate(
+            **inputs,
+            max_length=150,  # Limit output to 100-150 words
+            num_return_sequences=1,
+            temperature=0.7,
+            top_k=50,
+            top_p=0.95,
+            do_sample=True,
+            repetition_penalty=1.2  # Prevent repetition in descriptions
+        )
+
+        # Decode generated text
+        description = tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+        # Display the result
+        st.write("### Generated Description")
+        st.write(description)
 else:
     st.warning("Please select a brand.")