update code to fail gracefully when logging no available

app.py

@@ -1,5 +1,6 @@
 import os
 import re
+import sys
 import time
 import json
 from itertools import cycle
@@ -152,8 +153,12 @@ def gradio_interface_function(template, text, is_example):
         # yield gr.update(value=chunk_info), gr.update(value=progress), gr.update(value=full_pred), gr.update(value=html_content)
         yield progress, full_pred, html_content
 
-    if not is_example and os.environ.get("LOG_SERVER") is not None:
-        log_event(text, template, full_pred)
+    # Conditionally log event if not an example and logging is configured
+    if not is_example:
+        try:
+            log_event(text, template, full_pred)
+        except Exception as e:
+            print(f"Warning: Could not log event: {e}", file=sys.stderr)
         
 
 # Set up the Gradio interface

nuextract_logging.py

@@ -3,14 +3,30 @@ import ssl
 import json
 import socket
 import datetime
+import sys
 
 
 def log_event(input_text, template, prediction):
-    timestamp = datetime.datetime.now(datetime.timezone.utc).isoformat()
-    log_message = f'<6>1 {timestamp} 149.202.165.20 example.org - - [exampleSDID@8485 X-OVH-TOKEN="{os.environ["LOG_TOKEN"]}" template={json.dumps(template)} text={json.dumps(input_text)} prediction={json.dumps(prediction)}] {timestamp} / {json.dumps(input_text[:20])}\n'
+    """Log extraction events to a remote server if configured through environment variables"""
+    # Check if environment variables are set
+    if not all(key in os.environ for key in ["LOG_SERVER", "LOG_PORT", "LOG_TOKEN"]):
+        print("Logging disabled: environment variables LOG_SERVER, LOG_PORT, or LOG_TOKEN not set", file=sys.stderr)
+        return
+    
+    try:
+        timestamp = datetime.datetime.now(datetime.timezone.utc).isoformat()
+        # Create a structured log message
+        log_message = f'<6>1 {timestamp} 149.202.165.20 example.org - - [exampleSDID@8485 X-OVH-TOKEN="{os.environ["LOG_TOKEN"]}" template={json.dumps(template)} text={json.dumps(input_text)} prediction={json.dumps(prediction)}] {timestamp} / {json.dumps(input_text[:20])}\n'
+
+        server = os.environ["LOG_SERVER"]
+        port = int(os.environ["LOG_PORT"])
+    except Exception as e:
+        print(f"Error preparing log message: {e}", file=sys.stderr)
+        return
 
-    server = os.environ["LOG_SERVER"]
-    port = int(os.environ["LOG_PORT"])
+    # Create a TCP connection and wrap it with SSL
+    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    ssl_sock = ssl.wrap_socket(sock)
 
     # Create a TCP connection and wrap it with SSL
     sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
@@ -22,8 +38,14 @@ def log_event(input_text, template, prediction):
         
         # Send the log message
         ssl_sock.sendall(log_message.encode('utf-8'))
-        print('Log message sent successfully.')
+        print('Log message sent successfully')
 
+    except Exception as e:
+        print(f"Error sending log message: {e}", file=sys.stderr)
+        
     finally:
         # Close the connection
-        ssl_sock.close()
+        try:
+            ssl_sock.close()
+        except:
+            pass