Update app.py

app.py

@@ -7,257 +7,171 @@ from pathlib import Path
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# --------------------------- التكوين الأساسي (The Core) ---------------------------
-# نستخدم Qwen لأنه الأقوى حالياً في الكود والمنطق المعقد
-MODEL_ID = os.environ.get('HF_MODEL_ID', 'Qwen/Qwen2.5-Coder-32B-Instruct')
+# --------------------------- الإعدادات ---------------------------
+# قائمة شاملة لضمان قبول كل ملفات المشروع دون أخطاء
+ALLOWED_EXTENSIONS = {
+    '.py', '.js', '.html', '.css', '.txt', '.json', '.md', 
+    '.php', '.xml', '.csv', '.zip', '.png', '.jpg', '.svg', '.ico'
+}
+MODEL_ID = os.environ.get('HF_MODEL_ID', 'moonshotai/Kimi-K2-Instruct')
 HF_TOKEN = os.environ.get('HF_TOKEN')
 
-class PowerTankIDE:
+class ProjectTank:
     def __init__(self):
-        # إنشاء بيئة عمل معزولة (Sandbox)
-        self.work_dir = tempfile.mkdtemp(prefix="tank_power_")
-        self.files_memory = {}  # الذاكرة الحية: {filename: content}
-        self.project_tree = ""  # الخريطة الذهنية للمشروع
+        self.work_dir = tempfile.mkdtemp(prefix="tank_dev_")
+        self.files_cache = {} # لتخزين محتوى الملفات: {الاسم: المحتوى}
 
-    def load_project(self, files):
-        """تحميل المشروع وتفكيكه بذكاء"""
-        if not files: return "⚠️ لا توجد ملفات.", []
+    def sync_files(self, file_objs):
+        """رفع الملفات وفكها وتجهيزها للمعالجة"""
+        if not file_objs: return "⚠️ لم يتم اختيار ملفات.", []
         
-        # تنظيف الساحة لبدء معركة جديدة
+        # تنظيف المجلد القديم
         if os.path.exists(self.work_dir):
             shutil.rmtree(self.work_dir)
-        self.work_dir = tempfile.mkdtemp(prefix="tank_power_")
-        
-        file_list = files if isinstance(files, list) else [files]
-        for f in file_list:
-            if f.name.lower().endswith('.zip'):
+        self.work_dir = tempfile.mkdtemp(prefix="tank_dev_")
+
+        files = file_objs if isinstance(file_objs, list) else [file_objs]
+        for f in files:
+            name = os.path.basename(f.name)
+            if name.lower().endswith('.zip'):
                 with zipfile.ZipFile(f.name, 'r') as z:
                     z.extractall(self.work_dir)
             else:
-                dest = os.path.join(self.work_dir, os.path.basename(f.name))
+                dest = os.path.join(self.work_dir, name)
                 shutil.copy(f.name, dest)
         
-        self.sync_system()
-        return f"✅ تم تحميل النظام: {len(self.files_memory)} ملفات جاهزة.", list(self.files_memory.keys())
-
-    def sync_system(self):
-        """مزامنة الواقع (القرص) مع الوعي (الذاكرة)"""
-        self.files_memory = {}
-        tree_lines = []
-        
-        # قراءة الملفات وبناء الهيكل
-        for path in Path(self.work_dir).rglob('*'):
-            if path.is_file() and not path.name.startswith('.'): # تجاهل ملفات النظام المخفية
-                rel_path = str(path.relative_to(self.work_dir)).replace("\\", "/")
-                
-                # قراءة الملفات النصية فقط للذاكرة
-                if path.suffix in ['.html', '.css', '.js', '.py', '.json', '.md', '.txt', '.xml']:
-                    try:
-                        content = path.read_text(encoding='utf-8', errors='ignore')
-                        self.files_memory[rel_path] = content
-                        tree_lines.append(f"- {rel_path}")
-                    except:
-                        tree_lines.append(f"- {rel_path} (Binary/Unreadable)")
-                else:
-                    tree_lines.append(f"- {rel_path} (Asset)")
-                    
-        self.project_tree = "\n".join(tree_lines)
-
-    def get_context(self, query):
-        """استدعاء الملفات ذات الصلة فقط (توفير الذاكرة للقوة العقلية)"""
-        context_str = f"### Project Structure ###\n{self.project_tree}\n"
-        
-        # خوارزمية بسيطة لتحديد الملفات المهمة بناءً على الطلب
-        # 1. دائماً نرفق ملفات الهيكل الأساسي
-        priority_files = ['index.html', 'styles.css', 'style.css', 'main.js', 'app.js']
-        
-        # 2. نرفق الملفات التي ذكرها المستخدم صراحة
-        query_lower = query.lower()
-        
-        attached_count = 0
-        for name, content in self.files_memory.items():
-            is_priority = any(p == name for p in priority_files)
-            is_requested = name.lower() in query_lower
-            
-            # نرفق الملف إذا كان أولوية أو تم طلبه، بشرط ألا نغرق الذاكرة
-            if (is_priority or is_requested) and attached_count < 5:
-                context_str += f"\n--- FILE: {name} ---\n{content}\n"
-                attached_count += 1
-                
-        return context_str
-
-    def force_update(self, filename, content):
-        """تحديث الملف بقوة على القرص والذاكرة"""
-        full_path = os.path.join(self.work_dir, filename)
-        os.makedirs(os.path.dirname(full_path), exist_ok=True)
-        with open(full_path, 'w', encoding='utf-8') as f:
+        self._reload_memory()
+        return f"🚀 تم تحميل {len(self.files_cache)} ملف في السيرفر المحلي.", list(self.files_cache.keys())
+
+    def _reload_memory(self):
+        self.files_cache = {}
+        for p in Path(self.work_dir).rglob('*'):
+            if p.is_file():
+                rel = str(p.relative_to(self.work_dir)).replace("\\", "/")
+                try:
+                    self.files_cache[rel] = p.read_text(encoding='utf-8', errors='ignore')
+                except: pass
+
+    def update_file_content(self, filename, content):
+        path = os.path.join(self.work_dir, filename)
+        os.makedirs(os.path.dirname(path), exist_ok=True)
+        with open(path, 'w', encoding='utf-8') as f:
             f.write(content)
-        self.files_memory[filename] = content
-
-    def compile_preview(self):
-        """المجمع (Compiler): يجمع الشتات ليصنع واجهة تعمل"""
-        # البحث عن نقطة الدخول
-        index = next((f for f in ["index.html", "main.html"] if f in self.files_memory), 
-                     next((f for f in self.files_memory if f.endswith('.html')), None))
+        self.files_cache[filename] = content
+
+    def build_preview(self):
+        """محرك المعاينة المحلي: حقن كل التبعيات في ملف واحد"""
+        # البحث عن نقطة الدخول (Entry Point)
+        entry = next((f for f in ["index.html", "main.html"] if f in self.files_cache), None)
+        if not entry:
+            entry = next((f for f in self.files_cache if f.endswith('.html')), None)
         
-        if not index:
-            return "<div style='display:flex;justify-content:center;align-items:center;height:100vh;color:#e74c3c;font-family:sans-serif;'><h2>⛔ خطأ: لا يوجد ملف HTML رئيسي (index.html)</h2></div>"
+        if not entry:
+            return "<div style='text-align:center; padding:50px;'>⚠️ لم يتم العثور على ملف HTML</div>"
 
-        html_content = self.files_memory[index]
-        
-        # حقن الـ CSS قسرياً في الرأس
-        css_blob = "\n".join([f"<style>/* {n} */\n{c}\n</style>" for n, c in self.files_memory.items() if n.endswith('.css')])
-        if "</head>" in html_content:
-            html_content = html_content.replace("</head>", f"{css_blob}\n</head>")
-        else:
-            html_content = f"<head>{css_blob}</head>" + html_content
+        html = self.files_cache[entry]
 
-        # حقن الـ JS قسرياً في النهاية
-        js_blob = "\n".join([f"<script>/* {n} */\n{c}\n</script>" for n, c in self.files_memory.items() if n.endswith('.js')])
-        if "</body>" in html_content:
-            html_content = html_content.replace("</body>", f"{js_blob}\n</body>")
-        else:
-            html_content += js_blob
+        # حقن CSS
+        for name, code in self.files_cache.items():
+            if name.endswith('.css'):
+                style = f"<style data-file='{name}'>\n{code}\n</style>"
+                html = re.sub(f'<link.*?href=["\'].*?{name}["\'].*?>', style, html)
 
-        return html_content
+        # حقن JS
+        for name, code in self.files_cache.items():
+            if name.endswith('.js'):
+                script = f"<script data-file='{name}'>\n{code}\n</script>"
+                html = re.sub(f'<script.*?src=["\'].*?{name}["\'].*?></script>', script, html)
 
-    def export_zip(self):
-        out = tempfile.mktemp()
-        shutil.make_archive(out, 'zip', self.work_dir)
-        return out + ".zip"
+        return html
 
-# تهيئة المحرك
-tank = PowerTankIDE()
+tank = ProjectTank()
 client = InferenceClient(api_key=HF_TOKEN) if HF_TOKEN else None
 
-# --------------------------- العقل المدبر (AI Logic) ---------------------------
+# --------------------------- منطق الذكاء الاصطناعي ---------------------------
 
-def engine_response(message, history):
+def clean_messages(history):
+    """إصلاح خطأ Data Incompatible - تنظيف مصفوفة الرسائل"""
+    return [{"role": m["role"], "content": str(m["content"])} for m in history if "role" in m and "content" in m]
+
+def chat_and_code(message, history):
     if not client:
-        yield history + [{"role": "assistant", "content": "⚠️ النظام متوقف: HF_TOKEN غير موجود."}]
+        yield history + [{"role": "assistant", "content": "⚠️ مفقود HF_TOKEN"}]
         return
 
-    context = tank.get_context(message)
-    system_prompt = (
-        "أنت 'Tank Engine'، بيئة تطوير قوية (IDE). هدفك: كتابة كود يعمل (Functional Code).\n"
-        "المخرجات: [FILE: filename] يليها ```code```. أعد كتابة الملف كاملاً."
+    # جمع السياق الكامل (وضع الدبابة)
+    context = "### CURRENT PROJECT STRUCTURE & CONTENT ###\n"
+    for name, content in tank.files_cache.items():
+        context += f"\nFILE: {name}\n{content}\n--- END ---\n"
+
+    system_instr = (
+        "You are an AI IDE. Provide short explanations then output the FULL modified files using this format:\n"
+        "[TARGET: filename]\n```\nUpdated Code\n```"
     )
 
-    messages = [{"role": "system", "content": system_prompt}]
-    for msg in history:
-        if isinstance(msg, dict): messages.append(msg)
-    messages.append({"role": "user", "content": f"{context}\n\nTask: {message}"})
+    clean_hist = clean_messages(history)
+    payload = [{"role": "system", "content": system_instr}]
+    payload.append({"role": "user", "content": f"{context}\n\nUser Request: {message}"})
 
-    full_response = ""
+    response = ""
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": ""})
 
-    try:
-        stream = client.chat_completion(model=MODEL_ID, messages=messages, max_tokens=8000, stream=True)
-        for chunk in stream:
-            # التحصين هنا: نتحقق من وجود 'choices' ومن أنها ليست فارغة
-            if hasattr(chunk, 'choices') and chunk.choices and len(chunk.choices) > 0:
-                token = chunk.choices[0].delta.content
-                if token:
-                    full_response += token
-                    history[-1]["content"] = full_response
-                    yield history
-            else:
-                # إذا وصلنا هنا، يعني أن هناك Chunk تقني لا يحتوي على نص
-                continue 
-    except Exception as e:
-        history[-1]["content"] = f"❌ انقطع الاتصال: {str(e)}"
-        yield history
-
-def apply_updates(history):
-    """تنفيذ التعديلات واستخراج الملفات"""
-    if not history: return "لا يوجد بيانات", None, None
+    stream = client.chat_completion(model=MODEL_ID, messages=payload, max_tokens=4000, stream=True)
+    for chunk in stream:
+        token = chunk.choices[0].delta.content
+        if token:
+            response += token
+            history[-1]["content"] = response
+            yield history
+
+def apply_and_preview(history):
+    """استخراج الكود وتحديث السيرفر المحلي"""
+    if not history: return "لا توجد بيانات", None
     
-    raw_text = history[-1]["content"]
-    # نمط استخراج قوي
-    pattern = r"\[FILE:\s*(.*?)\]\s*```.*?\n(.*?)```"
-    matches = re.findall(pattern, raw_text, re.DOTALL | re.IGNORECASE)
+    content = history[-1]["content"]
+    matches = re.findall(r"\[TARGET:\s*(.*?)\]\s*```.*?\n(.*?)```", content, re.DOTALL)
     
-    logs = []
-    if not matches:
-        return "⚠️ لم يتم العثور على كود قابل للتنفيذ.", tank.compile_preview(), list(tank.files_memory.keys())
-
     for filename, code in matches:
-        clean_name = filename.strip()
-        tank.force_update(clean_name, code.strip())
-        logs.append(f"✅ تم تحديث: {clean_name}")
+        tank.update_file_content(filename.strip(), code.strip())
     
-    tank.sync_system() # تحديث الخريطة
-    return "\n".join(logs), tank.compile_preview(), list(tank.files_memory.keys())
+    return "✅ تم التعديل والمعاينة بنجاح", tank.build_preview()
 
-# --------------------------- واجهة التحكم (Cockpit) ---------------------------
+# --------------------------- الواجهة ---------------------------
 
-with gr.Blocks(theme=gr.themes.Monochrome(), title="Tank IDE - Power Edition") as demo:
-    gr.Markdown("# 🛡️ Tank IDE <span style='font-size:0.6em; opacity:0.5'>Power Edition</span>")
-    
-    active_file = gr.State()
 
+
+with gr.Blocks(theme=gr.themes.Soft(), title="AI Tank IDE") as demo:
+    gr.Markdown("# 🛡️ AI Tank IDE Pro\n**سيرفر محلي + رؤية كاملة للملفات**")
+    
     with gr.Row():
-        # العمود الأيسر: التحكم والملفات
-        with gr.Column(scale=1, variant="panel"):
-            with gr.Group():
-                upload = gr.UploadButton("📂 تحميل المشروع (ZIP)", file_count="multiple", variant="secondary")
-                file_dropdown = gr.Dropdown(label="ملفات المشروع", interactive=True)
-                download = gr.DownloadButton("💾 تصدير المشروع (ZIP)")
-            
-            status = gr.Markdown("جاهز للعمل.")
+        with gr.Column(scale=1):
+            upload = gr.UploadButton("📁 ارفع مشروعك (ZIP/Files)", file_count="multiple")
+            file_list = gr.Dropdown(label="ملفات المشروع المكتشفة", choices=[])
+            status = gr.Markdown("🟢 جاهز للعمل")
             
-            # الشات - العقل المدبر
-            chatbot = gr.Chatbot(height=500, type="messages", show_label=False)
-            msg = gr.Textbox(placeholder="اكتب الأمر البرمجي هنا (مثلاً: أضف مكتبة Charts واعرض بيانات المبيعات)", show_label=False)
-            
-            with gr.Row():
-                btn_send = gr.Button("تنفيذ الأمر", variant="primary")
-                btn_apply = gr.Button("تطبيق التغييرات ⚡", variant="stop")
+            with gr.Group():
+                chat = gr.Chatbot(type="messages", height=400)
+                msg = gr.Textbox(placeholder="اطلب تعديل المشروع...", show_label=False)
+                with gr.Row():
+                    send = gr.Button("إرسال", variant="primary")
+                    apply = gr.Button("⚡ تطبيق المعاينة", variant="stop")
 
-        # العمود الأيمن: المعاينة والكود
         with gr.Column(scale=2):
             with gr.Tabs():
-                with gr.TabItem("🌐 المعاينة الحية"):
-                    preview = gr.HTML(label="المخرج النهائي", min_height=600)
-                
-                with gr.TabItem("💻 الكود المصدري"):
-                    editor = gr.Code(label="محرر الملفات", language="python", lines=30, interactive=True)
-                    btn_save_manual = gr.Button("حفظ تعديل يدوي", size="sm")
-
-    # --------------------------- التشبيك (Wiring) ---------------------------
-    
-    # 1. التحميل
-    def on_load(f):
-        log, files = tank.load_project(f)
-        return log, gr.update(choices=files), tank.compile_preview()
+                with gr.TabItem("🌐 المعاينة الفورية (Local Server)"):
+                    preview = gr.HTML(value="<div style='text-align:center; padding:50px;'>ارفع مشروعك لرؤية النتيجة</div>")
+                with gr.TabItem("📝 محرر الكود"):
+                    editor = gr.Code(label="كود الملف المختار", lines=25, interactive=True)
     
-    upload.upload(on_load, upload, [status, file_dropdown, preview])
-
-    # 2. الشات والتنفيذ
-    btn_send.click(engine_response, [msg, chatbot], chatbot).then(lambda: "", None, msg)
+    # ربط الأحداث
+    upload.upload(tank.sync_files, upload, [status, file_list]).then(tank.build_preview, None, preview)
     
-    def on_apply_click(h):
-        log, html, files = apply_updates(h)
-        return log, html, gr.update(choices=files)
+    send.click(chat_and_code, [msg, chat], chat).then(lambda: "", None, msg)
     
-    btn_apply.click(on_apply_click, chatbot, [status, preview, file_dropdown])
-
-    # 3. عرض الملفات وتعديلها يدوياً
-    def on_select_file(f):
-        return tank.files_memory.get(f, ""), f
-    
-    file_dropdown.change(on_select_file, file_dropdown, [editor, active_file])
-    
-    def on_manual_save(code, name):
-        if not name: return "⚠️ اختر ملفاً أولاً"
-        tank.force_update(name, code)
-        return f"✅ تم حفظ {name} يدوياً"
-        
-    btn_save_manual.click(on_manual_save, [editor, active_file], status).then(tank.compile_preview, None, preview)
+    apply.click(apply_and_preview, chat, [status, preview])
     
-    # 4. التصدير
-    download.click(tank.export_zip, None, download)
+    file_list.change(lambda n: tank.files_cache.get(n, ""), file_list, editor)
 
 if __name__ == "__main__":
     demo.queue().launch()