Update app.py

app.py

@@ -241,7 +241,7 @@ def chat(slug, user_id):
 
     # User-provided prompt
     if prompt := st.chat_input():
-        increment_user_chat_count(user_id)
+        # increment_user_chat_count(user_id)
         st.session_state.messages.append({"role": "user", "content": prompt})
         st.session_state.prompts = prompt
         with st.chat_message("user"):
@@ -260,6 +260,8 @@ def chat(slug, user_id):
                 {"role": "system", "content": message_content}
             )
             st.write(message_content)
+            if not is_user_in_unlimited_chat_group(user_id):
+                increment_user_chat_count(user_id)  # Increment count after response
 
 
 def init():
@@ -292,7 +294,7 @@ def main():
     load_dotenv()
     st.title("Chat with GPT :books:")
 
-    if slug:
+    if slug and user_id:
         chat(slug, user_id)
 
     else: