Update app2.py

app2.py

@@ -5,7 +5,6 @@ from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
 from sqlalchemy.future import select  # Correct async query API
 from sqlalchemy.orm import sessionmaker
 import logging
-
 import os
 import sys
 
@@ -15,6 +14,11 @@ sys.path.insert(0, parent_dir)
 
 logger = logging.getLogger(__name__)
 
+# Global variables for database session and engine
+db_session = None
+engine = None
+monitoring_task = None  # Global variable to track the monitoring task
+
 # This will constantly check the database status and update the textbox
 async def update_db_status(db_status_textbox):
     while True:
@@ -74,7 +78,7 @@ async def set_db_connection(host, port, user, password, db_name):
 
 # Main application that runs Gradio UI and background tasks
 async def main():
-    global db_session
+    global db_session, monitoring_task
     engine = None
 
     demo = gr.Blocks()
@@ -117,14 +121,24 @@ async def main():
                 message_input = gr.Textbox(placeholder="Type your message here...")
                 send_button = gr.Button("Send")
 
-        # Define button actions
+                # Define button actions
         async def on_start_click(target_urls_str: str, storage_loc: str, feed_enabled: bool, 
                                   host: str, port: str, user: str, password: str, db_name: str):
+            global monitoring_task
             urls = [url.strip() for url in target_urls_str.split(",")]
             await set_db_connection(host, port, user, password, db_name)  # Connect to the DB
-            asyncio.create_task(start_monitoring(urls, storage_loc, feed_enabled))  # Start monitoring
+            monitoring_task = asyncio.create_task(start_monitoring(urls, storage_loc, feed_enabled))  # Start monitoring
             return "Monitoring started."
 
+        async def on_stop_click():  # Define the on_stop_click function
+            global monitoring_task
+            if monitoring_task:
+                monitoring_task.cancel()  # Cancel the monitoring task
+                monitoring_task = None
+                return "Monitoring stopped."
+            else:
+                return "Monitoring is not running."
+
         async def on_view_feed_click(feed_url: str):
             # Logic to fetch and view RSS feed data based on URL
             return await fetch_feed_content(feed_url)
@@ -148,9 +162,8 @@ async def main():
     # Launch the Gradio demo
     await demo.launch()
 
-async def fetch_feed_content(feed_url: str):
+    async def fetch_feed_content(feed_url: str):
     # Logic to fetch RSS feed content from the provided URL
-    # You would replace this with actual RSS fetching and parsing logic
     return {
         'title': 'Sample Feed',
         'link': feed_url,
@@ -165,17 +178,12 @@ async def start_monitoring(urls, storage_location, feed_enabled):
     print(f"Starting monitoring for {urls}, saving to {storage_location}, RSS enabled: {feed_enabled}")
     return
 
-def stop_monitoring(url):
-    # Logic to stop monitoring a specific URL
-    print(f"Stopping monitoring for {url}")
-    return
-
 async def chatbot_response(message, chat_interface):
     # Example chatbot logic to respond to a user message
     response = f"Echo: {message}"
     chat_interface.append((message, response))
     return chat_interface, ""
 
-# Launch the app using asyncio
+    # Launch the app using asyncio
 if __name__ == "__main__":
     asyncio.run(main())