rss-mcp-client / rss_client.py
gperdrizet's picture
Updated readme and UI.
a28b1b4 verified
raw
history blame
2.48 kB
'''RSS MCP server demonstration client app.'''
import os
import logging
from pathlib import Path
from logging.handlers import RotatingFileHandler
import gradio as gr
import client.interface as interface
from client.mcp_client import MCPClientWrapper
from client.anthropic_bridge import AnthropicBridge
# Make sure log directory exists
Path('logs').mkdir(parents=True, exist_ok=True)
# Set-up logger
logger = logging.getLogger()
logging.basicConfig(
handlers=[RotatingFileHandler(
'logs/rss_client.log',
maxBytes=100000,
backupCount=10,
mode='w'
)],
level=logging.INFO,
format='%(levelname)s - %(name)s - %(message)s'
)
logger = logging.getLogger(__name__)
RSS_CLIENT = MCPClientWrapper(
'https://agents-mcp-hackathon-rss-mcp-server.hf.space/gradio_api/mcp/sse'
)
BRIDGE = AnthropicBridge(
RSS_CLIENT,
api_key=os.environ['ANTHROPIC_API_KEY']
)
async def send_message(message: str, chat_history: list) -> str:
'''Submits user message to agent'''
function_logger = logging.getLogger(__name__ + '.submit_input')
function_logger.info('Submitting user message: %s', message)
chat_history.append({"role": "user", "content": message})
chat_history = await interface.agent_input(BRIDGE, chat_history)
return '', chat_history
with gr.Blocks(title='MCP RSS client') as demo:
gr.Markdown('# Agentic RSS reader')
gr.Markdown("""
Uses sister Space
[RSS feed reader](https://huggingface.co/spaces/Agents-MCP-Hackathon/rss-mcp-server)
via MCP. Click 'Connect to MCP server' to get started. Check out the
[main project repo on GitHub](https://github.com/gperdrizet/MCP-hackathon/tree/main).
Both Spaces by [George Perdrizet](https://www.linkedin.com/in/gperdrizet/).
""")
connect_btn = gr.Button('Connect to MCP server')
status = gr.Textbox(label='MCP server tool dump', interactive=False, lines=4)
chatbot = gr.Chatbot(
value=[],
height=500,
type='messages',
show_copy_button=True
)
msg = gr.Textbox(
label='Ask about content or articles on a site or platform',
placeholder='Is there anything new on Hacker News?',
scale=4
)
connect_btn.click(RSS_CLIENT.list_tools, outputs=status) # pylint: disable=no-member
msg.submit(send_message, [msg, chatbot], [msg, chatbot]) # pylint: disable=no-member
if __name__ == '__main__':
demo.launch(debug=True)