Update app2.py

app2.py

@@ -1,13 +1,14 @@
-
+import asyncio
 import csv
 import logging
 import os
 from typing import List, Tuple
-import asyncio
+
+import aiohttp
 import datetime
 import hashlib
+from pathlib import Path
 
-import aiohttp
 import feedparser
 import gradio as gr
 from huggingface_hub import InferenceClient
@@ -53,7 +54,6 @@ async def create_db_engine(db_url):
     except SQLAlchemyError as e:
         logger.error(f"Database error: {e}")
         raise
-
 def sanitize_url(url: str) -> str:
     return url if validators.url(url) else None
 
@@ -148,20 +148,40 @@ def stop_monitoring(url: str):
         asyncio.create_task(cleanup_resources(url))
         del monitoring_tasks[url]
     return "Monitoring stopped"
+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 chatbot_response(message: str, history: List[Tuple[str, str]]):
     try:
         client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1", token=HUGGINGFACE_API_KEY)
         response = await client.text_generation(message, max_new_tokens=100)
 
-        history.append({"role": "user", "content": message})
-        history.append({"role": "assistant", "content": response[0]['generated_text']})
+        history.append((message, response[0]['generated_text']))
 
         return history, history
     except Exception as e:
         logger.error(f"Chatbot error: {e}")
-        history.append({"role": "user", "content": message})
-        history.append({"role": "assistant", "content": "Error: Could not get a response from the chatbot."})
+        history.append((message, "Error: Could not get a response from the chatbot."))
         return history, history
 
 async def update_db_status(db_status):
@@ -175,7 +195,7 @@ async def update_db_status(db_status):
 
 async def update_feed_content(db_session):
     try:
-        articles = db_session.query(Article).order_by(Article.timestamp.desc()).limit(20).all()
+        articles = await db_session.       query(Article).order_by(Article.timestamp.desc()).limit(20).all()
         feed = {
             'title': 'Website Changes Feed',
             'link': 'http://yourwebsite.com/feed',
@@ -192,6 +212,7 @@ async def update_feed_content(db_session):
         logger.error(f"Database error: {e}")
         return None
 
+
 async def periodic_update_with_error_handling(db_session):
     while True:
         try:
@@ -200,6 +221,7 @@ async def periodic_update_with_error_handling(db_session):
         except Exception as e:
             logger.error(f"Error in periodic update: {e}")
 
+
 async def main():
     global db_session
     try:
@@ -216,48 +238,56 @@ async def main():
 
         with gr.Row():
             with gr.Column():
-                db_url = gr.Textbox(label="Database URL", value="sqlite:///monitoring.db")
-                db_status = gr.Textbox(label="Database Status", interactive=False, value="Connected")
+                db_url = gr.Textbox(
+                    label="Database URL", value="sqlite:///monitoring.db")
+                db_status = gr.Textbox(label="Database Status",
+                                      interactive=False,
+                                      value="Connected")
 
             with gr.Column():
                 with gr.Tab("Configuration"):
-                    target_urls = gr.Textbox(label="Target URLs (comma-separated)", placeholder="https://example.com, https://another-site.com")
-                    storage_location = gr.Textbox(label="Storage Location (CSV file path)", placeholder="/path/to/your/file.csv")
+                    target_urls = gr.Textbox(
+                        label="Target URLs (comma-separated)",
+                        placeholder=
+                        "https://example.com, https://another-site.com")
+                    storage_location = gr.Textbox(
+                        label="Storage Location (CSV file path)",
+                        placeholder="/path/to/your/file.csv")
                     feed_rss_checkbox = gr.Checkbox(label="Enable RSS Feed")
                     start_button = gr.Button("Start Monitoring")
                     stop_button = gr.Button("Stop Monitoring")
-                    status_text = gr.Textbox(label="Status", interactive=False)
-                    history_text = gr.Textbox(label="History", lines=10, interactive=False)
+                    status_text = gr.Textbox(label="Status",
+                                              interactive=False)
+                    history_text = gr.Textbox(label="History",
+                                              lines=10,
+                                              interactive=False)
 
                 with gr.Tab("User-End View"):
                     feed_content = gr.JSON(label="RSS Feed Content")
 
                 with gr.Tab("Chatbot"):
-                    chatbot_interface = gr.Chatbot()
-                    message_input = gr.Textbox(placeholder="Type your message here...")
+                    chatbot_interface = gr.Chatbot(type='messages')
+                    message_input = gr.Textbox(
+                        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
-        )
+            outputs=status_text)
 
-        stop_button.click(
-            lambda url: stop_monitoring(url),
-            inputs=target_urls,
-            outputs=status_text
-        )
+        stop_button.click(lambda url: stop_monitoring(url),
+                          inputs=target_urls,
+                          outputs=status_text)
 
         send_button.click(
             chatbot_response,
             inputs=[message_input, chatbot_interface],
-            outputs=[chatbot_interface, message_input]
-        )
+            outputs=[chatbot_interface, message_input])
 
     asyncio.create_task(periodic_update_with_error_handling(db_session))
     asyncio.create_task(update_db_status(db_status))
-    
+
     try:
         await demo.launch()
     finally:
@@ -265,5 +295,6 @@ async def main():
             await db_session.close()
             engine.dispose()
 
+
 if __name__ == "__main__":
-    asyncio.run(main())
+    asyncio.run(main())