Update api/utils.py

api/utils.py

@@ -25,7 +25,7 @@ logger = setup_logger(__name__)
 # Define the blocked message
 BLOCKED_MESSAGE = "Generated by BLACKBOX.AI, try unlimited chat https://www.blackbox.ai"
 
-# Function to create chat completion data
+# Helper function to create chat completion data
 def create_chat_completion_data(
     content: str, model: str, timestamp: int, finish_reason: Optional[str] = None
 ) -> Dict[str, Any]:
@@ -46,20 +46,20 @@ def create_chat_completion_data(
 
 # Function to convert message to dictionary format, ensuring base64 data and optional model prefix
 def message_to_dict(message, model_prefix: Optional[str] = None):
-    content = message.content
+    content = message.content if isinstance(message.content, str) else message.content[0]["text"]
     if model_prefix:
         content = f"{model_prefix} {content}"
     message_dict = {"role": message.role, "content": content}
-
-    # Check if the message includes an image in base64 format
+    
+    # Check if the message includes image data
     if hasattr(message, 'image_base64') and message.image_base64:
-        # Ensure the base64 image data has the correct data URI prefix
         image_base64 = message.image_base64
+        # Ensure the base64 image data has the correct data URI prefix
         if not image_base64.startswith('data:image/'):
             # Assuming JPEG format; adjust if necessary
             image_base64 = f"data:image/jpeg;base64,{image_base64}"
-        # Generate a unique file path or name
-        file_name = f"image_{uuid.uuid4()}.jpg"
+        # Generate a unique file path
+        file_name = f"MultipleFiles/{uuid.uuid4().hex}.jpg"
         message_dict['data'] = {
             'imagesData': [
                 {
@@ -124,7 +124,7 @@ async def process_streaming_response(request: ChatRequest):
         "userId": None,
         "userSelectedModel": MODEL_MAPPING.get(request.model, request.model),
         "userSystemPrompt": None,
-        "validated": h_value,  # Dynamically set the validated field
+        "validated": h_value,
         "visitFromDelta": False,
         "webSearchModePrompt": False,
         "imageGenerationMode": False,  # Added this line
@@ -209,7 +209,7 @@ async def process_non_streaming_response(request: ChatRequest):
         "userId": None,
         "userSelectedModel": MODEL_MAPPING.get(request.model, request.model),
         "userSystemPrompt": None,
-        "validated": h_value,  # Dynamically set the validated field
+        "validated": h_value,
         "visitFromDelta": False,
         "webSearchModePrompt": False,
         "imageGenerationMode": False,  # Added this line