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

Added live client log display to Gradio UI.

Browse files
Files changed (1) hide show
  1. rss_client.py +27 -6
rss_client.py CHANGED
@@ -60,23 +60,46 @@ async def send_message(message: str, chat_history: list) -> str:
60
 
61
  return '', chat_history
62
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
63
 
64
  with gr.Blocks(title='MCP RSS client') as demo:
65
  gr.Markdown('# Agentic RSS reader')
66
- gr.Markdown("""
67
  Uses sister Space
68
  [RSS feed reader](https://huggingface.co/spaces/Agents-MCP-Hackathon/rss-mcp-server)
69
  via MCP. Click 'Connect to MCP server' to get started. Check out the
70
  [main project repo on GitHub](https://github.com/gperdrizet/MCP-hackathon/tree/main).
71
  Both Spaces by [George Perdrizet](https://www.linkedin.com/in/gperdrizet/).
72
- """)
73
 
 
74
  connect_btn = gr.Button('Connect to MCP server')
75
  status = gr.Textbox(label='MCP server tool dump', interactive=False, lines=4)
 
 
 
 
 
 
76
 
 
77
  chatbot = gr.Chatbot(
78
  value=[],
79
- height=800,
80
  type='messages',
81
  show_copy_button=True
82
  )
@@ -88,8 +111,6 @@ with gr.Blocks(title='MCP RSS client') as demo:
88
  scale=4
89
  )
90
 
91
- connect_btn.click(RSS_CLIENT.list_tools, outputs=status) # pylint: disable=no-member
92
-
93
  msg.submit( # pylint: disable=no-member
94
  send_message,
95
  [msg, chatbot],
@@ -101,7 +122,7 @@ if __name__ == '__main__':
101
  current_directory = os.getcwd()
102
 
103
  if 'pyrite' in current_directory:
104
- demo.launch(server_name="0.0.0.0", server_port=7860)
105
 
106
  else:
107
  demo.launch()
 
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')
81
+ gr.Markdown('''
82
  Uses sister Space
83
  [RSS feed reader](https://huggingface.co/spaces/Agents-MCP-Hackathon/rss-mcp-server)
84
  via MCP. Click 'Connect to MCP server' to get started. Check out the
85
  [main project repo on GitHub](https://github.com/gperdrizet/MCP-hackathon/tree/main).
86
  Both Spaces by [George Perdrizet](https://www.linkedin.com/in/gperdrizet/).
87
+ ''')
88
 
89
+ # MCP connection/tool dump
90
  connect_btn = gr.Button('Connect to MCP server')
91
  status = gr.Textbox(label='MCP server tool dump', interactive=False, lines=4)
92
+ connect_btn.click(RSS_CLIENT.list_tools, outputs=status) # pylint: disable=no-member
93
+
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(
101
  value=[],
102
+ height=500,
103
  type='messages',
104
  show_copy_button=True
105
  )
 
111
  scale=4
112
  )
113
 
 
 
114
  msg.submit( # pylint: disable=no-member
115
  send_message,
116
  [msg, chatbot],
 
122
  current_directory = os.getcwd()
123
 
124
  if 'pyrite' in current_directory:
125
+ demo.launch(server_name='0.0.0.0', server_port=7860)
126
 
127
  else:
128
  demo.launch()