Update app.py

app.py

@@ -10,6 +10,7 @@ from datetime import datetime, timedelta, timezone
 from uuid import uuid4
 import random
 import string
+import queue
 
 from flask import Flask, render_template_string, request, redirect, url_for, flash, jsonify, Response
 from PIL import Image
@@ -21,6 +22,17 @@ from werkzeug.utils import secure_filename
 from dotenv import load_dotenv
 import requests
 
+from selenium import webdriver
+from selenium.webdriver.chrome.service import Service as ChromeService
+from selenium.webdriver.chrome.options import Options as ChromeOptions
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.common.keys import Keys
+from selenium.common.exceptions import TimeoutException, NoSuchElementException
+from webdriver_manager.chrome import ChromeDriverManager
+
+
 load_dotenv()
 
 app = Flask(__name__)
@@ -66,6 +78,165 @@ COLOR_SCHEMES = {
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
+driver = None
+whatsapp_thread = None
+whatsapp_env_id = None
+whatsapp_queues = {
+    'incoming': queue.Queue(),
+    'outgoing': queue.Queue()
+}
+whatsapp_active = threading.Event()
+
+def whatsapp_bot_thread(env_id):
+    global driver, whatsapp_env_id
+    whatsapp_env_id = env_id
+    
+    try:
+        chrome_options = ChromeOptions()
+        user_data_dir = os.path.join(os.getcwd(), "whatsapp_profile")
+        os.makedirs(user_data_dir, exist_ok=True)
+        chrome_options.add_argument(f"--user-data-dir={user_data_dir}")
+        chrome_options.add_argument("--headless=new")
+        chrome_options.add_argument("--no-sandbox")
+        chrome_options.add_argument("--disable-dev-shm-usage")
+        chrome_options.add_argument("--disable-gpu")
+        chrome_options.add_argument("user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36")
+
+        try:
+            service = ChromeService(ChromeDriverManager().install())
+            driver = webdriver.Chrome(service=service, options=chrome_options)
+        except Exception as e:
+            logging.error(f"Could not start WebDriver with manager: {e}. Falling back to default.")
+            driver = webdriver.Chrome(options=chrome_options)
+
+        driver.get("https://web.whatsapp.com")
+        logging.info("WhatsApp Web page opened. Please scan the QR code if required.")
+
+        wait = WebDriverWait(driver, 60)
+        wait.until(EC.presence_of_element_located((By.XPATH, '//*[@id="side"]/div[1]/div/div[2]/div[2]/div/div[1]/p'))) # Wait for search bar
+        logging.info("WhatsApp Web is logged in and ready.")
+        whatsapp_active.set()
+        
+        processor = threading.Thread(target=process_whatsapp_queues, daemon=True)
+        processor.start()
+
+        last_processed_messages = {}
+
+        while whatsapp_active.is_set():
+            try:
+                unread_chats = driver.find_elements(By.XPATH, '//span[@data-testid="icon-unread-count"]')
+                for chat in unread_chats:
+                    try:
+                        parent_element = chat.find_element(By.XPATH, './ancestor::div[contains(@class, "_aou8")]')
+                        parent_element.click()
+                        time.sleep(1) 
+
+                        chat_title_element = driver.find_element(By.XPATH, '//header//span[@dir="auto"]')
+                        chat_name = chat_title_element.text
+                        
+                        messages = driver.find_elements(By.XPATH, '//div[contains(@class, "message-in")]')
+                        if messages:
+                            last_message = messages[-1]
+                            message_text_element = last_message.find_element(By.XPATH, './/span[contains(@class, "selectable-text")]/span')
+                            message_text = message_text_element.text
+                            
+                            if last_processed_messages.get(chat_name) != message_text:
+                                logging.info(f"New message from '{chat_name}': {message_text}")
+                                whatsapp_queues['incoming'].put({'sender': chat_name, 'message': message_text})
+                                last_processed_messages[chat_name] = message_text
+
+                    except (NoSuchElementException, StaleElementReferenceException):
+                        continue
+                time.sleep(5)
+            except Exception as e:
+                logging.error(f"Error in WhatsApp listener loop: {e}")
+                time.sleep(10)
+
+    except TimeoutException:
+        logging.error("Timed out waiting for WhatsApp Web to load. Please scan the QR code faster or check your connection.")
+    except Exception as e:
+        logging.error(f"An error occurred in the WhatsApp bot thread: {e}")
+    finally:
+        if driver:
+            driver.quit()
+        driver = None
+        whatsapp_active.clear()
+        logging.info("WhatsApp bot thread has stopped.")
+
+
+def process_whatsapp_queues():
+    while whatsapp_active.is_set():
+        try:
+            # Process incoming messages
+            if not whatsapp_queues['incoming'].empty():
+                data = whatsapp_queues['incoming'].get()
+                sender = data['sender']
+                message = data['message']
+                
+                # Simple chat history management (in-memory for this example)
+                # For a real app, this should be persisted in data.json
+                if 'whatsapp_chats' not in app.config:
+                    app.config['whatsapp_chats'] = {}
+                if sender not in app.config['whatsapp_chats']:
+                    app.config['whatsapp_chats'][sender] = []
+
+                history = app.config['whatsapp_chats'][sender]
+
+                ai_response = generate_chat_response(message, history, whatsapp_env_id)
+                
+                history.append({'role': 'user', 'text': message})
+                history.append({'role': 'ai', 'text': ai_response})
+                # Trim history to keep it from growing too large
+                app.config['whatsapp_chats'][sender] = history[-20:]
+
+                whatsapp_queues['outgoing'].put({'recipient': sender, 'message': ai_response})
+
+            # Process outgoing messages
+            if not whatsapp_queues['outgoing'].empty():
+                data = whatsapp_queues['outgoing'].get()
+                send_whatsapp_message(data['recipient'], data['message'])
+
+            time.sleep(1)
+        except Exception as e:
+            logging.error(f"Error processing WhatsApp queues: {e}")
+
+
+def send_whatsapp_message(recipient, message):
+    if not driver or not whatsapp_active.is_set():
+        logging.warning("WhatsApp driver not ready, cannot send message.")
+        return
+
+    try:
+        search_box = WebDriverWait(driver, 10).until(
+            EC.presence_of_element_located((By.XPATH, '//*[@id="side"]/div[1]/div/div[2]/div[2]/div/div[1]/p'))
+        )
+        search_box.clear()
+        search_box.send_keys(recipient)
+        time.sleep(2)
+
+        chat = WebDriverWait(driver, 10).until(
+            EC.presence_of_element_located((By.XPATH, f'//span[@title="{recipient}"]'))
+        )
+        chat.click()
+        time.sleep(1)
+
+        message_box = WebDriverWait(driver, 10).until(
+            EC.presence_of_element_located((By.XPATH, '//*[@id="main"]/footer/div[1]/div/span[2]/div/div[2]/div[1]/div/div[1]/p'))
+        )
+
+        for line in message.split('\n'):
+            message_box.send_keys(line)
+            message_box.send_keys(Keys.SHIFT, Keys.ENTER)
+        
+        message_box.send_keys(Keys.ENTER)
+        logging.info(f"Message sent to '{recipient}'.")
+
+    except TimeoutException:
+        logging.error(f"Could not find chat or message box for '{recipient}'.")
+    except Exception as e:
+        logging.error(f"Failed to send WhatsApp message to '{recipient}': {e}")
+
+
 def download_db_from_hf(specific_file=None, retries=DOWNLOAD_RETRIES, delay=DOWNLOAD_DELAY):
     if not HF_TOKEN_READ and not HF_TOKEN_WRITE:
         logging.warning("HF_TOKEN_READ/HF_TOKEN_WRITE not set. Download might fail for private repos.")
@@ -2501,6 +2672,17 @@ ADMIN_TEMPLATE = '''
                  {% endif %}
             </div>
 
+            <div class="section">
+                <h2><i class="fab fa-whatsapp"></i> Интеграция с WhatsApp</h2>
+                {% if whatsapp_is_active %}
+                    <p style="color: green;">WhatsApp бот активен для среды <strong>{{ whatsapp_env_id }}</strong>.</p>
+                    <a href="{{ url_for('stop_whatsapp_bot', env_id=env_id) }}" class="button delete-button">Остановить WhatsApp Бота</a>
+                {% else %}
+                    <p>Запустите ИИ-консультанта в WhatsApp. После запуска, откроется браузер. Если потребуется, отсканируйте QR-код.</p>
+                    <a href="{{ url_for('start_whatsapp_bot', env_id=env_id) }}" class="button add-button" style="background-color: #25D366; color: white;"><i class="fab fa-whatsapp"></i> Запустить WhatsApp Бота</a>
+                {% endif %}
+            </div>
+
             <div class="section">
                 <h2><i class="fas fa-sync-alt"></i> Синхронизация с Датацентром</h2>
                 <div class="sync-buttons">
@@ -3663,7 +3845,9 @@ def admin(env_id):
         currencies=CURRENCIES,
         color_schemes=COLOR_SCHEMES,
         env_id=env_id,
-        chat_status=chat_status
+        chat_status=chat_status,
+        whatsapp_is_active=whatsapp_active.is_set(),
+        whatsapp_env_id=whatsapp_env_id
     )
 
 @app.route('/generate_description_ai', methods=['POST'])
@@ -3769,6 +3953,33 @@ def force_download(env_id):
         flash(f"Ошибка при скачивании с Hugging Face: {e}", 'error')
     return redirect(url_for('admin', env_id=env_id))
 
+@app.route('/<env_id>/whatsapp/start')
+def start_whatsapp_bot(env_id):
+    global whatsapp_thread
+    if whatsapp_thread and whatsapp_thread.is_alive():
+        flash('WhatsApp бот уже запущен.', 'warning')
+    else:
+        whatsapp_thread = threading.Thread(target=whatsapp_bot_thread, args=(env_id,), daemon=True)
+        whatsapp_thread.start()
+        flash('Запуск WhatsApp бота... Пожалуйста, будьте готовы отсканировать QR-код в новом окне браузера.', 'success')
+    return redirect(url_for('admin', env_id=env_id))
+
+@app.route('/<env_id>/whatsapp/stop')
+def stop_whatsapp_bot(env_id):
+    global driver, whatsapp_thread
+    if not (whatsapp_thread and whatsapp_thread.is_alive()):
+        flash('WhatsApp бот не был запущен.', 'warning')
+    else:
+        whatsapp_active.clear()
+        if driver:
+            try:
+                driver.quit()
+            except:
+                pass
+        flash('WhatsApp бот остановлен.', 'success')
+    return redirect(url_for('admin', env_id=env_id))
+
+
 if __name__ == '__main__':
     configure_gemini()
     download_db_from_hf()