Update app.py

app.py

@@ -2,173 +2,276 @@ import os
 import re
 import shutil
 import tempfile
+import zipfile
 from pathlib import Path
-from typing import Dict, Tuple, Optional, List
+from typing import Dict, List, Optional
 
-import pyzipper
 import gradio as gr
 from huggingface_hub import InferenceClient
-from docx import Document
-import PyPDF2
 
-# --------------------------- إعدادات عامة ---------------------------
-ALLOWED_EXTENSIONS = {'.py', '.js', '.html', '.css', '.txt', '.json', '.md', '.php', '.yml', '.yaml', '.docx', '.pdf', '.zip'}
-MODEL_ID = os.environ.get('HF_MODEL_ID', 'moonshotai/Kimi-K2-Instruct')
-HF_TOKEN = os.environ.get('HF_TOKEN') 
+# --------------------------- إعدادات النظام ---------------------------
+# ندعم ملفات الكود النصية والصور الأساسية (للمعاينة المستقبلية)
+ALLOWED_EXTENSIONS = {'.py', '.js', '.html', '.css', '.txt', '.json', '.md', '.php', '.xml', '.csv'}
+MODEL_ID = os.environ.get('HF_MODEL_ID', 'moonshotai/Kimi-K2-Instruct') # نموذج قوي بذاكرة كبيرة
+HF_TOKEN = os.environ.get('HF_TOKEN')
 
-class Sandbox:
+class ProjectManager:
     def __init__(self):
-        self.dir = None
-        self.files_content = {}
-
-    def cleanup(self):
-        if self.dir and os.path.exists(self.dir):
-            try: shutil.rmtree(self.dir)
-            except: pass
-        self.dir = None
-        self.files_content = {}
-
-    def process_files(self, file_objs, password: Optional[str] = None):
-        if not file_objs:
-            return "⚠️ لم يتم اختيار ملفات.", []
+        self.work_dir = tempfile.mkdtemp(prefix="ai_ide_")
+        self.files: Dict[str, str] = {} # تخزين المسار النسبي والمحتوى
+        self.active_file = None
+
+    def load_files(self, file_objs):
+        """تحميل الملفات أو فك ضغط ZIP مع الحفاظ على الهيكلية"""
+        if not file_objs: return "⚠️ لم يتم اختيار ملفات.", []
         
-        self.cleanup()
-        self.dir = tempfile.mkdtemp(prefix="dev_sandbox_")
+        # تنظيف القديم
+        self.files = {}
+        if os.path.exists(self.work_dir):
+            shutil.rmtree(self.work_dir)
+        self.work_dir = tempfile.mkdtemp(prefix="ai_ide_")
+
+        file_list_objs = file_objs if isinstance(file_objs, list) else [file_objs]
         
-        # التأكد من التعامل مع قائمة ملفات
-        files = file_objs if isinstance(file_objs, list) else [file_objs]
+        for f_obj in file_list_objs:
+            file_path = f_obj.name
+            filename = os.path.basename(file_path)
 
-        for f in files:
-            original_path = f.name
-            filename = os.path.basename(original_path)
-            
             if filename.lower().endswith('.zip'):
                 try:
-                    with pyzipper.AESZipFile(original_path, 'r') as zf:
-                        if password: zf.setpassword(password.encode('utf-8'))
-                        zf.extractall(path=self.dir)
-                except Exception as e: return f"❌ خطأ في ZIP: {str(e)}", []
+                    with zipfile.ZipFile(file_path, 'r') as zf:
+                        zf.extractall(self.work_dir)
+                except Exception as e:
+                    return f"❌ خطأ في فك الضغط: {str(e)}", []
             else:
-                dest = os.path.join(self.dir, filename)
-                shutil.copy(original_path, dest)
-        
-        self.files_content = self.read_allowed_files(self.dir)
-        # إرجاع الحالة وقائمة بأسماء الملفات للمستكشف
-        file_list = list(self.files_content.keys())
-        return f"✅ تم تحميل {len(file_list)} ملف بنجاح.", file_list
-
-    def read_allowed_files(self, base_dir: str) -> Dict[str, str]:
-        content_map = {}
-        for path in Path(base_dir).rglob('*'):
+                shutil.copy(file_path, os.path.join(self.work_dir, filename))
+
+        # قراءة الملفات من القرص إلى الذاكرة
+        self.refresh_file_memory()
+        return f"✅ تم تحميل المشروع ({len(self.files)} ملف).", list(self.files.keys())
+
+    def refresh_file_memory(self):
+        """قراءة الملفات فعلياً من المجلد المؤقت"""
+        self.files = {}
+        base_path = Path(self.work_dir)
+        for path in base_path.rglob('*'):
             if path.is_file() and path.suffix.lower() in ALLOWED_EXTENSIONS:
-                rel = str(path.relative_to(base_dir))
+                rel_path = str(path.relative_to(base_path)).replace("\\", "/")
                 try:
-                    ext = path.suffix.lower()
-                    if ext == '.docx':
-                        doc = Document(path)
-                        text = "\n".join([p.text for p in doc.paragraphs])
-                    elif ext == '.pdf':
-                        with open(path, 'rb') as pdf_f:
-                            reader = PyPDF2.PdfReader(pdf_f)
-                            text = "\n".join([page.extract_text() or "" for page in reader.pages])
-                    else:
-                        text = path.read_text(encoding='utf-8', errors='ignore')
-                    content_map[rel] = text
-                except: content_map[rel] = "/* خطأ في القراءة */"
-        return content_map
-
-    def update_file(self, name: str, content: str):
-        if not self.dir: return
-        full_path = os.path.join(self.dir, name)
+                    content = path.read_text(encoding='utf-8', errors='ignore')
+                    self.files[rel_path] = content
+                except:
+                    pass
+
+    def get_file_content(self, filename):
+        return self.files.get(filename, "")
+
+    def update_file(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_content[name] = content
+        self.files[filename] = content
+        return f"تم تحديث {filename}"
+
+    def generate_preview(self):
+        """
+        🔥 المحرك السحري للمعاينة 🔥
+        يقوم بدمج ملفات CSS و JS داخل ملف HTML ليعمل كملف واحد.
+        """
+        index_file = None
+        # البحث عن ملف البداية
+        candidates = ["index.html", "main.html", "home.html"]
+        for c in candidates:
+            if c in self.files:
+                index_file = c
+                break
+        
+        if not index_file:
+            # البحث عن أي ملف html
+            for f in self.files:
+                if f.endswith(".html"):
+                    index_file = f
+                    break
+        
+        if not index_file:
+            return "<div style='padding:20px; text-align:center'>⚠️ لم يتم العثور على ملف HTML للمعاينة.</div>"
+
+        html_content = self.files[index_file]
+
+        # 1. حقن CSS (بحث عن <link rel="stylesheet">)
+        def replace_css(match):
+            href = match.group(1)
+            # محاولة إيجاد الملف بناء على المسار النسبي
+            if href in self.files:
+                return f"<style>\n/* Injected: {href} */\n{self.files[href]}\n</style>"
+            return match.group(0) # إبقاءه كما هو إذا لم يوجد الملف
+
+        html_content = re.sub(r'<link[^>]+href=["\'](.*?)["\'][^>]*>', replace_css, html_content)
+
+        # 2. حقن JS (بحث عن <script src="...">)
+        def replace_js(match):
+            src = match.group(1)
+            if src in self.files:
+                return f"<script>\n/* Injected: {src} */\n{self.files[src]}\n</script>"
+            return match.group(0)
 
-    def create_zip(self):
-        if not self.dir: return None
-        out_base = tempfile.mktemp()
-        shutil.make_archive(out_base, 'zip', self.dir)
-        return out_base + ".zip"
+        html_content = re.sub(r'<script[^>]+src=["\'](.*?)["\'][^>]*>\s*</script>', replace_js, html_content)
 
-sandbox = Sandbox()
+        return html_content
+
+    def export_zip(self):
+        if not self.work_dir: return None
+        out_file = tempfile.mktemp()
+        shutil.make_archive(out_file, 'zip', self.work_dir)
+        return out_file + ".zip"
+
+# تهيئة المدير والعميل
+pm = ProjectManager()
 client = InferenceClient(api_key=HF_TOKEN) if HF_TOKEN else None
 
-# --------------------------- منطق المعالجة ---------------------------
+# --------------------------- منطق الذكاء الاصطناعي ---------------------------
 
-def handle_ai_chat(history):
+def ai_chat_handler(message, history):
     if not client:
-        history.append({"role": "assistant", "content": "⚠️ HF_TOKEN غير متوفر."})
-        yield history
-        return
+        return "⚠️ يجب وضع HF_TOKEN في متغيرات البيئة."
 
-    context = "الملفات المرفوعة:\n"
-    for name, content in sandbox.files_content.items():
-        context += f"\n[ملف: {name}]\n{content[:1000]}\n"
+    # تجميع السياق الكامل (بلا حدود أحرف - الدبابة!)
+    context_str = "الملفات الحالية في المشروع:\n"
+    for name, content in pm.files.items():
+        context_str += f"\n--- FILE: {name} ---\n{content}\n"
 
     system_prompt = (
-        "أنت مساعد برمجي. لإجراء تعديلات، استخدم التنسيق التالي حصراً:\n"
-        "[ملف: اسم_الملف]\n"
-        "```\nمحتوى الملف المحدث بالكامل\n```"
+        "أنت بيئة تطوير متكاملة (AI IDE). "
+        "مهمتك: تعديل الكود وتطويره بناء على الطلب. "
+        "القواعد الصارمة:\n"
+        "1. اقرأ كل الملفات بعناية.\n"
+        "2. عند تعديل ملف، يجب كتابة الكود كاملاً داخل البلوك.\n"
+        "3. استخدم الصيغة التالية للإخراج:\n"
+        "[TARGET: filename.ext]\n```\n... code ...\n```"
     )
-    
-    messages = [{"role": "system", "content": system_prompt}, {"role": "user", "content": f"{context}\n\nطلب المستخدم: {history[-1]['content']}"}]
-    
-    full_res = ""
-    history.append({"role": "assistant", "content": ""})
+
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": context_str + f"\n\nالمطلوب: {message}"}
+    ]
+
+    response_text = ""
     try:
-        for chunk in client.chat_completion(model=MODEL_ID, messages=messages, max_tokens=2500, stream=True):
-            if chunk.choices:
-                token = chunk.choices[0].delta.content
-                if token:
-                    full_res += token
-                    history[-1]["content"] = full_res
-                    yield history
+        # استخدام stream للحصول على استجابة سريعة
+        stream = client.chat_completion(model=MODEL_ID, messages=messages, max_tokens=4000, stream=True)
+        for chunk in stream:
+            if chunk.choices and chunk.choices[0].delta.content:
+                response_text += chunk.choices[0].delta.content
+                yield response_text
     except Exception as e:
-        history[-1]["content"] = f"❌ خطأ اتصال: {str(e)}"
-        yield history
+        yield f"❌ خطأ: {str(e)}"
 
-def execute_changes(history):
-    if not history: return None, "الدردشة فارغة."
-    text = history[-1]["content"]
-    matches = re.findall(r"\[\s*(?:ملف|file)\s*:\s*(.*?)\s*\]\s*```.*?\n(.*?)```", text, re.DOTALL | re.IGNORECASE)
+def apply_ai_changes(response_text):
+    """تحليل رد الذكاء الاصطناعي وتنفيذ التغييرات"""
+    pattern = r"\[TARGET:\s*(.*?)\]\s*```.*?\n(.*?)```"
+    matches = re.findall(pattern, response_text, re.DOTALL | re.IGNORECASE)
     
-    if not matches: return None, "لم يتم العثور على تعديلات بصيغة [ملف: ...]"
+    log = []
+    for filename, code in matches:
+        filename = filename.strip()
+        pm.update_file(filename, code.strip())
+        log.append(f"✅ تم تحديث {filename}")
     
-    for name, code in matches:
-        sandbox.update_file(name.strip(), code.strip())
+    if not log:
+        return "⚠️ لم يتم العثور على كود للتنفيذ. تأكد أن الرد يحتوي على [TARGET: file]...", pm.generate_preview()
     
-    return sandbox.create_zip(), f"✅ تم تحديث {len(matches)} ملفات."
+    return "\n".join(log), pm.generate_preview()
+
+# --------------------------- واجهة المستخدم (Gradio Blocks) ---------------------------
 
-# --------------------------- الواجهة ---------------------------
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🚀 Sandbox المحرر الذكي \n ارفع ملفاتك، اطلب التعديل، وحمل النتيجة.")
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="slate"), title="AI IDE Pro") as demo:
     
+    # حالة التطبيق (State)
+    selected_file_state = gr.State()
+
     with gr.Row():
-        with gr.Column(scale=2):
-            chatbot = gr.Chatbot(type="messages", height=500)
-            msg_in = gr.Textbox(placeholder="اكتب طلبك هنا (مثلاً: عدل README.md ليصبح بالعربية)...", show_label=False)
-            send_btn = gr.Button("إرسال", variant="primary")
-        
-        with gr.Column(scale=1):
-            # استخدام UploadButton بدلاً من File لضمان التوافق مع الجوال
-            upload_btn = gr.UploadButton("📁 اختر ملفات أو ZIP", file_types=list(ALLOWED_EXTENSIONS), file_count="multiple")
-            pwd_in = gr.Textbox(label="كلمة سر ZIP", type="password")
-            status_md = gr.Markdown("🟢 بانتظار الملفات...")
+        gr.Markdown("## 🛠️ AI IDE Pro: بيئة تطوير كاملة")
+    
+    with gr.Row(equal_height=True):
+        # العمود الأيسر: الأدوات والملفات
+        with gr.Column(scale=1, variant="panel"):
+            upload_btn = gr.UploadButton("📂 فتح مشروع (ZIP/Files)", file_count="multiple", file_types=list(ALLOWED_EXTENSIONS))
+            file_list = gr.Dropdown(label="ملفات المشروع", interactive=True)
+            status_bar = gr.Markdown("⚪ جاهز...")
             
-            with gr.Accordion("📄 مستكشف الملفات المرفوعة", open=False):
-                file_explorer = gr.JSON(value=[], label="الملفات النشطة في Sandbox")
+            gr.Markdown("### 🤖 المبرمج الذكي")
+            chatbot = gr.Chatbot(height=300, type="messages")
+            msg_input = gr.Textbox(placeholder="اطلب تعديلاً (مثلاً: غير لون الخلفية للأحمر)...", lines=2)
+            send_btn = gr.Button("إرسال الطلب", variant="primary")
+            apply_btn = gr.Button("⚡ تنفيذ التعديلات", variant="stop")
+
+        # العمود الأيمن: المحرر والمعاينة
+        with gr.Column(scale=2):
+            with gr.Tabs():
+                with gr.TabItem("📝 المحرر (Code Editor)"):
+                    code_editor = gr.Code(label="Source Code", language="html", interactive=True, lines=20)
+                    save_manual_btn = gr.Button("حفظ التعديلات اليدوية", size="sm")
+                
+                with gr.TabItem("🌐 معاينة حية (Live Preview)"):
+                    preview_html = gr.HTML(label="Web Preview")
+                    refresh_preview_btn = gr.Button("🔄 تحديث المعاينة")
             
-            apply_btn = gr.Button("✨ تنفيذ التعديلات وحفظ", variant="stop")
-            download_file = gr.File(label="تحميل المشروع النهائي")
+            download_btn = gr.File(label="تصدير المشروع النهائي")
 
-    # ربط العمليات
-    upload_btn.upload(sandbox.process_files, [upload_btn, pwd_in], [status_md, file_explorer])
+    # ------------------ ربط الأحداث (Event Wiring) ------------------
+
+    # 1. رفع الملفات
+    def on_upload(files):
+        msg, list_files = pm.load_files(files)
+        preview = pm.generate_preview()
+        return msg, gr.update(choices=list_files, value=list_files[0] if list_files else None), preview
     
-    send_btn.click(lambda m, h: ("", h + [{"role": "user", "content": m}]), [msg_in, chatbot], [msg_in, chatbot]).then(
-        handle_ai_chat, [chatbot], [chatbot]
+    upload_btn.upload(on_upload, upload_btn, [status_bar, file_list, preview_html])
+
+    # 2. اختيار ملف من القائمة لعرضه في المحرر
+    def on_file_select(filename):
+        content = pm.get_file_content(filename)
+        lang = "html"
+        if filename.endswith(".js"): lang = "javascript"
+        elif filename.endswith(".py"): lang = "python"
+        elif filename.endswith(".css"): lang = "css"
+        return content, lang, filename
+
+    file_list.change(on_file_select, file_list, [code_editor, code_editor, selected_file_state])
+
+    # 3. حفظ يدوي من المحرر
+    def on_manual_save(code, filename):
+        if not filename: return "⚠️ لم يتم اختيار ملف.", pm.generate_preview()
+        msg = pm.update_file(filename, code)
+        return msg, pm.generate_preview()
+
+    save_manual_btn.click(on_manual_save, [code_editor, selected_file_state], [status_bar, preview_html])
+
+    # 4. الشات والذكاء الاصطناعي
+    msg_input.submit(lambda m, h: ("", h + [{"role": "user", "content": m}]), [msg_input, chatbot], [msg_input, chatbot]).then(
+        ai_chat_handler, [msg_input, chatbot], chatbot
     )
-    
-    apply_btn.click(execute_changes, [chatbot], [download_file, status_md])
+    send_btn.click(lambda m, h: ("", h + [{"role": "user", "content": m}]), [msg_input, chatbot], [msg_input, chatbot]).then(
+        ai_chat_handler, [msg_input, chatbot], chatbot
+    )
+
+    # 5. تنفيذ تعديلات الذكاء الاصطناعي
+    def execute_ai(history):
+        if not history: return "لا يوجد محادثة", None, None
+        last_response = history[-1]['content']
+        log, preview = apply_ai_changes(last_response)
+        # تحديث قائمة الملفات لأن الـ AI قد ينشئ ملفات جديدة
+        new_file_list = list(pm.files.keys())
+        zip_path = pm.export_zip()
+        return log, preview, gr.update(choices=new_file_list), zip_path
+
+    apply_btn.click(execute_ai, [chatbot], [status_bar, preview_html, file_list, download_btn])
+
+    # 6. تحديث المعاينة يدوياً
+    refresh_preview_btn.click(lambda: pm.generate_preview(), None, preview_html)
 
-demo.launch()
+# تشغيل التطبيق
+if __name__ == "__main__":
+    demo.queue().launch()