Merge pull request #5 from gperdrizet/dev

assets/html.py

@@ -0,0 +1,20 @@
+'''Collection of HTML elements for Gradio interface.'''
+
+TITLE = (
+    '''
+        <center> 
+            <h1>Agentic RSS reader</h1>
+        </center>
+    '''
+)
+
+DESCRIPTION = (
+    '''
+        <p>Uses sister Space 
+        <a href='https://huggingface.co/spaces/Agents-MCP-Hackathon/rss-mcp-server'>
+        RSS feed reader</a> via MCP. Click 'Connect to MCP server' to get started. 
+        Check out the <a href='https://github.com/gperdrizet/MCP-hackathon/tree/main'>
+        main project repo on GitHub</a>. Both Spaces by
+        <a href=https://www.linkedin.com/in/gperdrizet/'>George Perdrizet</a>.</p>
+    '''
+)

client/gradio_functions.py

@@ -0,0 +1,36 @@
+'''Collections of helper functions for Gradio user interface.'''
+
+import os
+import re
+
+def update_log(n: int = 10):
+    '''Gets updated logging output from disk to display to user.
+    
+    Args:
+        n: number of most recent lines of logging output to display
+
+    Returns:
+        Logging output as string
+    '''
+
+    with open('logs/rss_client.log', 'r', encoding='utf-8') as log_file:
+        lines = log_file.readlines()
+
+    return ''.join(lines[-n:])
+
+
+def delete_old_logs(directory:str, basename:str) -> None:
+    '''Deletes old log files from previous optimization sessions, if present.
+    
+    Args:
+        directory: path to log file directory as string
+        basename: log file base name as string
+        
+    Returns:
+        None
+    '''
+
+    for filename in os.listdir(directory):
+        file_path = os.path.join(directory, filename)
+        if re.search(basename, filename):
+            os.remove(file_path)

rss_client.py

@@ -6,6 +6,8 @@ from pathlib import Path
 from logging.handlers import RotatingFileHandler
 
 import gradio as gr
+import assets.html as html
+import client.gradio_functions as gradio_funcs
 import client.interface as interface
 from client.mcp_client import MCPClientWrapper
 from client.anthropic_bridge import AnthropicBridge
@@ -13,6 +15,9 @@ from client.anthropic_bridge import AnthropicBridge
 # Make sure log directory exists
 Path('logs').mkdir(parents=True, exist_ok=True)
 
+# Clear old logs if present
+gradio_funcs.delete_old_logs('logs', 'rss_client')
+
 # Set-up logger
 logger = logging.getLogger()
 
@@ -62,21 +67,29 @@ async def send_message(message: str, chat_history: list) -> str:
 
 
 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/).
-    """)
+    with gr.Row():
+        gr.HTML(html.TITLE)
+
+    gr.Markdown(html.DESCRIPTION)
 
+    # MCP connection/tool dump
     connect_btn = gr.Button('Connect to MCP server')
     status = gr.Textbox(label='MCP server tool dump', interactive=False, lines=4)
+    connect_btn.click(RSS_CLIENT.list_tools, outputs=status) # pylint: disable=no-member
+
+    # Log output
+    logs = gr.Textbox(label='Client logs', lines=10, max_lines=10)
+    timer = gr.Timer(1, active=True)
 
+    timer.tick( # pylint: disable=no-member
+        lambda: gradio_funcs.update_log(), # pylint: disable=unnecessary-lambda
+        outputs=logs
+    )
+
+    # Chat interface
     chatbot = gr.Chatbot(
         value=[],
-        height=800,
+        height=500,
         type='messages',
         show_copy_button=True
     )
@@ -88,8 +101,6 @@ with gr.Blocks(title='MCP RSS client') as demo:
         scale=4
     )
 
-    connect_btn.click(RSS_CLIENT.list_tools, outputs=status) # pylint: disable=no-member
-
     msg.submit( # pylint: disable=no-member
         send_message,
         [msg, chatbot],
@@ -101,7 +112,7 @@ if __name__ == '__main__':
     current_directory = os.getcwd()
 
     if 'pyrite' in current_directory:
-        demo.launch(server_name="0.0.0.0", server_port=7860)
+        demo.launch(server_name='0.0.0.0', server_port=7860)
 
     else:
         demo.launch()