Update app.py

app.py

@@ -6,42 +6,50 @@ import numpy as np
 from datetime import datetime, date
 import os
 import json
+import uuid
 
 # File to store session data
-SESSION_FILE = "session_data.json"
+SESSION_FILE = "user_session_data.json"
 
 # OpenAI API Key
 api_key = os.getenv("OPENAI_API_KEY")
 
 
-# Function to load session data
-def load_session_data():
+# Function to load user session data
+def load_user_session_data(user_id):
     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):
+            all_user_data = json.load(f)
+    else:
+        all_user_data = {}
+    
+    if user_id not in all_user_data:
+        all_user_data[user_id] = {"last_reset_date": str(date.today()), "test_counter": 0}
+    
+    return all_user_data
+
+# Function to save user session data
+def save_user_session_data(all_user_data):
     with open(SESSION_FILE, 'w') as f:
-        json.dump(data, f)
+        json.dump(all_user_data, f)
 
-# Function to check and reset the counter if necessary
-def check_and_reset_counter():
-    session_data = load_session_data()
+# Function to check and reset the counter if necessary for a specific user
+def check_and_reset_user_counter(user_id):
+    all_user_data = load_user_session_data(user_id)
+    user_data = all_user_data[user_id]
     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"]
+    if user_data["last_reset_date"] != today:
+        user_data["last_reset_date"] = today
+        user_data["test_counter"] = 0
+    save_user_session_data(all_user_data)
+    return user_data["test_counter"]
+
+# Function to increment the counter for a specific user
+def increment_user_counter(user_id):
+    all_user_data = load_user_session_data(user_id)
+    all_user_data[user_id]["test_counter"] += 1
+    save_user_session_data(all_user_data)
+    return all_user_data[user_id]["test_counter"]
 
 # Function to encode the image
 def encode_image(image_array):
@@ -153,8 +161,8 @@ css = """
   }
 """
 
-# Initialize the test counter
-test_counter = 0
+# Initialize user ID
+user_id = str(uuid.uuid4())
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown("<h1>WordLift Product Description Generation - [FREE]</h1>")
@@ -196,7 +204,6 @@ with gr.Blocks(css=css) as demo:
                     value="Short Formal 📝",
                     elem_classes="gr-box"
                 )
-
                 custom_instruction = gr.Textbox(
                     label="Custom Instruction (Only for 'Other')", visible=False, elem_classes="gr-box"
                 )
@@ -206,6 +213,7 @@ with gr.Blocks(css=css) as demo:
                 redirect_message = gr.Markdown(visible=False, elem_classes="redirect-message")
                 test_counter_display = gr.Markdown(visible=True, label="Test Counter")
 
+
         # Toggle visibility of custom instruction based on selected type
         def toggle_custom_instruction(type_selection):
             return gr.update(visible=(type_selection == "Other"))
@@ -216,10 +224,11 @@ with gr.Blocks(css=css) as demo:
             outputs=[custom_instruction],
         )
 
-        # Modified function to handle test limit
+        # Modified function to handle test limit with improved messaging
         def handle_submit(image, description_type, custom_instruction):
-            check_and_reset_counter()  # Check and reset counter if it's a new day
-            test_counter = increment_counter()
+            global user_id
+            check_and_reset_user_counter(user_id)  # Check and reset counter if it's a new day
+            test_counter = increment_user_counter(user_id)
             if test_counter <= 4:
                 result = generate_product_description(image, description_type, custom_instruction)
                 remaining_tests = 4 - test_counter