Update app.py

app.py

@@ -56,24 +56,50 @@ def get_or_create_user_id(user_id):
     if not user_id:
         return str(uuid.uuid4())
     return user_id
-
-# Function to encode the image
+    
+# Modify the encode_image function to handle AVIF files
 def encode_image(image_array):
-    # Convert numpy array to an image file and encode it in base64
-    img = Image.fromarray(np.uint8(image_array))
-    img_buffer = os.path.join(
-        "/tmp", f"temp_image_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
-    )
-    img.save(img_buffer, format="JPEG")
-
-    with open(img_buffer, "rb") as image_file:
-        return base64.b64encode(image_file.read()).decode("utf-8")
+    try:
+        # Convert numpy array to PIL Image
+        img = Image.fromarray(np.uint8(image_array))
+        
+        # Create a temporary file path
+        img_buffer = os.path.join(
+            "/tmp", f"temp_image_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
+        )
+        
+        # Convert to RGB mode if the image is in a different mode
+        if img.mode != 'RGB':
+            img = img.convert('RGB')
+        
+        # Save as JPEG with error handling
+        try:
+            img.save(img_buffer, format="JPEG", quality=95)
+        except Exception as e:
+            print(f"Error saving image: {e}")
+            # If saving fails, try to convert to RGB first
+            img = img.convert('RGB')
+            img.save(img_buffer, format="JPEG", quality=95)
 
+        # Read and encode the saved image
+        with open(img_buffer, "rb") as image_file:
+            encoded_image = base64.b64encode(image_file.read()).decode("utf-8")
+        
+        # Clean up temporary file
+        if os.path.exists(img_buffer):
+            os.remove(img_buffer)
+            
+        return encoded_image
+    except Exception as e:
+        raise ValueError(f"Error processing image: {str(e)}")
 
 # Function to generate product description using OpenAI API
 def generate_product_description(image, description_type, custom_instruction=None):
-    # Encode the uploaded image
-    base64_image = encode_image(image)
+    try:
+        # Encode the uploaded image
+        base64_image = encode_image(image)
+    except Exception as e:
+        raise gr.Error(f"Error processing image: {str(e)}. Please try uploading a different image format.")
 
     headers = {"Content-Type": "application/json", "Authorization": f"Bearer {api_key}"}
 
@@ -234,80 +260,100 @@ with gr.Blocks(css=custom_css) as demo:
             outputs=[custom_instruction],
         )
 
-        # Modify the handle_submit function
         def handle_submit(image, description_type, custom_instruction, user_id):
-            if not user_id:
-                user_id = str(uuid.uuid4())
+            try:
+                if not user_id:
+                    user_id = str(uuid.uuid4())
+                
+                check_and_reset_user_counter(user_id)
+                test_counter = increment_user_counter(user_id)
+                
+                demo_link = "https://wordlift.io/book-a-demo/?utm_source=free-app&utm_medium=app&utm_campaign=product-description-generator"
+                
+                if test_counter <= 4:
+                    try:
+                        result = generate_product_description(image, description_type, custom_instruction)
+                    except gr.Error as e:
+                        return (
+                            str(e),
+                            gr.update(visible=False),
+                            gr.update(interactive=True),
+                            f"Error occurred. You still have {4 - test_counter + 1} free test(s) remaining today.",
+                            f"User ID: {user_id}",
+                            user_id
+                        )
+                    
+                    remaining_tests = 4 - test_counter
+        
+                    if remaining_tests > 0:
+                        return (
+                            result,
+                            gr.update(visible=False),
+                            gr.update(interactive=True),
+                            f"You have {remaining_tests} free test(s) remaining today.",
+                            f"User ID: {user_id}",
+                            user_id
+                        )
+                    else:
+                        redirect_text = f"""
+                        Thank you for trying our product description generation tool!
             
-            check_and_reset_user_counter(user_id)  # Check and reset counter if it's a new day
-            test_counter = increment_user_counter(user_id)
+                        You've used all 4 of your free tests for today. We hope you found them helpful and insightful.
             
-            # UTM-tagged demo booking link
-            demo_link = "https://wordlift.io/book-a-demo/?utm_source=free-app&utm_medium=app&utm_campaign=product-description-generator"
+                        To unlock unlimited access and discover how our AI-powered solution can revolutionize your content creation process:
             
-            if test_counter <= 4:
-                result = generate_product_description(image, description_type, custom_instruction)
-                remaining_tests = 4 - test_counter
-                if remaining_tests > 0:
-                    return (
-                        result,
-                        gr.update(visible=False),
-                        gr.update(interactive=True),
-                        f"You have {remaining_tests} free test(s) remaining today.",
-                        f"User ID: {user_id}",
-                        user_id
-                    )
+                        [Book a Personalized Demo with Our Experts]({demo_link})
+            
+                        During the demo, you'll get:<br>
+                        • A tailored walkthrough of our full suite of features<br>
+                        • Insights on how to optimize your specific content workflow<br>
+                        • Answers to any questions you may have about our solution<br>
+            
+                        We're excited to show you how we can help scale your content creation!
+            
+                        (Your free tests will reset tomorrow, allowing you to try 4 more if you need additional time to decide.)
+                        """
+                        return (
+                            result,
+                            gr.update(visible=True, value=redirect_text),
+                            gr.update(interactive=False),
+                            "You've used all your free tests for today. We invite you to book a demo for full access!",
+                            f"User ID: {user_id}",
+                            user_id
+                        )
                 else:
                     redirect_text = f"""
-                    Thank you for trying our product description generation tool!
-        
-                    You've used all 4 of your free tests for today. We hope you found them helpful and insightful.
-        
-                    To unlock unlimited access and discover how our AI-powered solution can revolutionize your content creation process:
-        
+                    Thank you for your interest in our product description generation tool!
+            
+                    You've already used your 4 free tests for today. We hope they've given you a glimpse of how our AI can enhance your content creation.
+            
+                    Ready to explore the full potential of our solution?
+            
                     [Book a Personalized Demo with Our Experts]({demo_link})
-        
-                    During the demo, you'll get:<br>
-                    • A tailored walkthrough of our full suite of features<br>
-                    • Insights on how to optimize your specific content workflow<br>
-                    • Answers to any questions you may have about our solution<br>
-        
-                    We're excited to show you how we can help scale your content creation!
-        
-                    (Your free tests will reset tomorrow, allowing you to try 4 more if you need additional time to decide.)
+            
+                    In this demo, you'll discover:<br>
+                    • How our AI can be customized for your specific needs<br>
+                    • Advanced features for scaling your content production using a [Product Knowledge Graph](https://wordlift.io/blog/en/product-knowledge-graph/)<br>
+                    • ROI projections based on your current content workflow<br>
+            
+                    Let's discuss how we can help you achieve your content goals!
+            
+                    (Remember, your free tests will reset tomorrow if you need more time to evaluate.)
                     """
                     return (
-                        result,
+                        "",
                         gr.update(visible=True, value=redirect_text),
                         gr.update(interactive=False),
-                        "You've used all your free tests for today. We invite you to book a demo for full access!",
+                        "All free tests used. We invite you to book a demo for full access!",
                         f"User ID: {user_id}",
                         user_id
                     )
-            else:
-                redirect_text = f"""
-                Thank you for your interest in our product description generation tool!
-        
-                You've already used your 4 free tests for today. We hope they've given you a glimpse of how our AI can enhance your content creation.
-        
-                Ready to explore the full potential of our solution?
-        
-                [Book a Personalized Demo with Our Experts]({demo_link})
-        
-                In this demo, you'll discover:<br>
-                • How our AI can be customized for your specific needs<br>
-                • Advanced features for scaling your content production using a [Product Knowledge Graph](https://wordlift.io/blog/en/product-knowledge-graph/)<br>
-                • ROI projections based on your current content workflow<br>
-        
-                Let's discuss how we can help you achieve your content goals!
-        
-                (Remember, your free tests will reset tomorrow if you need more time to evaluate.)
-                """
+            except Exception as e:
                 return (
-                    "",
-                    gr.update(visible=True, value=redirect_text),
-                    gr.update(interactive=False),
-                    "All free tests used. We invite you to book a demo for full access!",
+                    f"An error occurred: {str(e)}",
+                    gr.update(visible=False),
+                    gr.update(interactive=True),
+                    "Error processing request",
                     f"User ID: {user_id}",
                     user_id
                 )