Update app2.py

app2.py

@@ -26,6 +26,9 @@ engine = None
 monitoring_task = None
 logger = logging.getLogger(__name__)
 
+# Configure logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+
 # Function for dynamically setting the database connection
 async def set_db_connection(host: str, port: str, user: str, password: str, db_name: str):
     global db_session, engine
@@ -95,11 +98,19 @@ async def predict_website_traffic(url: str) -> Dict[str, Any]:
         model = LinearRegression()
         model.fit(X_train, y_train)
         y_pred = model.predict(X_test)
-        return {'traffic': y_pred[0]}
+        return {'traffic': y_pred}
     except Exception as e:
         logger.error(f"Error predicting website traffic: {e}")
         return {}
 
+# Function to update database status
+async def update_db_status():
+    try:
+        await db_session.execute("SELECT 1")
+        return "Connected"
+    except SQLAlchemyError:
+        return "Disconnected"
+
 # Main application that runs Gradio UI and background tasks
 async def main():
     global db_session, monitoring_task
@@ -128,15 +139,15 @@ async def main():
                 stop_button = gr.Button("Stop Monitoring")
             with gr.Column():
                 feed_content = gr.JSON(label="RSS Feed Content")
-                chatbot_interface = gr.Chatbot(type='messages')
+                chatbot_interface = gr.Chatbot(type='messages', stateful=True)  # Enable session state
                 message_input = gr.Textbox(placeholder="Type your message here...")
                 send_button = gr.Button("Send")
                 scrape_button = gr.Button("Scrape Website")
                 analyze_button = gr.Button("Analyze Website Content")
                 predict_button = gr.Button("Predict Website Traffic")
                 scrape_output = gr.Textbox(label="Scraped Website Content", interactive=False)
-                analyze_output = gr.JSON(label="Website Content Analysis", interactive=False)
-                predict_output = gr.JSON(label="Website Traffic Prediction", interactive=False)
+                analyze_output = gr.JSON(label="Website Content Analysis")  # Removed interactive=False
+                predict_output = gr.JSON(label="Website Traffic Prediction")  # Removed interactive=False
         # 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
@@ -170,8 +181,21 @@ async def main():
         # Load and check database status when the UI is loaded
         demo.load(update_db_status, outputs=db_status_textbox)
         asyncio.create_task(periodic_update_with_error_handling())
-        # Launch the Gradio demo
-        await demo.launch()
+        # Launch the Gradio demo with a custom theme
+        await demo.launch(theme="default", title="Website Monitor and Chatbot")
+
+# Function to handle chatbot responses with session state
+async def chatbot_response(message: str, chat_history: List[str]) -> List[str]:
+    chat_history = chat_history or []
+    response = f"Echo: {message}"
+    chat_history.append((message, response))
+    return chat_history, ""
+
+# Function to start monitoring
+async def start_monitoring(urls: List[str], storage_location: str, feed_enabled: bool):
+    # Logic to start monitoring URLs and optionally save to CSV or enable RSS
+    print(f"Starting monitoring for {urls}, saving to {storage_location}, RSS enabled: {feed_enabled}")
+    return
 
 # Launch the app using asyncio
 if __name__ == "__main__":