Update app.py

app.py

@@ -14,31 +14,47 @@ import os
 
 # --- Ensure chat_log.txt exists ---
 log_file = "chat_log.txt"
-if not os.path.exists(log_file):
-    with open(log_file, 'w') as f:
-        pass  # Just create the file
+try:
+    if not os.path.exists(log_file):
+        with open(log_file, 'w') as f:
+            f.write("Log file created.\n")  # Write a simple message to the log file
+    print(f"{log_file} is ready for logging.")
+except Exception as e:
+    print(f"Error creating log file: {e}")
 
 # --- Logging Setup ---
-logging.basicConfig(filename=log_file, level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
-logging.debug("Logging setup complete.")
+try:
+    logging.basicConfig(filename=log_file, level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
+    logging.debug("Logging setup complete.")
+except Exception as e:
+    print(f"Error setting up logging: {e}")
 
 # Load the model
-ner_pipeline = pipeline("ner", model="Sevixdd/roberta-base-finetuned-ner")
-logging.debug("NER pipeline loaded.")
+try:
+    ner_pipeline = pipeline("ner", model="Sevixdd/roberta-base-finetuned-ner")
+    logging.debug("NER pipeline loaded.")
+except Exception as e:
+    logging.error(f"Error loading NER pipeline: {e}")
 
 # Load the dataset
-dataset = load_dataset("surrey-nlp/PLOD-filtered")
-logging.debug("Dataset loaded.")
+try:
+    dataset = load_dataset("surrey-nlp/PLOD-filtered")
+    logging.debug("Dataset loaded.")
+except Exception as e:
+    logging.error(f"Error loading dataset: {e}")
 
 # --- Prometheus Metrics Setup ---
-REQUEST_COUNT = Counter('gradio_request_count', 'Total number of requests')
-REQUEST_LATENCY = Histogram('gradio_request_latency_seconds', 'Request latency in seconds')
-ERROR_COUNT = Counter('gradio_error_count', 'Total number of errors')
-RESPONSE_SIZE = Histogram('gradio_response_size_bytes', 'Size of responses in bytes')
-CPU_USAGE = Gauge('system_cpu_usage_percent', 'System CPU usage in percent')
-MEM_USAGE = Gauge('system_memory_usage_percent', 'System memory usage in percent')
-QUEUE_LENGTH = Gauge('chat_queue_length', 'Length of the chat queue')
-logging.debug("Prometheus metrics setup complete.")
+try:
+    REQUEST_COUNT = Counter('gradio_request_count', 'Total number of requests')
+    REQUEST_LATENCY = Histogram('gradio_request_latency_seconds', 'Request latency in seconds')
+    ERROR_COUNT = Counter('gradio_error_count', 'Total number of errors')
+    RESPONSE_SIZE = Histogram('gradio_response_size_bytes', 'Size of responses in bytes')
+    CPU_USAGE = Gauge('system_cpu_usage_percent', 'System CPU usage in percent')
+    MEM_USAGE = Gauge('system_memory_usage_percent', 'System memory usage in percent')
+    QUEUE_LENGTH = Gauge('chat_queue_length', 'Length of the chat queue')
+    logging.debug("Prometheus metrics setup complete.")
+except Exception as e:
+    logging.error(f"Error setting up Prometheus metrics: {e}")
 
 # --- Queue and Metrics ---
 chat_queue = Queue()  # Define chat_queue globally