Update app.py

app.py

@@ -5,11 +5,44 @@ from PIL import Image
 import numpy as np
 from datetime import datetime
 import os
+import json
+
+# File to store session data
+SESSION_FILE = "session_data.json"
 
 # OpenAI API Key
 api_key = os.getenv("OPENAI_API_KEY")
 
 
+# Function to load session data
+def load_session_data():
+    if os.path.exists(SESSION_FILE):
+        with open(SESSION_FILE, 'r') as f:
+            return json.load(f)
+    return {"last_reset_date": str(date.today()), "test_counter": 0}
+
+# Function to save session data
+def save_session_data(data):
+    with open(SESSION_FILE, 'w') as f:
+        json.dump(data, f)
+
+# Function to check and reset the counter if necessary
+def check_and_reset_counter():
+    session_data = load_session_data()
+    today = str(date.today())
+    if session_data["last_reset_date"] != today:
+        session_data["last_reset_date"] = today
+        session_data["test_counter"] = 0
+        save_session_data(session_data)
+    return session_data["test_counter"]
+
+# Function to increment the counter
+def increment_counter():
+    session_data = load_session_data()
+    session_data["test_counter"] += 1
+    save_session_data(session_data)
+    return session_data["test_counter"]
+
 # Function to encode the image
 def encode_image(image_array):
     # Convert numpy array to an image file and encode it in base64
@@ -168,7 +201,6 @@ with gr.Blocks(css=css) as demo:
                     label="Custom Instruction (Only for 'Other')", visible=False, elem_classes="gr-box"
                 )
                 submit_btn = gr.Button(value="Submit", elem_classes="gr-box")
-
             with gr.Column():
                 output_text = gr.Markdown(label="Output Text", show_copy_button=True, elem_classes="output-text")
                 redirect_message = gr.Markdown(visible=False, elem_classes="redirect-message")
@@ -186,9 +218,9 @@ with gr.Blocks(css=css) as demo:
 
         # Modified function to handle test limit
         def handle_submit(image, description_type, custom_instruction):
-            global test_counter
-            if test_counter < 4:
-                test_counter += 1
+            check_and_reset_counter()  # Check and reset counter if it's a new day
+            test_counter = increment_counter()
+            if test_counter <= 4:
                 result = generate_product_description(image, description_type, custom_instruction)
                 remaining_tests = 4 - test_counter
                 if remaining_tests > 0:
@@ -196,27 +228,29 @@ with gr.Blocks(css=css) as demo:
                         result,
                         gr.update(visible=False),
                         gr.update(interactive=True),
-                        f"You have {remaining_tests} test(s) remaining.",
+                        f"You have {remaining_tests} test(s) remaining today.",
                     )
                 else:
                     redirect_text = """
-                    You've reached the limit of 4 free tests. To continue using our product description generation tool, 
+                    You've reached the limit of 4 free tests for today. 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/)
+                    
+                    The test counter will reset tomorrow, allowing you to try 4 more tests.
                     """
                     return (
                         result,
                         gr.update(visible=True, value=redirect_text),
                         gr.update(interactive=False),
-                        "You've used all your free tests.",
+                        "You've used all your free tests for today.",
                     )
             else:
                 return (
                     "",
                     gr.update(visible=True),
                     gr.update(interactive=False),
-                    "You've used all your free tests.",
+                    "You've used all your free tests for today.",
                 )
 
         submit_btn.click(
@@ -225,7 +259,6 @@ with gr.Blocks(css=css) as demo:
             outputs=[output_text, redirect_message, submit_btn, test_counter_display],
         )
 
-
 # Launch Gradio app
 demo.queue(api_open=False)
 demo.launch(debug=True)