Update app2.py

app2.py

@@ -7,7 +7,6 @@ from typing import List, Tuple
 import aiohttp
 import datetime
 import hashlib
-from pathlib import Path
 
 import feedparser
 import gradio as gr
@@ -19,8 +18,7 @@ import validators
 from bs4 import BeautifulSoup
 
 # Configure logging
-logging.basicConfig(level=logging.INFO,
-                    format='%(asctime)s - %(levelname)s - %(message)s')
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
 # Configuration
@@ -87,7 +85,8 @@ async def save_to_csv(storage_location: str, url: str, title: str,
                     content: str, timestamp: datetime.datetime):
     try:
         os.makedirs(os.path.dirname(storage_location), exist_ok=True)
-        with open(storage_location, "a", newline='', encoding="utf-8") as csvfile:
+        with open(storage_location, "a", newline='',
+                  encoding="utf-8") as csvfile:
             csv_writer = csv.writer(csvfile)
             csv_writer.writerow([
                 timestamp.strftime("%Y-%m-%d %H:%M:%S"), url, title, content
@@ -220,17 +219,18 @@ async def update_db_status(db_status):
     while True:
         try:
             await db_session.execute("SELECT 1")
-            await db_status.update(value="Connected")
+            db_status = "Connected"
         except SQLAlchemyError:
-            await db_status.update(value="Disconnected")
+            db_status = "Disconnected"
         await asyncio.sleep(60)  # Check every minute
+        return db_status
 
 
 async def update_feed_content(db_session):
     try:
         articles = await db_session.query(Article).order_by(
             Article.timestamp.desc()).limit(20).all()
-        feed        {
+        feed = {
             'title': 'Website Changes Feed',
             'link': 'http://yourwebsite.com/feed',
             'description': 'Feed of changes detected on monitored websites.',
@@ -305,20 +305,34 @@ async def main():
                         placeholder="Type your message here...")
                     send_button = gr.Button("Send")
 
-        start_button.click(
-            start_monitoring,
-            inputs=[target_urls, storage_location, feed_rss_checkbox],
-            outputs=status_text)
+        async def on_start_click(target_urls_str: str, storage_loc: str,
+                                  feed_enabled: bool):
+            urls = [url.strip() for url in target_urls_str.split(",")]
+            await start_monitoring(urls,
+                                      storage_loc if storage_loc else None,
+                                      feed_enabled)
+            return "Monitoring started for valid URLs."
 
-        stop_button.click(lambda url: stop_monitoring(url),
-                          inputs=target_urls,
-                          outputs=status_text)
+        async def on_stop_click():
+            for url in list(monitoring_tasks.keys()):
+                stop_monitoring(url)
+            return "Monitoring stopped for all URLs."
 
+        start_button.click(
+            on_start_click,
+            inputs=[target_urls, storage_location, feed_rss_checkbox],
+            outputs=[status_text])
+        stop_button.click(on_stop_click, outputs=[status_text])
         send_button.click(
             chatbot_response,
             inputs=[message_input, chatbot_interface],
             outputs=[chatbot_interface, message_input])
 
+        feed_updater = gr.Timer(every=300)
+        feed_updater.tick(fn=update_feed_content,
+                            inputs=[db_session],
+                            outputs=feed_content)
+
     asyncio.create_task(periodic_update_with_error_handling(db_session))
     asyncio.create_task(update_db_status(db_status))
 
@@ -331,4 +345,4 @@ async def main():
 
 
 if __name__ == "__main__":
-    asyncio.run(main())
+    asyncio.run(main())