Update app.py

app.py

@@ -7,28 +7,26 @@ from pathlib import Path
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# --------------------------- الإعدادات الأساسية ---------------------------
-# قائمة واسعة لضمان عدم حدوث Crash عند الرفع
-ALLOWED_EXTENSIONS = {
-    '.py', '.js', '.html', '.css', '.txt', '.json', '.md', 
-    '.php', '.xml', '.csv', '.zip', '.png', '.jpg', '.svg'
-}
-MODEL_ID = os.environ.get('HF_MODEL_ID', 'deepseek-ai/DeepSeek-V3.2')
+# --------------------------- التكوين الأساسي (The Core) ---------------------------
+# نستخدم Qwen لأنه الأقوى حالياً في الكود والمنطق المعقد
+MODEL_ID = os.environ.get('HF_MODEL_ID', 'Qwen/Qwen2.5-Coder-32B-Instruct')
 HF_TOKEN = os.environ.get('HF_TOKEN')
 
-class TankProjectManager:
+class PowerTankIDE:
     def __init__(self):
-        self.work_dir = tempfile.mkdtemp(prefix="tank_ide_")
-        self.files_memory = {} # اسم الملف: المحتوى
+        # إنشاء بيئة عمل معزولة (Sandbox)
+        self.work_dir = tempfile.mkdtemp(prefix="tank_power_")
+        self.files_memory = {}  # الذاكرة الحية: {filename: content}
+        self.project_tree = ""  # الخريطة الذهنية للمشروع
 
     def load_project(self, files):
-        """تحميل الملفات واستخراجها مع الحفاظ على الهيكل"""
-        if not files: return "⚠️ لم يتم رفع أي ملفات.", []
+        """تحميل المشروع وتفكيكه بذكاء"""
+        if not files: return "⚠️ لا توجد ملفات.", []
         
-        # تنظيف المجلد قبل التحميل الجديد
+        # تنظيف الساحة لبدء معركة جديدة
         if os.path.exists(self.work_dir):
             shutil.rmtree(self.work_dir)
-        self.work_dir = tempfile.mkdtemp(prefix="tank_ide_")
+        self.work_dir = tempfile.mkdtemp(prefix="tank_power_")
         
         file_list = files if isinstance(files, list) else [files]
         for f in file_list:
@@ -39,103 +37,134 @@ class TankProjectManager:
                 dest = os.path.join(self.work_dir, os.path.basename(f.name))
                 shutil.copy(f.name, dest)
         
-        self.refresh_memory()
-        return f"✅ تم تحميل {len(self.files_memory)} ملف بنجاح.", list(self.files_memory.keys())
+        self.sync_system()
+        return f"✅ تم تحميل النظام: {len(self.files_memory)} ملفات جاهزة.", list(self.files_memory.keys())
 
-    def refresh_memory(self):
-        """قراءة كل حرف في المشروع (وضع الدبابة)"""
+    def sync_system(self):
+        """مزامنة الواقع (القرص) مع الوعي (الذاكرة)"""
         self.files_memory = {}
+        tree_lines = []
+        
+        # قراءة الملفات وبناء الهيكل
         for path in Path(self.work_dir).rglob('*'):
-            if path.is_file():
+            if path.is_file() and not path.name.startswith('.'): # تجاهل ملفات النظام المخفية
                 rel_path = str(path.relative_to(self.work_dir)).replace("\\", "/")
-                try:
-                    # قراءة كاملة للملف
-                    content = path.read_text(encoding='utf-8', errors='ignore')
-                    self.files_memory[rel_path] = content
-                except: pass
-
-    def update_file(self, filename, content):
-        """حفظ التعديلات على القرص وفي الذاكرة"""
+                
+                # قراءة الملفات النصية فقط للذاكرة
+                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:
             f.write(content)
         self.files_memory[filename] = content
 
-    def get_preview_html(self):
-        """محرك حقن التبعيات للمعاينة السريعة"""
-        index_file = None
-        for name in ["index.html", "main.html"]:
-            if name in self.files_memory:
-                index_file = name
-                break
-        if not index_file:
-            # البحث عن أول ملف html متاح
-            html_files = [f for f in self.files_memory if f.endswith('.html')]
-            if not html_files: return "<div style='text-align:center;padding:20px;'>⚠️ لا يوجد ملف HTML للمعاينة</div>"
-            index_file = html_files[0]
-
-        content = self.files_memory[index_file]
-        
-        # حقن الـ CSS
-        for name, css_code in self.files_memory.items():
-            if name.endswith('.css'):
-                style_tag = f"<style>/* {name} */\n{css_code}\n</style>"
-                content = re.sub(f'<link.*?href=["\'].*?{name}["\'].*?>', style_tag, 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))
         
-        # حقن الـ JS
-        for name, js_code in self.files_memory.items():
-            if name.endswith('.js'):
-                script_tag = f"<script>/* {name} */\n{js_code}\n</script>"
-                content = re.sub(f'<script.*?src=["\'].*?{name}["\'].*?></script>', script_tag, content)
+        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>"
+
+        html_content = self.files_memory[index]
         
-        return content
+        # حقن الـ 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
+
+        # حقن الـ 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
+
+        return html_content
 
     def export_zip(self):
         out = tempfile.mktemp()
         shutil.make_archive(out, 'zip', self.work_dir)
         return out + ".zip"
 
-# تهيئة الأدوات
-pm = TankProjectManager()
+# تهيئة المحرك
+tank = PowerTankIDE()
 client = InferenceClient(api_key=HF_TOKEN) if HF_TOKEN else None
 
-# --------------------------- منطق الذكاء الاصطناعي ---------------------------
-
-def clean_history(history):
-    """إصلاح خطأ Data incompatible عبر تنظيف القائمة"""
-    cleaned = []
-    for msg in history:
-        if isinstance(msg, dict) and "role" in msg and "content" in msg:
-            cleaned.append({"role": msg["role"], "content": str(msg["content"])})
-    return cleaned
+# --------------------------- العقل المدبر (AI Logic) ---------------------------
 
-def ai_handler(message, history):
+def engine_response(message, history):
     if not client:
-        yield history + [{"role": "assistant", "content": "⚠️ HF_TOKEN مفقود!"}]
+        yield history + [{"role": "assistant", "content": "⚠️ النظام متوقف: HF_TOKEN غير موجود."}]
         return
 
-    # بناء السياق الكامل للمشروع
-    context = "### هيكل المشروع الحالي ومحتوى الملفات بالكامل ###\n"
-    for name, content in pm.files_memory.items():
-        context += f"\n--- FILE: {name} ---\n{content}\n--- END {name} ---\n"
-
+    # 1. التجهيز
+    context = tank.get_context(message)
+    
+    # 2. التوجيه الصارم (The Strict System Prompt)
     system_prompt = (
-        "أنت مبرمج خبير (AI IDE). لقد أرسلت لك المشروع كاملاً. "
-        "عند طلب تعديلات، أجب بالشرح ثم ضع الكود المحدث لكل ملف بالصيغة التالية:\n"
-        "[FILE: filename.ext]\n```\nالكود المحدث هنا\n```"
+        "أنت 'Tank Engine'، بيئة تطوير قوية (IDE). هدفك: كتابة كود يعمل (Functional Code) وليس مجرد نصوص.\n"
+        "القواعد الصارمة:\n"
+        "1. المخرجات: يجب أن تكون بصيغة [FILE: filename] يليها بلوك الكود.\n"
+        "2. التبعيات: لا يوجد NPM. إذا احتجت مكتبة (React, Vue, Tailwind, Chart.js)، استخدم رابط CDN في كود HTML فوراً.\n"
+        "3. الشمولية: عند تعديل ملف، أعد كتابته كاملاً. لا تستخدم عبارات مثل '...بقية الكود...'.\n"
+        "4. التصميم: استخدم Tailwind CSS عبر CDN إذا طلب منك تصميم عصري، إلا إذا طلب المستخدم CSS مخصص.\n"
+        "5. لا تعتذر ولا تشرح كثيراً. اكتب الكود."
     )
 
     messages = [{"role": "system", "content": system_prompt}]
-    messages.extend(clean_history(history))
-    messages.append({"role": "user", "content": f"{context}\n\nالمطلوب الآن: {message}"})
+    # تنظيف التاريخ لتجنب الأخطاء
+    for msg in history:
+        if isinstance(msg, dict): messages.append(msg)
+        
+    messages.append({"role": "user", "content": f"{context}\n\nTask: {message}"})
 
+    # 3. التنفيذ
     full_response = ""
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": ""})
 
     try:
-        stream = client.chat_completion(model=MODEL_ID, messages=messages, max_tokens=4000, stream=True)
+        stream = client.chat_completion(model=MODEL_ID, messages=messages, max_tokens=8000, stream=True)
         for chunk in stream:
             token = chunk.choices[0].delta.content
             if token:
@@ -143,98 +172,98 @@ def ai_handler(message, history):
                 history[-1]["content"] = full_response
                 yield history
     except Exception as e:
-        history[-1]["content"] = f"❌ خطأ: {str(e)}"
+        history[-1]["content"] = f"❌ فشل المعالجة: {str(e)}"
         yield history
 
-def apply_ai_updates(history):
-    """استخراج الكود من الشات وتحديث الـ Sandbox"""
-    if not history: return "لا توجد بيانات", None, None
+def apply_updates(history):
+    """تنفيذ التعديلات واستخراج الملفات"""
+    if not history: return "لا يوجد بيانات", None, None
     
-    text = history[-1]["content"]
-    # البحث عن نمط الملفات [FILE: name] ```code```
+    raw_text = history[-1]["content"]
+    # نمط استخراج قوي
     pattern = r"\[FILE:\s*(.*?)\]\s*```.*?\n(.*?)```"
-    matches = re.findall(pattern, text, re.DOTALL | re.IGNORECASE)
+    matches = re.findall(pattern, raw_text, re.DOTALL | re.IGNORECASE)
     
     logs = []
+    if not matches:
+        return "⚠️ لم يتم العثور على كود قابل للتنفيذ.", tank.compile_preview(), list(tank.files_memory.keys())
+
     for filename, code in matches:
-        filename = filename.strip()
-        pm.update_file(filename, code.strip())
-        logs.append(f"✅ تم تحديث {filename}")
+        clean_name = filename.strip()
+        tank.force_update(clean_name, code.strip())
+        logs.append(f"✅ تم تحديث: {clean_name}")
     
-    if not logs:
-        return "⚠️ لم يتم العثور على كود بصيغة [FILE: name] لتنفيذه.", pm.get_preview_html(), list(pm.files_memory.keys())
-    
-    return "\n".join(logs), pm.get_preview_html(), list(pm.files_memory.keys())
+    tank.sync_system() # تحديث الخريطة
+    return "\n".join(logs), tank.compile_preview(), list(tank.files_memory.keys())
 
-# --------------------------- واجهة المستخدم ---------------------------
+# --------------------------- واجهة التحكم (Cockpit) ---------------------------
 
-with gr.Blocks(theme=gr.themes.Soft(), title="AI Tank IDE") as demo:
-    gr.Markdown("# 🛡️ AI Tank IDE Pro\n**نظام برمجة متكامل: رؤية كاملة للملفات + معاينة فورية**")
+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_filename = gr.State()
+    active_file = gr.State()
 
     with gr.Row():
-        # الجانب الأيسر: التحكم والملفات
+        # العمود الأيسر: التحكم والملفات
         with gr.Column(scale=1, variant="panel"):
-            upload_btn = gr.UploadButton("📂 رفع المشروع (ZIP/ملفات)", file_count="multiple")
-            file_tree = gr.Dropdown(label="مستكشف الملفات", choices=[])
-            status_log = gr.Markdown("🟢 جاهز")
+            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.Accordion("🤖 مساعد البرمجة", open=True):
-                chatbot = gr.Chatbot(type="messages", height=350)
-                msg_input = gr.Textbox(placeholder="اطلب تعديلاً كبيراً على المشروع...", show_label=False)
-                with gr.Row():
-                    send_btn = gr.Button("إرسال", variant="primary")
-                    apply_btn = gr.Button("⚡ تنفيذ التعديلات", variant="stop")
-
-        # الجانب الأيمن: المحرر والمعاينة
+            # الشات - العقل المدبر
+            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.Column(scale=2):
             with gr.Tabs():
-                with gr.TabItem("🌐 المعاينة الحية (Preview)"):
-                    preview_area = gr.HTML(value="<div style='text-align:center;'>بانتظار رفع المشروع...</div>")
-                    refresh_btn = gr.Button("🔄 تحديث المعاينة")
+                with gr.TabItem("🌐 المعاينة الحية"):
+                    preview = gr.HTML(label="المخرج النهائي", min_height=600)
                 
-                with gr.TabItem("📝 محرر الكود"):
-                    code_editor = gr.Code(label="Source Code", lines=25, interactive=True)
-                    save_manual = gr.Button("💾 حفظ التعديل اليدوي", size="sm")
+                with gr.TabItem("💻 الكود المصدري"):
+                    editor = gr.Code(label="محرر الملفات", language="python", lines=30, interactive=True)
+                    btn_save_manual = gr.Button("حفظ تعديل يدوي", size="sm")
 
-            download_file = gr.File(label="تحميل المشروع النهائي")
-
-    # --- ربط الأحداث ---
-
-    # 1. الرفع
-    def on_upload(files):
-        msg, files_list = pm.load_project(files)
-        preview = pm.get_preview_html()
-        return msg, gr.update(choices=files_list), preview
+    # --------------------------- التشبيك (Wiring) ---------------------------
     
-    upload_btn.upload(on_upload, upload_btn, [status_log, file_tree, preview_area])
-
-    # 2. الشات
-    send_btn.click(ai_handler, [msg_input, chatbot], chatbot).then(lambda: "", None, msg_input)
-
-    # 3. تنفيذ تعديلات AI
-    def on_apply(h):
-        log, preview, files_list = apply_ai_updates(h)
-        zip_path = pm.export_zip()
-        return log, preview, gr.update(choices=files_list), zip_path
+    # 1. التحميل
+    def on_load(f):
+        log, files = tank.load_project(f)
+        return log, gr.update(choices=files), tank.compile_preview()
     
-    apply_btn.click(on_apply, chatbot, [status_log, preview_area, file_tree, download_file])
+    upload.upload(on_load, upload, [status, file_dropdown, preview])
 
-    # 4. التبديل بين الملفات في المحرر
-    def load_to_editor(name):
-        return pm.files_memory.get(name, ""), name
+    # 2. الشات والتنفيذ
+    btn_send.click(engine_response, [msg, chatbot], chatbot).then(lambda: "", None, msg)
     
-    file_tree.change(load_to_editor, file_tree, [code_editor, active_filename])
+    def on_apply_click(h):
+        log, html, files = apply_updates(h)
+        return log, html, gr.update(choices=files)
+    
+    btn_apply.click(on_apply_click, chatbot, [status, preview, file_dropdown])
 
-    # 5. الحفظ اليدوي
+    # 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 "⚠️ اختر ملفاً أولاً"
-        pm.update_file(name, code)
+        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)
     
-    save_manual.click(on_manual_save, [code_editor, active_filename], status_log)
-    refresh_btn.click(pm.get_preview_html, None, preview_area)
+    # 4. التصدير
+    download.click(tank.export_zip, None, download)
 
 if __name__ == "__main__":
     demo.queue().launch()