Merge pull request #5 from gperdrizet/dev

.gitignore

@@ -0,0 +1,4 @@
+__pycache__
+.vscode
+.venv
+logs

assets/html.py

@@ -3,9 +3,9 @@
 TITLE = (
     '''
         <center> 
-            <h1>Letter counter</h1>
+            <h1>RSS feed reader</h1>
         </center>
     '''
 )
 
-DESCRIPTION = 'Enter text and a letter to count how many times the letter appears in the text.'
+DESCRIPTION = 'Enter a website to extract RSS feed entry titles.'

functions/helper_functions.py

@@ -0,0 +1,56 @@
+'''Helper functions for MCP tools.'''
+
+import logging
+import feedparser
+from findfeed import search
+
+
+def get_feed(website_url: str) -> str:
+    '''Finds the RSS feed URI for a website given the website's url.
+    
+    Args:
+        website_url: The url for the website to find the RSS feed for
+        
+    Returns:
+        The website's RSS feed URI as a string
+    '''
+
+    logger = logging.getLogger(__name__ + '.get_content')
+
+    feeds = search(website_url)
+
+    if len(feeds) > 0:
+        return str(feeds[0].url)
+
+    else:
+        return f'No feed found for {website_url}'
+
+
+def parse_feed(feed_uri: str) -> list:
+    '''Gets content from a remote RSS feed URI.
+    
+    Args:
+        feed_uri: The RSS feed to get content from
+
+    Returns:
+        List of titles for the 10 most recent entries in the RSS feed.
+    '''
+
+    logger = logging.getLogger(__name__ + '.parse_feed')
+
+    feed = feedparser.parse(feed_uri)
+    logger.info('%s yieled %s entries', feed_uri, len(feed.entries))
+
+    titles = []
+
+    for entry in feed.entries:
+
+        logger.debug('Entry attributes: %s', list(entry.keys()))
+
+        if 'title' in entry:
+            titles.append(entry.title)
+
+        if len(titles) >= 10:
+            break
+
+    return titles

functions/tools.py

@@ -1,18 +1,26 @@
 '''Tool functions for MCP server'''
 
+import logging
+import functions.helper_functions as helper_funcs
 
-def letter_counter(word, letter):
-    """
-    Count the number of occurrences of a letter in a word or text.
 
+def get_content(website_url: str) -> list:
+    '''Gets RSS feed content from a given website.
+    
     Args:
-        word (str): The input text to search through
-        letter (str): The letter to search for
+        website_url: URL of website to extract RSS feed content from
 
     Returns:
-        str: A message indicating how many times the letter appears
-    """
-    word = word.lower()
-    letter = letter.lower()
-    count = word.count(letter)
-    return count
+        List of titles for the 10 most recent entries in the RSS feed.
+    '''
+
+    logger = logging.getLogger(__name__ + '.get_content')
+    logger.info('Getting feed content for: %s', website_url)
+
+    feed_uri = helper_funcs.get_feed(website_url)
+    logger.info('get_feed() returned %s', feed_uri)
+
+    content = helper_funcs.parse_feed(feed_uri)
+    logger.info('parse_feed() returned %s', content)
+
+    return '\n'.join(content)

rss_server.py

@@ -1,9 +1,32 @@
 '''Main script to run gradio interface and MCP server.'''
 
+import logging
+from pathlib import Path
+from logging.handlers import RotatingFileHandler
+
 import gradio as gr
 import assets.html as html
 from functions import tools as tool_funcs
 
+# 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_server.log',
+        maxBytes=100000,
+        backupCount=10,
+        mode='w'
+    )],
+    level=logging.DEBUG,
+    format='%(levelname)s - %(name)s - %(message)s'
+)
+
+logger = logging.getLogger(__name__)
+
 
 with gr.Blocks() as demo:
 
@@ -11,16 +34,15 @@ with gr.Blocks() as demo:
         gr.HTML(html.TITLE)
 
     gr.Markdown(html.DESCRIPTION)
-    input_word = gr.Textbox('strawberry', label='Text')
-    target_letter = gr.Textbox('r', label='Word')
-    output = gr.Number(label='Letter count')
-    count_button = gr.Button('Count')
-
-    count_button.click( # pylint: disable=no-member
-        fn=tool_funcs.letter_counter,
-        inputs=[input_word, target_letter],
+    website_url = gr.Textbox('hackernews.com', label='Website URL')
+    output = gr.Textbox(label='RSS entry titles', lines=10)
+    submit_button = gr.Button('Submit')
+
+    submit_button.click( # pylint: disable=no-member
+        fn=tool_funcs.get_content,
+        inputs=website_url,
         outputs=output,
-        api_name='letter count'
+        api_name='Get RSS feed content'
     )