Update app.py

app.py

@@ -11,8 +11,11 @@ from sklearn.metrics import precision_score, recall_score, f1_score, accuracy_sc
 import requests
 from datasets import load_dataset
 import os
-from logging.handlers import RotatingFileHandler
-
+from logging import FileHandler
+from __future__ import annotations
+from typing import Iterable
+from gradio.themes.base import Base
+from gradio.themes.utils import colors, fonts, sizes
 
 
 # Ensure the log files exist
@@ -20,10 +23,10 @@ log_file_path = 'chat_log.log'
 debug_log_file_path = 'debug.log'
 if not os.path.exists(log_file_path):
     with open(log_file_path, 'w') as f:
-        f.write("")
+        f.write(" ")
 if not os.path.exists(debug_log_file_path):
     with open(debug_log_file_path, 'w') as f:
-        f.write("")
+        f.write(" ")
 
 
 # Create logger instance
@@ -35,13 +38,11 @@ formatter = logging.Formatter(
     '%(asctime)s - %(name)s - %(levelname)s - %(message)s', datefmt='%d-%b-%y %H:%M:%S')
 
 # Create handlers
-info_handler = RotatingFileHandler(
-    filename=log_file_path, mode='w', maxBytes=5*1024*1024, backupCount=2)
+info_handler = FileHandler( filename=log_file_path,  mode='w+')
 info_handler.setLevel(logging.INFO)
 info_handler.setFormatter(formatter)
 
-debug_handler = RotatingFileHandler(
-    filename=debug_log_file_path, mode='w', maxBytes=5*1024*1024, backupCount=2)
+debug_handler = FileHandler(filename=debug_log_file_path, mode='w+')
 debug_handler.setLevel(logging.DEBUG)
 debug_handler.setFormatter(formatter)
 
@@ -185,8 +186,8 @@ def stress_test(num_requests, message, delay):
                 "data": [message],
                 "fn_index": 0  # This might need to be updated based on your Gradio app's function index
             })
-            logger.debug(f"Request payload: {message}")
-            logger.debug(f"Response: {response.json()}")
+            logger.debug(f"Request payload: {message}",exc_info=True)
+            logger.debug(f"Response: {response.json()}",exc_info=True)
         except Exception as e:
             logger.debug(f"Error during stress test request: {e}", exc_info=True)
 
@@ -200,14 +201,10 @@ def stress_test(num_requests, message, delay):
     for t in threads:
         t.join()
 
+
 # --- Gradio Interface with Background Image and Three Windows ---
-with gr.Blocks(css="""
-body {
-    background-image: url("stag.jpeg"); 
-    background-size: cover; 
-    background-repeat: no-repeat;
-}
-""", title="PLOD Filtered with Monitoring") as demo:  # Load CSS for background image
+with gr.Blocks(title="PLOD Filtered with Monitoring") as demo:  # Load CSS for background image
+    
     with gr.Tab("Sentence input"):
         gr.Markdown("## Chat with the Bot")
         index_input = gr.Textbox(label="Enter A sentence:", lines=1)
@@ -250,7 +247,9 @@ body {
                 stress_test_status.value = f"Stress test failed: {e}"
 
         stress_test_button.click(run_stress_test, [num_requests_input, index_input_stress, delay_input], stress_test_status)
-
+    img = gr.Image(
+                "stag.jpeg", label="Image"
+            )
     # --- Update Functions ---
     def update_metrics(request_count_display, avg_latency_display):
         while True:
@@ -274,15 +273,10 @@ body {
     def update_logs(logs_display):
         while True:
             info_log_vector = []
-            # with open('debug.log', "r") as log_file_handler:
-            #     for line in log_file_handler:  # Skip empty lines
-            #         info_log_vector.append(line)
-            #         debugger.debug(info_log_vector)
-            #     logs_display.value = info_log_vector  # Display last 10 lines
             logs = []
             while not logs_queue.empty():
                 logs.append(logs_queue.get())
-            logs_display.value = "\n".join(logs)
+            logs_display.value = "\n".join(logs[-10:])
             time.sleep(1)  # Update every 1 second
 
     def display_model_params(model_params_display):
@@ -301,7 +295,7 @@ body {
     threading.Thread(target=start_http_server, args=(8000,), daemon=True).start()
     threading.Thread(target=update_metrics, args=(request_count_display, avg_latency_display), daemon=True).start()
     threading.Thread(target=update_usage, args=(cpu_usage_display, mem_usage_display), daemon=True).start()
-    threading.Thread(target=update_logs, args=(logs_display,), daemon=True).start()
+    threading.Thread(target=update_logs, args=(logs_display), daemon=True).start()
     threading.Thread(target=display_model_params, args=(model_params_display,), daemon=True).start()
     threading.Thread(target=update_queue_length, daemon=True).start()