Update app.py

app.py

@@ -4,8 +4,7 @@ import logging
 import asyncio
 import aiohttp
 from fastapi import FastAPI, HTTPException, BackgroundTasks
-from fastapi.responses import HTMLResponse
-from fastapi.staticfiles import StaticFiles
+from fastapi.responses import HTMLResponse, JSONResponse
 from pydantic import BaseModel, Field
 from typing import List, Dict, Optional, Any
 import torch
@@ -35,9 +34,6 @@ os.makedirs("/tmp/hf_cache", exist_ok=True)
 
 app = FastAPI(title="نظام الذكاء الاصطناعي المتقدم للتحكم في Windows")
 
-# Mount static files
-app.mount("/static", StaticFiles(directory="static"), name="static")
-
 # إعداد التسجيل
 logging.basicConfig(
     level=logging.INFO,
@@ -53,7 +49,7 @@ class WindowsTask(BaseModel):
     priority: str = "normal"
 
 class ClientCommand(BaseModel):
-    command_type: str  # execute, analyze, create_bot, install, monitor
+    command_type: str
     parameters: Dict[str, Any]
     client_id: Optional[str] = None
 
@@ -121,11 +117,14 @@ class WindowsAIAdvisor:
             
         except Exception as e:
             logger.error(f"فشل تحميل النماذج: {e}")
-            raise e
+            # لا نرفع الخطأ لنتمكن من تشغيل التطبيق بدون النموذج
 
     def generate_ai_response(self, prompt: str, context: List[Dict] = None) -> str:
         """توليد رد الذكاء الاصطناعي"""
         try:
+            if not self.is_loaded:
+                return "النظام قيد التهيئة. يرجى المحاولة لاحقاً."
+            
             system_context = """
             أنت مساعد ذكي متخصص في تحليل واختراق ألعاب Windows وبناء البوتات المتقدمة.
             خبرتك تشمل:
@@ -160,28 +159,37 @@ class WindowsAIAdvisor:
 
     def analyze_task(self, task_description: str) -> Dict:
         """تحليل المهمة وإنشاء خطة تنفيذ"""
-        analysis_prompt = f"""
-        قم بتحليل المهمة التالية وإنشاء خطة تنفيذ مفصلة للعميل على Windows:
-        {task_description}
-        
-        قدم خطة تنفيذ تشمل:
-        1. الأدوات المطلوبة
-        2. الخطوات التقنية
-        3. الأوامر المطلوبة
-        4. المخاطر المحتملة
-        5. البدائل
-        
-        ركز على الجانب العملي والتنفيذي.
-        """
-        
-        plan = self.generate_ai_response(analysis_prompt)
-        
-        return {
-            "plan": plan,
-            "steps": self.extract_execution_steps(plan),
-            "tools": self.extract_required_tools(plan),
-            "commands": self.generate_client_commands(plan, task_description)
-        }
+        try:
+            analysis_prompt = f"""
+            قم بتحليل المهمة التالية وإنشاء خطة تنفيذ مفصلة للعميل على Windows:
+            {task_description}
+            
+            قدم خطة تنفيذ تشمل:
+            1. الأدوات المطلوبة
+            2. الخطوات التقنية
+            3. الأوامر المطلوبة
+            4. المخاطر المحتملة
+            5. البدائل
+            
+            ركز على الجانب العملي والتنفيذي.
+            """
+            
+            plan = self.generate_ai_response(analysis_prompt)
+            
+            return {
+                "plan": plan,
+                "steps": self.extract_execution_steps(plan),
+                "tools": self.extract_required_tools(plan),
+                "commands": self.generate_client_commands(plan, task_description)
+            }
+        except Exception as e:
+            logger.error(f"فشل تحليل المهمة: {e}")
+            return {
+                "plan": f"خطأ في التحليل: {str(e)}",
+                "steps": [],
+                "tools": [],
+                "commands": []
+            }
 
     def extract_execution_steps(self, plan: str) -> List[str]:
         """استخراج خطوات التنفيذ من الخطة"""
@@ -192,11 +200,10 @@ class WindowsAIAdvisor:
             line = line.strip()
             if any(marker in line for marker in ['1.', '2.', '3.', '4.', '5.', '- ', '• ', 'خطوة']):
                 if len(line) > 10:
-                    # تنظيف الخطوة
                     clean_step = re.sub(r'^[0-9•\-\.,]+\s*', '', line)
                     steps.append(clean_step)
         
-        return steps[:15]  # إرجاع أول 15 خطوة فقط
+        return steps[:15]
 
     def extract_required_tools(self, plan: str) -> List[str]:
         """استخراج الأدوات المطلوبة"""
@@ -218,7 +225,6 @@ class WindowsAIAdvisor:
         """توليد أوامر للعميل"""
         commands = []
         
-        # تحليل نوع المهمة
         task_lower = task_description.lower()
         
         if any(word in task_lower for word in ['تحليل', 'analyze', 'فحص']):
@@ -246,15 +252,6 @@ class WindowsAIAdvisor:
                     "depth": "deep"
                 },
                 "description": "تحليل عمليات وذاكرة النظام"
-            },
-            {
-                "command_type": "analyze",
-                "parameters": {
-                    "task": task,
-                    "analysis_type": "game_files",
-                    "depth": "standard"
-                },
-                "description": "فحص ملفات اللعبة"
             }
         ]
 
@@ -300,14 +297,6 @@ class WindowsAIAdvisor:
                     "depth": "deep"
                 },
                 "description": "مسح الثغرات الأمنية"
-            },
-            {
-                "command_type": "execute",
-                "parameters": {
-                    "script_type": "exploit",
-                    "task": task
-                },
-                "description": "تنفيذ استغلال الثغرات"
             }
         ]
 
@@ -377,23 +366,105 @@ class WindowsAIAdvisor:
         }
         logger.info(f"تم تسجيل العميل: {client_id}")
 
-    def update_client_status(self, client_id: str, status: str):
-        """تحديث حالة العميل"""
-        if client_id in self.connected_clients:
-            self.connected_clients[client_id].update({
-                "last_seen": datetime.now(),
-                "status": status
-            })
-
 # تهيئة النظام
 ai_advisor = WindowsAIAdvisor()
 
+# واجهة HTML بسيطة مدمجة في الكود
+HTML_INTERFACE = """
+<!DOCTYPE html>
+<html lang="ar" dir="rtl">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>نظام الذكاء الاصطناعي للتحكم في Windows</title>
+    <style>
+        * { margin: 0; padding: 0; box-sizing: border-box; font-family: Arial, sans-serif; }
+        body { background: #1a1a2e; color: white; padding: 20px; }
+        .container { max-width: 1200px; margin: 0 auto; }
+        .header { text-align: center; margin-bottom: 30px; padding: 20px; background: #16213e; border-radius: 10px; }
+        .card { background: #0f3460; padding: 20px; margin: 15px 0; border-radius: 8px; }
+        textarea, input, button { width: 100%; padding: 10px; margin: 10px 0; border-radius: 5px; border: none; }
+        button { background: #e94560; color: white; cursor: pointer; font-weight: bold; }
+        button:hover { background: #c23616; }
+        .result { background: #2d4059; padding: 15px; border-radius: 5px; margin: 10px 0; white-space: pre-wrap; }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <div class="header">
+            <h1>🖥️ نظام الذكاء الاصطناعي للتحكم في Windows</h1>
+            <p>أدخل المهمة المطلوبة وسيتم توليد خطة تنفيذ مفصلة</p>
+        </div>
+        
+        <div class="card">
+            <h3>🎯 إدخال المهمة</h3>
+            <textarea id="taskInput" rows="5" placeholder="مثال: قم بتحليل لعبة Genshin Impact وابني بوتاً للتعدين التلقائي..."></textarea>
+            <select id="taskType">
+                <option value="analysis">تحليل لعبة</option>
+                <option value="bot_creation">بناء بوت</option>
+                <option value="hacking">تحليل اختراق</option>
+                <option value="software_install">تثبيت برامج</option>
+            </select>
+            <button onclick="analyzeTask()">🚀 تحليل المهمة</button>
+        </div>
+        
+        <div class="card">
+            <h3>📊 النتائج</h3>
+            <div id="results"></div>
+        </div>
+    </div>
+
+    <script>
+        async function analyzeTask() {
+            const task = document.getElementById('taskInput').value;
+            const taskType = document.getElementById('taskType').value;
+            
+            if (!task) {
+                alert('يرجى إدخال المهمة');
+                return;
+            }
+            
+            document.getElementById('results').innerHTML = 'جاري التحليل...';
+            
+            try {
+                const response = await fetch('/api/analyze-task', {
+                    method: 'POST',
+                    headers: { 'Content-Type': 'application/json' },
+                    body: JSON.stringify({
+                        task_description: task,
+                        task_type: taskType,
+                        priority: 'high'
+                    })
+                });
+                
+                const data = await response.json();
+                displayResults(data);
+            } catch (error) {
+                document.getElementById('results').innerHTML = 'خطأ في الاتصال: ' + error;
+            }
+        }
+        
+        function displayResults(data) {
+            let html = '';
+            if (data.success) {
+                html += `<div class="result"><strong>✅ الخطة:</strong><br>${data.data.execution_plan}</div>`;
+                html += `<div class="result"><strong>🛠️ الأدوات المطلوبة:</strong><br>${data.data.analysis.tools.join(', ')}</div>`;
+                html += `<div class="result"><strong>📋 خطوات التنفيذ:</strong><br>${data.data.analysis.steps.join('<br>')}</div>`;
+            } else {
+                html = `<div class="result">❌ خطأ: ${data.detail}</div>`;
+            }
+            document.getElementById('results').innerHTML = html;
+        }
+    </script>
+</body>
+</html>
+"""
+
 # نقاط النهاية
-@app.get("/", response_class=HTMLResponse)
+@app.get("/")
 async def read_root():
-    """خدمة واجهة HTML الرئيسية"""
-    with open("static/index.html", "r", encoding="utf-8") as f:
-        return HTMLResponse(content=f.read())
+    """الصفحة الرئيسية"""
+    return HTMLResponse(content=HTML_INTERFACE)
 
 @app.post("/api/analyze-task")
 async def analyze_windows_task(task: WindowsTask):
@@ -439,7 +510,7 @@ async def generate_bot_script(task: WindowsTask):
                 "bot_script": bot_code,
                 "language": "cpp",
                 "features": ai_advisor.extract_bot_features(task.task_description),
-                "compilation_instructions": self.generate_compilation_instructions(bot_code)
+                "compilation_instructions": generate_compilation_instructions(bot_code)
             }
         }
     except Exception as e:
@@ -467,7 +538,7 @@ async def generate_exploit_code(task: WindowsTask):
             "success": True,
             "data": {
                 "exploit_code": exploit_code,
-                "techniques_used": self.extract_exploit_techniques(exploit_code),
+                "techniques_used": extract_exploit_techniques(exploit_code),
                 "risk_level": "high",
                 "instructions": "يتطلب خبرة متقدمة في التنفيذ"
             }
@@ -493,25 +564,6 @@ async def register_client(client_info: Dict):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-@app.post("/api/client/command")
-async def send_client_command(command: ClientCommand):
-    """إرسال أمر للعميل"""
-    try:
-        # هنا يمكن حفظ الأمر في قاعدة بيانات pending commands
-        command_id = f"cmd_{int(time.time())}"
-        
-        return {
-            "success": True,
-            "data": {
-                "command_id": command_id,
-                "command": command,
-                "status": "sent_to_client",
-                "timestamp": datetime.now().isoformat()
-            }
-        }
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
-
 @app.get("/api/system/status")
 async def get_system_status():
     """الحصول على حالة النظام"""
@@ -525,7 +577,12 @@ async def get_system_status():
         }
     }
 
-def generate_compilation_instructions(self, code: str) -> str:
+@app.get("/health")
+async def health_check():
+    """فحص صحة النظام"""
+    return {"status": "healthy", "timestamp": datetime.now().isoformat()}
+
+def generate_compilation_instructions(code: str) -> str:
     """توليد تعليمات التجميع"""
     if "cpp" in code.lower() or "c++" in code.lower():
         return """
@@ -547,7 +604,7 @@ def generate_compilation_instructions(self, code: str) -> str:
     else:
         return "تعليمات التجميع: استخدم المترجم المناسب للغة المستخدمة"
 
-def extract_exploit_techniques(self, code: str) -> List[str]:
+def extract_exploit_techniques(code: str) -> List[str]:
     """استخراج تقنيات الاستغلال المستخدمة"""
     techniques = []
     code_lower = code.lower()