Remove message logging for privacy

app.py

@@ -1,21 +1,24 @@
 import os
 import sys
+import datetime
 
 from openai import OpenAI
 import gradio as gr
 from gradio.components.chatbot import ChatMessage, Message
 from typing import (
-    TYPE_CHECKING,
     Any,
     Literal,
-    Optional,
-    Union,
-    cast,
 )
 
+print(f"Gradio version: {gr.__version__}")
+
+DEBUG_LOG = False or os.environ.get("DEBUG_LOG") == "True"
+
 title = None  # "ServiceNow-AI Chat" # modelConfig.get('MODE_DISPLAY_NAME')
 description = None
 
+chat_start_count = 0
+
 model_config = {
     "MODEL_NAME": os.environ.get("MODEL_NAME"),
     "MODE_DISPLAY_NAME": os.environ.get("MODE_DISPLAY_NAME"),
@@ -31,6 +34,12 @@ client = OpenAI(
 )
 
 
+def log_message(message):
+    if DEBUG_LOG is True:
+        print(message)
+
+
+# Gradio 5.0.1 had issues with checking the message formats.  5.29.0 does not!
 def _check_format(messages: Any, type: Literal["messages", "tuples"] = "messages") -> None:
     if type == "messages":
         all_valid = all(
@@ -52,8 +61,8 @@ def _check_format(messages: Any, type: Literal["messages", "tuples"] = "messages
             raise Exception(
                 "Data incompatible with messages format. Each message should be a dictionary with 'role' and 'content' keys or a ChatMessage object."
             )
-        else:
-            print("_check_format() --> All messages are valid.")
+        # else:
+        #     print("_check_format() --> All messages are valid.")
     elif not all(
             isinstance(message, (tuple, list)) and len(message) == 2
             for message in messages
@@ -64,24 +73,26 @@ def _check_format(messages: Any, type: Literal["messages", "tuples"] = "messages
 
 
 def chat_fn(message, history):
-    print(f"{'-' * 80}\nchat_fn() --> Message: {message}")
+    log_message(f"{'-' * 80}\nchat_fn() --> Message: {message}")
+
+    global chat_start_count
+    chat_start_count = chat_start_count + 1
+    print(
+        f"{datetime.datetime.now()}: chat_start_count: {chat_start_count}, turns: {int(len(history if history else []) / 3)}")
+
     # Remove any assistant messages with metadata from history for multiple turns
-    print(f"Original History: {history}")
+    log_message(f"Original History: {history}")
     _check_format(history, "messages")
     history = [item for item in history if
                not (isinstance(item, dict) and
                     item.get("role") == "assistant" and
                     isinstance(item.get("metadata"), dict) and
                     item.get("metadata", {}).get("title") is not None)]
-    print(f"Updated History: {history}")
+    log_message(f"Updated History: {history}")
     _check_format(history, "messages")
 
-    # messages = history + [{"role": "user", "content": message}]
-    # print(f"Messages: {messages}")
-    # _check_format(messages, "messages")
-
     history.append({"role": "user", "content": message})
-    print(f"History with user message: {history}")
+    log_message(f"History with user message: {history}")
     _check_format(history, "messages")
 
     # Create the streaming response
@@ -97,7 +108,7 @@ def chat_fn(message, history):
         content="Thinking...",
         metadata={"title": "🧠 Thought"}
     ))
-    print(f"History added thinking: {history}")
+    log_message(f"History added thinking: {history}")
     _check_format(history, "messages")
 
     output = ""
@@ -137,7 +148,7 @@ def chat_fn(message, history):
         # _check_format(messages_to_yield, "messages")
         yield messages_to_yield
 
-    print(f"Final History: {history}")
+    log_message(f"Final History: {history}")
     _check_format(history, "messages")