Update app.py

app.py

@@ -52,6 +52,23 @@ class ChatResponse(BaseModel):
     message: str
     actions: List[Dict[str, Any]] = []
 
+class ClientRegistration(BaseModel):
+    client_id: str
+    machine_name: str
+    os_version: str
+    timestamp: str
+    agent_type: str
+
+def update_client_activity(client_id: str, activity: str = "chat"):
+    """تحديث نشاط العميل"""
+    connected_clients[client_id] = {
+        "last_seen": datetime.now().isoformat(),
+        "status": "active",
+        "last_activity": activity,
+        "message_count": connected_clients.get(client_id, {}).get("message_count", 0) + 1
+    }
+    logger.info(f"📊 تم تحديث نشاط العميل {client_id}: {activity}")
+
 @app.get("/")
 async def root():
     return {
@@ -59,30 +76,58 @@ async def root():
         "service": "Windows AI Controller Server",
         "timestamp": datetime.now().isoformat(),
         "connected_clients": len(connected_clients),
-        "pending_commands": len(pending_commands)
+        "pending_commands": len(pending_commands),
+        "active_clients": list(connected_clients.keys())
     }
 
 @app.post("/api/chat")
 async def chat_with_ai(request: ChatRequest):
-    """نقطة النهاية للمحادثة مع الذكاء الاصطناعي - هذه النقطة المفقودة!"""
+    """نقطة النهاية للمحادثة مع الذكاء الاصطناعي"""
     try:
-        logger.info(f"طلب محادثة من العميل {request.client_id}: {request.message}")
+        logger.info(f"💬 طلب محادثة من العميل {request.client_id}: {request.message}")
+        
+        # تحديث نشاط العميل فوراً
+        update_client_activity(request.client_id, "chat")
         
         # محاكاة استجابة من نموذج الذكاء الاصطناعي الحقيقي
         thinking = f"🤔 **التفكير الاستراتيجي**: تحليل متعمق لطلب المستخدم '{request.message}'. تحديد الإجراءات المطلوبة بناءً على نوع الطلب والمتطلبات."
         
         if "لعبة" in request.message or "game" in request.message:
             response_msg = "🎮 **الوكيل الاستراتيجي يحلل طلب الألعاب**\n\nتم استلام طلبك المتعلق بالألعاب بنجاح! 🚀\n\nجاري:\n• إعداد أدوات الهندسة العكسية المتقدمة\n• تحضير محلل الذاكرة والعمليات\n• تجهيز مكتبات تعديل القيم\n\n🔍 **أخبرني باسم اللعبة المحددة والميزة التي تريد تعديلها**"
-            actions = [{"type": "game_analysis", "description": "تحليل هياكل اللعبة والذاكرة باستخدام الذكاء الاستراتيجي"}]
+            actions = [
+                {
+                    "type": "game_analysis", 
+                    "description": "تحليل هياكل اللعبة والذاكرة باستخدام الذكاء الاستراتيجي",
+                    "parameters": {"game_detected": True, "analysis_level": "advanced"}
+                }
+            ]
         elif "بوت" in request.message or "bot" in request.message:
             response_msg = "🤖 **تطوير البوتات - تحليل استراتيجي**\n\nممتاز! جاري:\n• تحضير بيئة التطوير المتقدمة\n• إعداد مكتبات الأتمتة\n• تحليل متطلبات الأداء\n\n💡 **ما نوع البوت الذي تريد تطويره وأي لعبة يستهدفها؟**"
-            actions = [{"type": "bot_development", "description": "إعداد بيئة تطوير البوت باستراتيجيات متقدمة"}]
-        elif "مرحبا" in request.message or "اهلا" in request.message:
+            actions = [
+                {
+                    "type": "bot_development", 
+                    "description": "إعداد بيئة تطوير البوت باستراتيجيات متقدمة",
+                    "parameters": {"development_phase": "requirements_analysis"}
+                }
+            ]
+        elif "مرحبا" in request.message or "اهلا" in request.message or "hello" in request.message:
             response_msg = "🧠 **الوكيل الاستراتيجي يرحب بك!**\n\n✅ **تم الاتصال بنجاح بالخادم المركزي**\n\n🔧 **الميزات المتاحة عبر الوكيل الاستراتيجي:**\n• تحليل ذاكرة متقدم للألعاب\n• هندسة عكسية باستخدام الذكاء الاصطناعي\n• تطوير برامج وبوتات مخصصة\n• حل مشاكل تقنية معقدة\n• بحث واستخراج معلومات متقدم\n\n🚀 **ما المهمة التي تريدني مساعدتك فيها؟**"
-            actions = [{"type": "strategic_welcome", "description": "تهيئة البيئة الاستراتيجية للعميل"}]
+            actions = [
+                {
+                    "type": "strategic_welcome", 
+                    "description": "تهيئة البيئة الاستراتيجية للعميل",
+                    "parameters": {"welcome_complete": True, "client_registered": True}
+                }
+            ]
         else:
             response_msg = f"🧠 **الوكيل الاستراتيجي يحلل طلبك**: '{request.message}'\n\n✅ **تم استلام طلبك بنجاح عبر الخادم المركزي**\n\n🔧 **جاري المعالجة باستخدام الذكاء الاصطناعي المتقدم...**\n\n💡 **الميزات المتاحة:**\n• تحليل متقدم للذاكرة والعمليات\n• هندسة عكسية متطورة للألعاب\n• تطوير برامج وبوتات مخصصة\n• حل مشاكل تقنية معقدة\n• بحث واستخراج معلومات متقدم"
-            actions = [{"type": "strategic_analysis", "description": "تحليل استراتيجي متقدم للمهمة المطلوبة"}]
+            actions = [
+                {
+                    "type": "strategic_analysis", 
+                    "description": "تحليل استراتيجي متقدم للمهمة المطلوبة",
+                    "parameters": {"analysis_type": "general", "complexity": "high"}
+                }
+            ]
         
         response = ChatResponse(
             thinking_process=thinking,
@@ -90,18 +135,36 @@ async def chat_with_ai(request: ChatRequest):
             actions=actions
         )
         
-        # تحديث حالة العميل
-        connected_clients[request.client_id] = {
+        logger.info(f"✅ تم إرسال رد استراتيجي للعميل {request.client_id} - الإجراءات: {len(actions)}")
+        return response
+        
+    except Exception as e:
+        logger.error(f"❌ خطأ في معالجة المحادثة: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.post("/api/register-client")
+async def register_client(registration: ClientRegistration):
+    """تسجيل عميل جديد"""
+    try:
+        connected_clients[registration.client_id] = {
+            "machine_name": registration.machine_name,
+            "os_version": registration.os_version,
+            "agent_type": registration.agent_type,
+            "first_seen": registration.timestamp,
             "last_seen": datetime.now().isoformat(),
             "status": "active",
-            "last_message": request.message
+            "message_count": 0
         }
         
-        logger.info(f"تم إرسال رد استراتيجي للعميل {request.client_id}")
-        return response
+        logger.info(f"🆕 عميل جديد مسجل: {registration.client_id} - {registration.machine_name}")
         
+        return {
+            "success": True, 
+            "message": "تم تسجيل العميل بنجاح",
+            "client_id": registration.client_id
+        }
     except Exception as e:
-        logger.error(f"خطأ في معالجة المحادثة: {e}")
+        logger.error(f"❌ خطأ في تسجيل العميل: {e}")
         raise HTTPException(status_code=500, detail=str(e))
 
 @app.post("/api/send-command")
@@ -119,7 +182,9 @@ async def send_command(command: WindowsCommand):
             "status": "pending"
         }
         
-        logger.info(f"تم استلام أمر من العميل {command.client_id}: {command.action}")
+        update_client_activity(command.client_id, "command")
+        
+        logger.info(f"📨 تم استلام أمر من العميل {command.client_id}: {command.action}")
         
         return {
             "success": True, 
@@ -127,7 +192,7 @@ async def send_command(command: WindowsCommand):
             "message": "تم استلام الأمر بنجاح"
         }
     except Exception as e:
-        logger.error(f"خطأ في استقبال الأمر: {e}")
+        logger.error(f"❌ خطأ في استقبال الأمر: {e}")
         raise HTTPException(status_code=500, detail=str(e))
 
 @app.get("/api/get-commands/{client_id}")
@@ -148,7 +213,7 @@ async def get_commands(client_id: str):
         
         return {"success": True, "commands": commands}
     except Exception as e:
-        logger.error(f"خطأ في إرسال الأوامر: {e}")
+        logger.error(f"❌ خطأ في إرسال الأوامر: {e}")
         raise HTTPException(status_code=500, detail=str(e))
 
 @app.post("/api/submit-result")
@@ -162,16 +227,13 @@ async def submit_result(result: ClientResult):
             "timestamp": datetime.now().isoformat()
         }
         
-        connected_clients[result.client_id] = {
-            "last_seen": datetime.now().isoformat(),
-            "status": "active"
-        }
+        update_client_activity(result.client_id, "result")
         
-        logger.info(f"تم استلام نتيجة من العميل {result.client_id}")
+        logger.info(f"📊 تم استلام نتيجة من العميل {result.client_id}")
         
         return {"success": True, "message": "تم استلام النتيجة بنجاح"}
     except Exception as e:
-        logger.error(f"خطأ في استقبال النتيجة: {e}")
+        logger.error(f"❌ خطأ في استقبال النتيجة: {e}")
         raise HTTPException(status_code=500, detail=str(e))
 
 @app.get("/api/client-status/{client_id}")
@@ -183,21 +245,41 @@ async def client_status(client_id: str):
         "client_id": client_id,
         "status": client_data.get("status", "offline"),
         "last_seen": client_data.get("last_seen"),
+        "message_count": client_data.get("message_count", 0),
         "timestamp": datetime.now().isoformat()
     }
 
 @app.get("/api/system-status")
 async def system_status():
     """حالة النظام العام"""
+    active_clients = []
+    for client_id, data in connected_clients.items():
+        if data.get("status") == "active":
+            active_clients.append({
+                "client_id": client_id,
+                "machine_name": data.get("machine_name"),
+                "last_seen": data.get("last_seen"),
+                "message_count": data.get("message_count", 0)
+            })
+    
     return {
         "success": True,
         "status": "running",
         "connected_clients": len(connected_clients),
+        "active_clients": active_clients,
         "pending_commands": len(pending_commands),
         "total_results": len(client_results),
         "server_time": datetime.now().isoformat()
     }
 
+@app.get("/api/debug-clients")
+async def debug_clients():
+    """تصحيح الأخطاء - عرض جميع العملاء"""
+    return {
+        "connected_clients": connected_clients,
+        "total": len(connected_clients)
+    }
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)