gperdrizet commited on
Commit
b57bc26
·
verified ·
1 Parent(s): 53d2fb7

Refactored some Gradio UI helper functions out of rss_client.py

Browse files
Files changed (2) hide show
  1. client/gradio_functions.py +36 -0
  2. rss_client.py +9 -16
client/gradio_functions.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ '''Collections of helper functions for Gradio user interface.'''
2
+
3
+ import os
4
+ import re
5
+
6
+ def update_log(n: int = 10):
7
+ '''Gets updated logging output from disk to display to user.
8
+
9
+ Args:
10
+ n: number of most recent lines of logging output to display
11
+
12
+ Returns:
13
+ Logging output as string
14
+ '''
15
+
16
+ with open('logs/rss_client.log', 'r', encoding='utf-8') as log_file:
17
+ lines = log_file.readlines()
18
+
19
+ return ''.join(lines[-n:])
20
+
21
+
22
+ def delete_old_logs(directory:str, basename:str) -> None:
23
+ '''Deletes old log files from previous optimization sessions, if present.
24
+
25
+ Args:
26
+ directory: path to log file directory as string
27
+ basename: log file base name as string
28
+
29
+ Returns:
30
+ None
31
+ '''
32
+
33
+ for filename in os.listdir(directory):
34
+ file_path = os.path.join(directory, filename)
35
+ if re.search(basename, filename):
36
+ os.remove(file_path)
rss_client.py CHANGED
@@ -6,6 +6,7 @@ from pathlib import Path
6
  from logging.handlers import RotatingFileHandler
7
 
8
  import gradio as gr
 
9
  import client.interface as interface
10
  from client.mcp_client import MCPClientWrapper
11
  from client.anthropic_bridge import AnthropicBridge
@@ -13,6 +14,9 @@ from client.anthropic_bridge import AnthropicBridge
13
  # Make sure log directory exists
14
  Path('logs').mkdir(parents=True, exist_ok=True)
15
 
 
 
 
16
  # Set-up logger
17
  logger = logging.getLogger()
18
 
@@ -60,21 +64,6 @@ async def send_message(message: str, chat_history: list) -> str:
60
 
61
  return '', chat_history
62
 
63
- def update_log(n: int = 10):
64
- '''Gets updated logging output from disk to display to user.
65
-
66
- Args:
67
- n: number of most recent lines of logging output to display
68
-
69
- Returns:
70
- Logging output as string
71
- '''
72
-
73
- with open('logs/rss_client.log', 'r', encoding='utf-8') as log_file:
74
- lines = log_file.readlines()
75
-
76
- return ''.join(lines[-n:])
77
-
78
 
79
  with gr.Blocks(title='MCP RSS client') as demo:
80
  gr.Markdown('# Agentic RSS reader')
@@ -94,7 +83,11 @@ with gr.Blocks(title='MCP RSS client') as demo:
94
  # Log output
95
  logs = gr.Textbox(label='Client logs', lines=10, max_lines=10)
96
  timer = gr.Timer(1, active=True)
97
- timer.tick(lambda: update_log(), outputs=logs) # pylint: disable=no-member, disable=unnecessary-lambda
 
 
 
 
98
 
99
  # Chat interface
100
  chatbot = gr.Chatbot(
 
6
  from logging.handlers import RotatingFileHandler
7
 
8
  import gradio as gr
9
+ import client.gradio_functions as gradio_funcs
10
  import client.interface as interface
11
  from client.mcp_client import MCPClientWrapper
12
  from client.anthropic_bridge import AnthropicBridge
 
14
  # Make sure log directory exists
15
  Path('logs').mkdir(parents=True, exist_ok=True)
16
 
17
+ # Clear old logs if present
18
+ gradio_funcs.delete_old_logs('logs', 'rss_client')
19
+
20
  # Set-up logger
21
  logger = logging.getLogger()
22
 
 
64
 
65
  return '', chat_history
66
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
67
 
68
  with gr.Blocks(title='MCP RSS client') as demo:
69
  gr.Markdown('# Agentic RSS reader')
 
83
  # Log output
84
  logs = gr.Textbox(label='Client logs', lines=10, max_lines=10)
85
  timer = gr.Timer(1, active=True)
86
+
87
+ timer.tick( # pylint: disable=no-member
88
+ lambda: gradio_funcs.update_log(), # pylint: disable=unnecessary-lambda
89
+ outputs=logs
90
+ )
91
 
92
  # Chat interface
93
  chatbot = gr.Chatbot(