Added live client log display to Gradio UI.
Browse files- 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=
|
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=
|
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()
|