Update api/utils.py

api/utils.py

@@ -1,6 +1,3 @@
-# utils.py or your main application file
-
-import os  # Import os to access environment variables
 from datetime import datetime
 import json
 import uuid
@@ -30,9 +27,6 @@ logger = setup_logger(__name__)
 # Define the blocked message
 BLOCKED_MESSAGE = "Generated by BLACKBOX.AI, try unlimited chat https://www.blackbox.ai"
 
-# Get the custom message from the environment variable
-FREE_USER_MESSAGE = os.getenv('FREE_USER_MESSAGE', 'Thank you for using our service! Upgrade to premium for more features.')
-
 # Helper function to create chat completion data
 def create_chat_completion_data(
     content: str, model: str, timestamp: int, finish_reason: Optional[str] = None
@@ -151,12 +145,7 @@ async def process_streaming_response(request: ChatRequest):
                         cleaned_content = strip_model_prefix(content, model_prefix)
                         yield f"data: {json.dumps(create_chat_completion_data(cleaned_content, request.model, timestamp))}\n\n"
 
-                # Append the custom message for free users
-                if request.is_free_user and FREE_USER_MESSAGE:
-                    timestamp = int(datetime.now().timestamp())
-                    yield f"data: {json.dumps(create_chat_completion_data(FREE_USER_MESSAGE, request.model, timestamp, 'stop'))}\n\n"
-                else:
-                    yield f"data: {json.dumps(create_chat_completion_data('', request.model, timestamp, 'stop'))}\n\n"
+                yield f"data: {json.dumps(create_chat_completion_data('', request.model, timestamp, 'stop'))}\n\n"
                 yield "data: [DONE]\n\n"
         except httpx.HTTPStatusError as e:
             logger.error(f"HTTP error occurred for Request ID {request_id}: {e}")
@@ -242,10 +231,6 @@ async def process_non_streaming_response(request: ChatRequest):
 
     cleaned_full_response = strip_model_prefix(full_response, model_prefix)
 
-    # Append the custom message for free users
-    if request.is_free_user and FREE_USER_MESSAGE:
-        cleaned_full_response += f"\n\n{FREE_USER_MESSAGE}"
-
     return {
         "id": f"chatcmpl-{uuid.uuid4()}",
         "object": "chat.completion",
@@ -259,4 +244,4 @@ async def process_non_streaming_response(request: ChatRequest):
             }
         ],
         "usage": None,
-    }
+    }