Update app.py

app.py

@@ -18,24 +18,6 @@ from sqlalchemy.exc import SQLAlchemyError
 import validators
 import asyncio
 
-# Periodic update function
-async def periodic_update():
-    while True:
-        await asyncio.sleep(300)  # Wait for 5 minutes
-        await update_feed_content()
-
-async def update_feed_content():
-    return generate_rss_feed()
-    
-def start_periodic_task():
-    loop = asyncio.get_event_loop()
-    if loop.is_running():
-        asyncio.create_task(periodic_update())
-    else:
-        loop.run_until_complete(periodic_update())
-
-# Start the periodic update task
-start_periodic_task()
 # Configure logging
 logging.basicConfig(level=logging.INFO,
                     format='%(asctime)s - %(levelname)s - %(message)s')
@@ -57,6 +39,50 @@ engine = None  # Initialize the database engine globally
 # Database setup
 Base = declarative_base()
 
+def generate_rss_feed():
+    session = Session()
+    try:
+        articles = session.query(Article).order_by(
+            Article.timestamp.desc()).limit(20).all()
+        feed = feedparser.FeedParserDict()
+        feed['title'] = 'Website Changes Feed'
+        feed['link'] = 'http://yourwebsite.com/feed'  # Replace if needed
+        feed['description'] = 'Feed of changes detected on monitored websites.'
+        feed['entries'] = [{
+            'title': article.title,
+            'link': article.url,
+            'description': article.content,
+            'published': article.timestamp
+        } for article in articles]
+        return feedparser.FeedGenerator().feed_from_dictionary(
+            feed).writeString('utf-8')
+    except SQLAlchemyError as e:
+        logger.error(f"Database error: {e}")
+        return None
+    finally:
+        session.close()
+
+async def update_feed_content():
+    return generate_rss_feed()
+
+# Periodic update function
+async def periodic_update():
+    while True:
+        await asyncio.sleep(300)  # Wait for 5 minutes
+        await update_feed_content()
+
+async def update_feed_content():
+    return generate_rss_feed()
+    
+def start_periodic_task():
+    loop = asyncio.get_event_loop()
+    if loop.is_running():
+        asyncio.create_task(periodic_update())
+    else:
+        loop.run_until_complete(periodic_update())
+
+# Start the periodic update task
+start_periodic_task()
 
 class Article(Base):
     __tablename__ = 'articles'
@@ -151,7 +177,6 @@ async def monitor_url(url: str, interval: int, storage_location: str,
 
             await asyncio.sleep(interval)
 
-
 async def start_monitoring(urls: List[str], storage_location: str,
                           feed_rss: bool):
     for url in urls:
@@ -172,31 +197,6 @@ def stop_monitoring(url: str):
         monitoring_tasks[url].cancel()
         del monitoring_tasks[url]
 
-
-def generate_rss_feed():
-    session = Session()
-    try:
-        articles = session.query(Article).order_by(
-            Article.timestamp.desc()).limit(20).all()
-        feed = feedparser.FeedParserDict()
-        feed['title'] = 'Website Changes Feed'
-        feed['link'] = 'http://yourwebsite.com/feed'
-        feed['description'] = 'Feed of changes detected on monitored websites.'
-        feed['entries'] = [{
-            'title': article.title,
-            'link': article.url,
-            'description': article.content,
-            'published': article.timestamp
-        } for article in articles]
-        return feedparser.FeedGenerator().feed_from_dictionary(
-            feed).writeString('utf-8')
-    except SQLAlchemyError as e:
-        logger.error(f"Database error: {e}")
-        return None
-    finally:
-        session.close()
-
-
 async def chatbot_response(message: str, history: List[Tuple[str, str]]):
     try:
         client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1",
@@ -297,15 +297,6 @@ with gr.Blocks() as demo:
         inputs=[message_input, chatbot_interface],
         outputs=[chatbot_interface, chatbot_interface])
 
-    async def update_feed_content():
-        return generate_rss_feed()
-
-    # Periodic update loop
-    async def periodic_update():
-        while True:
-            await asyncio.sleep(300)  # Wait for 5 minutes
-            await update_feed_content()
-
     # Start the periodic update task
     asyncio.create_task(periodic_update())