Update app.py

app.py

@@ -110,7 +110,18 @@ css = """
     height: 500px;
     overflow: auto;
     border: 1px solid #ccc;
-  }"""
+  }
+  .redirect-message {
+    background-color: #f0f0f0;
+    border: 1px solid #ccc;
+    padding: 15px;
+    margin-top: 20px;
+    border-radius: 5px;
+  }
+"""
+
+# Initialize the test counter
+test_counter = 0
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown("<h1>WordLift Product Description Generation - [FREE]</h1>")
@@ -169,12 +180,51 @@ with gr.Blocks(css=css) as demo:
             outputs=[custom_instruction],
         )
 
+        # Modified function to handle test limit
+        def handle_submit(image, description_type, custom_instruction):
+            global test_counter
+            if test_counter < 4:
+                test_counter += 1
+                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} test(s) remaining.",
+                    )
+                else:
+                    redirect_text = """
+                    You've reached the limit of 4 free tests. To continue using our product description generation tool, 
+                    please book a demo of our solution for content generation with our experts.
+                    
+                    [Book a Demo](https://wordlift.io/book-a-demo/)
+                    """
+                    return (
+                        result,
+                        gr.update(visible=True, value=redirect_text),
+                        gr.update(interactive=False),
+                        "You've used all your free tests.",
+                    )
+            else:
+                return (
+                    "",
+                    gr.update(visible=True),
+                    gr.update(interactive=False),
+                    "You've used all your free tests.",
+                )
+
         submit_btn.click(
-            generate_product_description,
-            [input_img, description_type, custom_instruction],
-            [output_text],
+            handle_submit,
+            inputs=[input_img, description_type, custom_instruction],
+            outputs=[output_text, redirect_message, submit_btn, gr.Markdown(visible=True, label="Test Counter")],
         )
 
+# Launch Gradio app
+demo.queue(api_open=False)
+demo.launch(debug=True)
+
 # Launch Gradio app
 demo.queue(api_open=False)
 demo.launch(debug=True)