Update app.py

app.py

@@ -68,41 +68,57 @@ def enhance_description_with_gemini(description):
         return "No contents or parts in response"
     except requests.exceptions.RequestException as e:
         return f"Request failed: {e}"
+def create_label(image, ingredients, usage, expiry_date, barcode_data):
+    try:
+        # Create an image for the label
+        label_image = Image.new('RGB', (800, 1200), color='white')
+        draw = ImageDraw.Draw(label_image)
+
+        # Define font and size
+        try:
+            font = ImageFont.truetype("arial.ttf", 24)
+        except IOError:
+            font = ImageFont.load_default()
+        
+        # Draw the main image
+        image.thumbnail((600, 400))
+        label_image.paste(image, (100, 50))
+
+        # Draw text
+        text_position = (100, 500)
+        draw.text(text_position, f"Ingredients:\n{ingredients}", font=font, fill="black")
+        text_position = (100, 600)
+        draw.text(text_position, f"Usage:\n{usage}", font=font, fill="black")
+        text_position = (100, 700)
+        draw.text(text_position, f"Expiry Date: {expiry_date}", font=font, fill="black")
+
+        # Draw barcode
+        if len(barcode_data) == 12:  # Check length for EAN-13
+            barcode_image = generate_barcode(barcode_data)
+            label_image.paste(barcode_image, (100, 800))
+        else:
+            draw.text((100, 800), "Invalid Barcode Data", font=font, fill="red")
+        
+        return label_image
+    except Exception as e:
+        raise RuntimeError(f"Error generating label: {e}")
+
+# Streamlit interface continuation
+if st.button('Generate Label'):
+    if ingredients and usage and expiry_date and barcode_data:
+        try:
+            label_image = create_label(image, ingredients, usage, expiry_date, barcode_data)
+            st.image(label_image, caption="Generated Label")
+        except Exception as e:
+            st.error(f"Error generating label: {e}")
+    else:
+        st.error("Please provide all required details for the label.")
 
 def generate_barcode(data):
     code = barcode.get('ean13', data, writer=ImageWriter())
     barcode_image = code.render()
     return barcode_image
 
-def create_label(image, ingredients, usage, expiry_date, barcode_data):
-    # Create an image for the label
-    label_image = Image.new('RGB', (800, 1200), color='white')
-    draw = ImageDraw.Draw(label_image)
-
-    # Define font and size
-    try:
-        font = ImageFont.truetype("arial.ttf", 24)
-    except IOError:
-        font = ImageFont.load_default()
-    
-    # Draw the main image
-    image.thumbnail((600, 400))
-    label_image.paste(image, (100, 50))
-
-    # Draw text
-    text_position = (100, 500)
-    draw.text(text_position, f"Ingredients:\n{ingredients}", font=font, fill="black")
-    text_position = (100, 600)
-    draw.text(text_position, f"Usage:\n{usage}", font=font, fill="black")
-    text_position = (100, 700)
-    draw.text(text_position, f"Expiry Date: {expiry_date}", font=font, fill="black")
-
-    # Draw barcode
-    barcode_image = generate_barcode(barcode_data)
-    label_image.paste(barcode_image, (100, 800))
-
-    return label_image
-
 # Streamlit interface
 st.title("Image Detailed Description, Captions, Logo Suggestions Generator with Image Enhancement Options")