Update app.py

app.py

@@ -15,37 +15,73 @@ import fitz  # PyMuPDF
 from PIL import Image
 import io
 import requests
+from fastapi import FastAPI, Request, HTTPException
+from fastapi.responses import HTMLResponse
+import uvicorn
 
-# تكوين السجلات
+# ========== تكوين السجلات ==========
 logging.basicConfig(
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
     level=logging.INFO
 )
 logger = logging.getLogger(__name__)
 
-# التوكنات - سيتم تعيينها في متغيرات البيئة
+# ========== التوكنات ومتغيرات البيئة ==========
 TELEGRAM_BOT_TOKEN = os.environ.get('TELEGRAM_BOT_TOKEN')
 NVAPI_API_KEY = os.environ.get('NVAPI_API_KEY')
-WEBHOOK_URL = os.environ.get('WEBHOOK_URL', '')  # سيعينه Hugging Face تلقائياً
+WEBHOOK_URL = os.environ.get('SPACE_URL', '')
 
-# تكوين عميل NVIDIA
+# ========== تكوين عميل NVIDIA ==========
 nvidia_client = OpenAI(
     base_url="https://integrate.api.nvidia.com/v1",
     api_key=NVAPI_API_KEY
 )
 
-# مستودع Hugging Face للمواد
+# ========== مستودع Hugging Face للمواد ==========
 REPO_ID = "Riy777/Study"
 
-# حالات المحادثة
+# ========== حالات المحادثة ==========
 SELECTING_SUBJECT, SELECTING_ACTION, WAITING_FOR_QUESTION = range(3)
 
+# ========== تطبيق FastAPI ==========
+app = FastAPI(title="Medical Lab Bot", version="1.0.0")
+
+# ========== فئة البوت الرئيسية ==========
 class MedicalLabBot:
     def __init__(self):
         self.conversation_memory = {}
         self.available_materials = {}
         self.file_cache = {}
         self.load_all_materials()
+        self.application = None
+    
+    def initialize_application(self):
+        """تهيئة تطبيق التليجرام"""
+        if not self.application:
+            self.application = Application.builder().token(TELEGRAM_BOT_TOKEN).build()
+            self.setup_handlers()
+    
+    def setup_handlers(self):
+        """إعداد معالجات التليجرام"""
+        conv_handler = ConversationHandler(
+            entry_points=[CommandHandler('start', self.start)],
+            states={
+                SELECTING_SUBJECT: [
+                    CallbackQueryHandler(self.handle_subject_selection, pattern='^subject_|general_help|refresh_materials$')
+                ],
+                SELECTING_ACTION: [
+                    CallbackQueryHandler(self.handle_action_selection),
+                    CallbackQueryHandler(self.handle_subject_selection, pattern='^subject_')
+                ],
+                WAITING_FOR_QUESTION: [
+                    MessageHandler(filters.TEXT & ~filters.COMMAND, self.handle_message)
+                ]
+            },
+            fallbacks=[CommandHandler('start', self.start)]
+        )
+        
+        self.application.add_handler(conv_handler)
+        self.application.add_handler(CallbackQueryHandler(self.handle_callback_query, pattern='^more_questions|change_subject|back_to_actions$'))
     
     def load_all_materials(self):
         """تحميل جميع المواد والملفات من Hugging Face"""
@@ -755,37 +791,16 @@ class MedicalLabBot:
             )
             return SELECTING_ACTION
 
-# إنشاء كائن البوت
+# ========== إنشاء كائن البوت ==========
 bot = MedicalLabBot()
+bot.initialize_application()
 
-# إنشاء تطبيق التليجرام مع تكوين Webhook
-application = Application.builder().token(TELEGRAM_BOT_TOKEN).build()
-
-# إضافة handlers
-conv_handler = ConversationHandler(
-    entry_points=[CommandHandler('start', bot.start)],
-    states={
-        SELECTING_SUBJECT: [
-            CallbackQueryHandler(bot.handle_subject_selection, pattern='^subject_|general_help|refresh_materials$')
-        ],
-        SELECTING_ACTION: [
-            CallbackQueryHandler(bot.handle_action_selection),
-            CallbackQueryHandler(bot.handle_subject_selection, pattern='^subject_')
-        ],
-        WAITING_FOR_QUESTION: [
-            MessageHandler(filters.TEXT & ~filters.COMMAND, bot.handle_message)
-        ]
-    },
-    fallbacks=[CommandHandler('start', bot.start)]
-)
-
-application.add_handler(conv_handler)
-application.add_handler(CallbackQueryHandler(bot.handle_callback_query, pattern='^more_questions|change_subject|back_to_actions$'))
-
-async def setup_webhook():
-    """إعداد Webhook للبوت"""
+# ========== دوال FastAPI ==========
+@app.on_event("startup")
+async def on_startup():
+    """إعداد Webhook عند بدء التشغيل"""
     if WEBHOOK_URL:
-        await application.bot.set_webhook(
+        await bot.application.bot.set_webhook(
             url=f"{WEBHOOK_URL}/telegram",
             allowed_updates=Update.ALL_TYPES
         )
@@ -793,7 +808,59 @@ async def setup_webhook():
     else:
         logger.warning("WEBHOOK_URL not set, using polling instead")
 
-if __name__ == '__main__':
-    # عند التشغيل المحلي، استخدام Polling
-    logger.info("Starting bot in polling mode...")
-    application.run_polling()
+@app.get("/", response_class=HTMLResponse)
+async def root():
+    """الصفحة الرئيسية"""
+    return """
+    <html>
+        <head>
+            <title>Medical Lab Bot</title>
+            <style>
+                body { font-family: Arial, sans-serif; margin: 40px; }
+                .container { max-width: 800px; margin: 0 auto; }
+                .status { padding: 20px; background: #f0f8ff; border-radius: 10px; }
+            </style>
+        </head>
+        <body>
+            <div class="container">
+                <h1>🏥 بوت المختبرات الطبية الذكي</h1>
+                <div class="status">
+                    <h2>✅ البوت يعمل بنجاح</h2>
+                    <p>البوت متصل بـ Telegram وجاهز لاستقبال الرسائل.</p>
+                    <p>استخدم الرابط التالي للتواصل مع البوت:</p>
+                    <p><strong>https://t.me/your_bot_username</strong></p>
+                </div>
+                <h3>المميزات:</h3>
+                <ul>
+                    <li>📚 شرح المواد الدراسية من ملفات PDF وWord</li>
+                    <li>❓ توليد أسئلة متنوعة للمراجعة</li>
+                    <li>📖 تلخيص المحتوى الدراسي</li>
+                    <li>🧪 تفسير المفاهيم العلمية</li>
+                    <li>🎯 اختبار الفهم</li>
+                </ul>
+            </div>
+        </body>
+    </html>
+    """
+
+@app.post("/telegram")
+async def handle_telegram_update(request: Request):
+    """معالجة تحديثات Telegram"""
+    try:
+        update_data = await request.json()
+        update = Update.de_json(update_data, bot.application.bot)
+        await bot.application.process_update(update)
+        return {"status": "ok"}
+    except Exception as e:
+        logger.error(f"Error processing update: {e}")
+        raise HTTPException(status_code=400, detail="Invalid update")
+
+@app.get("/health")
+async def health_check():
+    """فحص صحة الخدمة"""
+    return {"status": "healthy", "service": "medical-lab-bot"}
+
+# ========== التشغيل الرئيسي ==========
+if __name__ == "__main__":
+    port = int(os.environ.get("PORT", 7860))
+    uvicorn.run(app, host="0.0.0.0", port=port)