Update app.py

app.py

@@ -1,13 +1,47 @@
+import os
+import sys
+import gc
+import json
+import logging
+import asyncio
+import traceback
+from datetime import datetime
+from pathlib import Path
+from typing import Dict, Optional
+from contextlib import asynccontextmanager
+import tempfile
+
+# تنظیم مسیرها - استفاده از مسیر temp برای حل مشکل دسترسی
+BASE_DIR = Path(tempfile.gettempdir()) / "ai_assistant"
+CACHE_DIR = BASE_DIR / "cache"
+LOG_DIR = BASE_DIR / "logs"
+STATIC_DIR = BASE_DIR / "static"
+TEMPLATES_DIR = BASE_DIR / "templates"
+
+# ایجاد مسیرها
+for path in [BASE_DIR, CACHE_DIR, LOG_DIR, STATIC_DIR, TEMPLATES_DIR]:
+    path.mkdir(exist_ok=True, parents=True)
+
+# تنظیم متغیرهای محیطی
+os.environ["TRANSFORMERS_CACHE"] = str(CACHE_DIR)
+os.environ["HF_HOME"] = str(CACHE_DIR)
+os.environ["TORCH_HOME"] = str(CACHE_DIR / "torch")
+
+# کتابخانه‌های هوش مصنوعی
 import torch
 import transformers
 from transformers import pipeline
+
+# کتابخانه‌های وب و تلگرام
+import uvicorn
 from fastapi import FastAPI, WebSocket, Request
 from fastapi.responses import HTMLResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
-import uvicorn
 from telegram import Update, Bot, InlineKeyboardButton, InlineKeyboardMarkup, BotCommand
 from telegram.ext import Application, CommandHandler, MessageHandler, CallbackQueryHandler, filters
+
+# کتابخانه‌های دیگر
 import hazm
 import httpx
 
@@ -24,18 +58,18 @@ logger = logging.getLogger(__name__)
 
 class Config:
     """تنظیمات اصلی برنامه"""
-    
+
     # تنظیمات تلگرام
-    TELEGRAM_TOKEN = "7437859619:AAGeGG3ZkLM0OVaw-Exx1uMRE55JtBCZZCY"
-    CHAT_ID = "-1002228627548"
-    
+    TELEGRAM_TOKEN = "YOUR_TELEGRAM_BOT_TOKEN"
+    CHAT_ID = "YOUR_CHAT_ID"
+
     # تنظیمات مسیرها
     BASE_DIR = BASE_DIR
     CACHE_DIR = CACHE_DIR
     LOG_DIR = LOG_DIR
     STATIC_DIR = STATIC_DIR
     TEMPLATES_DIR = TEMPLATES_DIR
-    
+
     # تنظیمات مدل
     MODEL_NAME = "bigscience/bloom-1b1"
     MAX_LENGTH = 200
@@ -44,7 +78,7 @@ class Config:
     TOP_K = 50
     TOP_P = 0.95
     DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
-    
+
     # تنظیمات عمومی
     DEBUG = True
     HOST = "0.0.0.0"
@@ -52,14 +86,17 @@ class Config:
     MAX_RETRIES = 3
     TIMEOUT = 30
     CLEANUP_INTERVAL = 3600  # یک ساعت
+    MAX_QUEUE_SIZE = 1000
+    MAX_CACHE_SIZE = 1000
+    MAX_MESSAGE_LENGTH = 500
 
 class BloomAI:
     """کلاس مدیریت مدل هوش مصنوعی"""
-    
+
     def __init__(self):
         self.pipeline = None
         self.device = Config.DEVICE
-        
+
     async def initialize(self) -> bool:
         try:
             logger.info(f"Loading AI model: {Config.MODEL_NAME}")
@@ -90,7 +127,7 @@ class BloomAI:
             if not self.pipeline:
                 logger.error("Model not initialized")
                 return "مدل در دسترس نیست."
-                
+
             outputs = self.pipeline(
                 text,
                 max_length=Config.MAX_LENGTH,
@@ -102,16 +139,16 @@ class BloomAI:
                 num_return_sequences=1,
                 pad_token_id=self.pipeline.tokenizer.eos_token_id
             )
-            
+
             generated_text = outputs[0]['generated_text']
             # حذف متن ورودی از پاسخ
             response = generated_text.replace(text, "").strip()
-            
+
             if not response:
                 return "متأسفم، نتوانستم پاسخ مناسبی تولید کنم."
-                
+
             return response
-            
+
         except Exception as e:
             logger.error(f"Error generating response: {e}")
             logger.error(traceback.format_exc())
@@ -129,7 +166,7 @@ class BloomAI:
 
 class TelegramUI:
     """کلاس مدیریت رابط کاربری تلگرام"""
-    
+
     @staticmethod
     def get_main_menu() -> InlineKeyboardMarkup:
         """منوی اصلی با دکمه‌های تخت"""
@@ -169,7 +206,7 @@ class TelegramUI:
 
 class TelegramBot:
     """کلاس اصلی ربات تلگرام"""
-    
+
     def __init__(self, ai_model: BloomAI):
         self.model = ai_model
         self.ui = TelegramUI()
@@ -191,10 +228,10 @@ class TelegramBot:
         self.application.add_handler(CommandHandler("help", self.help_command))
         self.application.add_handler(CommandHandler("menu", self.menu_command))
         self.application.add_handler(CommandHandler("cancel", self.cancel_command))
-        
+
         # هندلر دکمه‌ها
         self.application.add_handler(CallbackQueryHandler(self.handle_callback))
-        
+
         # هندلر پیام‌های متنی
         self.application.add_handler(
             MessageHandler(filters.TEXT & ~filters.COMMAND, self.handle_message)
@@ -250,7 +287,7 @@ class TelegramBot:
         # پاک کردن صف پیام‌ها
         while not self.message_queue.empty():
             self.message_queue.get_nowait()
-        
+
         await update.message.reply_text(
             "❌ عملیات فعلی لغو شد.",
             reply_markup=self.ui.get_main_menu()
@@ -260,19 +297,19 @@ class TelegramBot:
         """پردازش دکمه‌های فشرده شده"""
         query = update.callback_query
         await query.answer()
-        
+
         if query.data == "main_menu":
             await query.message.edit_text(
                 "🏠 منوی اصلی:",
                 reply_markup=self.ui.get_main_menu()
             )
-            
+
         elif query.data == "new_chat":
             await query.message.edit_text(
                 "💭 لطفاً نوع چت را انتخاب کنید:",
                 reply_markup=self.ui.get_chat_menu()
             )
-            
+
         elif query.data in ["general_chat", "qa_chat"]:
             chat_type = "عمومی" if query.data == "general_chat" else "سوال و جواب"
             await query.message.edit_text(
@@ -282,9 +319,19 @@ class TelegramBot:
     async def handle_message(self, update: Update, context):
         """پردازش پیام‌های دریافتی"""
         try:
+            # بررسی طول پیام
+            if len(update.message.text) > Config.MAX_MESSAGE_LENGTH:
+                await update.message.reply_text("⚠️ پیام شما بیش از حد طولانی است.")
+                return
+
+            # بررسی حجم صف پیام‌ها
+            if self.message_queue.qsize() >= Config.MAX_QUEUE_SIZE:
+                await update.message.reply_text("⚠️ سیستم در حال حاضر شلوغ است. لطفاً کمی صبر کنید.")
+                return
+
             # نرمال‌سازی متن
             text = self.normalizer.normalize(update.message.text)
-            
+
             # بررسی کش
             if text in self.response_cache:
                 response = self.response_cache[text]
@@ -294,25 +341,32 @@ class TelegramBot:
                 processing_message = await update.message.reply_text(
                     "🤔 در حال پردازش پیام شما..."
                 )
-                
+
                 # دریافت پاسخ از مدل
                 response = await self.model.generate_response(text)
-                
+
+                # بررسی حجم کش
+                if len(self.response_cache) >= Config.MAX_CACHE_SIZE:
+                    # حذف قدیمی‌ترین پاسخ‌ها
+                    oldest_keys = list(self.response_cache.keys())[:100]
+                    for key in oldest_keys:
+                        del self.response_cache[key]
+
                 # ذخیره در کش
                 self.response_cache[text] = response
-                
+
                 # حذف پیام در حال پردازش
                 await processing_message.delete()
-            
+
             # ارسال پاسخ با دکمه برگشت به منو
             keyboard = [[InlineKeyboardButton("🏠 بازگشت به منو", callback_data="main_menu")]]
             reply_markup = InlineKeyboardMarkup(keyboard)
-            
+
             await update.message.reply_text(
                 f"🤖 {response}",
                 reply_markup=reply_markup
             )
-            
+
         except Exception as e:
             logger.error(f"خطا در پردازش پیام: {e}")
             logger.error(traceback.format_exc())
@@ -322,24 +376,28 @@ class TelegramBot:
 
     async def periodic_cleanup(self):
         """پاکسازی دوره‌ای حافظه و کش"""
-        while True:
-            await asyncio.sleep(Config.CLEANUP_INTERVAL)
-            logger.info("Starting periodic cleanup...")
-            
-            # پاکسازی کش پاسخ‌ها
-            self.response_cache.clear()
-            
-            # پاکسازی حافظه
-            await self.model.cleanup()
-            
-            logger.info("Periodic cleanup completed")
+        try:
+            while True:
+                await asyncio.sleep(Config.CLEANUP_INTERVAL)
+                logger.info("Starting periodic cleanup...")
+
+                # پاکسازی کش پاسخ‌ها
+                self.response_cache.clear()
+
+                # پاکسازی حافظه
+                await self.model.cleanup()
+
+                logger.info("Periodic cleanup completed")
+        except Exception as e:
+            logger.error(f"Error in periodic cleanup: {e}")
+            logger.error(traceback.format_exc())
 
     async def run(self):
         """اجرای ربات"""
         try:
             # شروع پاکسازی دوره‌ای
             asyncio.create_task(self.periodic_cleanup())
-            
+
             # راه‌اندازی ربات
             await self.application.initialize()
             await self.application.start()
@@ -347,7 +405,7 @@ class TelegramBot:
                 allowed_updates=Update.ALL_TYPES,
                 drop_pending_updates=True
             )
-            
+
         except Exception as e:
             logger.error(f"خطا در اجرای ربات: {e}")
             logger.error(traceback.format_exc())
@@ -358,11 +416,180 @@ async def lifespan(app: FastAPI):
     """مدیریت چرخه حیات برنامه"""
     # راه‌اندازی
     logger.info("Starting application...")
-    
+
     # ایجاد نمونه از مدل
     app.state.model = BloomAI()
     if not await app.state.model.initialize():
         logger.error("Failed to initialize AI model")
         sys.exit(1)
-    
-   
+
+    # راه‌اندازی ربات تلگرام
+    app.state.bot = TelegramBot(app.state.model)
+    asyncio.create_task(app.state.bot.run())
+
+    yield
+
+    # خاتمه
+    tasks = [t for t in asyncio.all_tasks() if t is not asyncio.current_task()]
+    for task in tasks:
+        task.cancel()
+    await asyncio.gather(*tasks, return_exceptions=True)
+    await app.state.bot.application.shutdown()
+    logger.info("Application shutdown completed")
+
+# برنامه FastAPI
+app = FastAPI(title="AI Assistant", lifespan=lifespan)
+
+# پیوست کردن فایل‌های استاتیک
+app.mount("/static", StaticFiles(directory=Config.STATIC_DIR), name="static")
+templates = Jinja2Templates(directory=Config.TEMPLATES_DIR)
+
+# HTML پیش‌فرض
+DEFAULT_HTML = """
+<!DOCTYPE html>
+<html dir="rtl" lang="fa">
+<head>
+    <meta charset="UTF-8">
+    <title>دستیار هوشمند</title>
+    <style>
+        body {
+            font-family: Tahoma, Arial;
+            background-color: #f5f5f5;
+            margin: 0;
+            padding: 20px;
+        }
+        .container {
+            max-width: 800px;
+            margin: 0 auto;
+            background: white;
+            padding: 20px;
+            border-radius: 10px;
+            box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+        }
+        h1 { color: #1a73e8; }
+        .status { margin: 20px 0; }
+        .chat-container {
+            margin-top: 20px;
+            border: 1px solid #e0e0e0;
+            border-radius: 8px;
+            padding: 10px;
+            min-height: 300px;
+        }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <h1>دستیار هوشمند</h1>
+        <div class="status">
+            <p>برای استفاده از دستیار هوشمند، به ربات تلگرام مراجعه کنید یا از رابط وب استفاده نمایید.</p>
+        </div>
+        <div class="chat-container" id="chat-container">
+        </div>
+        <input type="text" id="message-input" placeholder="پیام خود را بنویسید..." style="width: 100%; padding: 10px; margin-top: 10px;">
+        <button onclick="sendMessage()" style="margin-top: 10px; padding: 10px 20px;">ارسال</button>
+    </div>
+    <script>
+        const ws = new WebSocket(`ws://${window.location.host}/ws`);
+        const chatContainer = document.getElementById('chat-container');
+        const messageInput = document.getElementById('message-input');
+
+        ws.onmessage = function(event) {
+            const response = JSON.parse(event.data);
+            appendMessage('bot', response.response);
+        };
+
+        function appendMessage(type, text) {
+            const messageDiv = document.createElement('div');
+            messageDiv.style.padding = '10px';
+            messageDiv.style.margin = '5px';
+            messageDiv.style.borderRadius = '5px';
+
+            if (type === 'user') {
+                messageDiv.style.backgroundColor = '#e3f2fd';
+                messageDiv.style.marginLeft = '20%';
+            } else {
+                messageDiv.style.backgroundColor = '#f5f5f5';
+                messageDiv.style.marginRight = '20%';
+            }
+
+            messageDiv.textContent = text;
+            chatContainer.appendChild(messageDiv);
+            chatContainer.scrollTop = chatContainer.scrollHeight;
+        }
+
+        function sendMessage() {
+            const message = messageInput.value.trim();
+            if (message) {
+                appendMessage('user', message);
+                ws.send(message);
+                messageInput.value = '';
+            }
+        }
+
+        messageInput.addEventListener('keypress', function(e) {
+            if (e.key === 'Enter') {
+                sendMessage();
+            }
+        });
+    </script>
+</body>
+</html>
+"""
+
+@app.get("/", response_class=HTMLResponse)
+async def home(request: Request):
+    """صفحه اصلی"""
+    try:
+        # می‌توانید اطلاعات بیشتری را به تمپلیت ارسال کنید
+        return templates.TemplateResponse(
+            "index.html",
+            {
+                "request": request,
+                "bot_status": "فعال" if request.app.state.is_ready.is_set() else "در حال آماده‌سازی"
+            }
+        )
+    except Exception as e:
+        logger.error(f"خطا در نمایش صفحه ��صلی: {e}")
+        return HTMLResponse(content=DEFAULT_HTML)
+
+@app.websocket("/ws")
+async def websocket_endpoint(websocket: WebSocket):
+    """اتصال WebSocket"""
+    await websocket.accept()
+
+    try:
+        while True:
+            # دریافت پیام
+            text = await websocket.receive_text()
+
+            # تولید پاسخ
+            response = await app.state.model.generate_response(text)
+
+            # ارسال پاسخ
+            await websocket.send_json({
+                "response": response
+            })
+    except WebSocketDisconnect:
+        logger.info("اتصال WebSocket قطع شد")
+    except Exception as e:
+        logger.error(f"خطا در WebSocket: {e}")
+        await websocket.close()
+
+@app.exception_handler(404)
+async def not_found_handler(request: Request, exc: Exception):
+    """پردازش خطای 404"""
+    return HTMLResponse(content=DEFAULT_HTML, status_code=404)
+
+@app.exception_handler(500)
+async def server_error_handler(request: Request, exc: Exception):
+    """پردازش خطای 500"""
+    return HTMLResponse(content=DEFAULT_HTML, status_code=500)
+
+if __name__ == "__main__":
+    # اجرای برنامه
+    uvicorn.run(
+        app,
+        host=Config.HOST,
+        port=Config.PORT,
+        log_level="info"
+    )