Update app.py

app.py

@@ -1,138 +1,216 @@
-""" Smart Dev Sandbox - تطبيق Gradio احترافي لتعديل ملفات ZIP / DOCX / PDF باستخدام نموذج AI ملف: app.py
-
-المتطلبات:
-
-gradio
-
-huggingface-hub
-
-pyzipper
-
-python-docx
-
-PyPDF2
-
-
-تثبيت سريع: python -m venv venv source venv/bin/activate   # أو venv\Scripts\activate على Windows pip install gradio huggingface-hub pyzipper python-docx PyPDF2
-
-متغيرات بيئة مطلوبة:
-
-HF_TOKEN : مفتاح الوصول إلى Hugging Face Inference API """
-
-
-import os import re import shutil import tempfile import traceback from pathlib import Path from typing import Dict, Tuple, Optional, List
-
-import pyzipper import gradio as gr from huggingface_hub import InferenceClient from docx import Document from PyPDF2 import PdfReader
-
-إعدادات عامة
-
-ALLOWED_EXTENSIONS = {'.py', '.js', '.html', '.css', '.txt', '.json', '.md', '.php', '.yml', '.yaml', '.docx', '.pdf'} MAX_FILE_SIZE_BYTES = 200 * 1024 MAX_TOTAL_FILES = 500 MAX_EXTRACTED_BYTES = 20 * 1024 * 1024 MODEL_ID = os.environ.get('HF_MODEL_ID', 'moonshotai/Kimi-K2-Instruct') HF_TOKEN = os.environ.get('HF_TOKEN')
-
-if not HF_TOKEN: print("تحذير: لم يتم ضبط HF_TOKEN. تأكد من تعيين HF_TOKEN لتشغيل نموذج Hugging Face.")
-
-صناديق الرمل
-
-class Sandbox: def init(self): self.dir: Optional[str] = None self.files_content: Dict[str, str] = {} self.original_file_path: Optional[str] = None self.password: Optional[str] = None
-
-def cleanup(self):
-    if self.dir and os.path.isdir(self.dir):
-        shutil.rmtree(self.dir, ignore_errors=True)
-    self.dir = None
-    self.files_content = {}
-    self.original_file_path = None
-    self.password = None
-
-def reset(self, file_path: str, password: Optional[str] = None) -> Tuple[Optional[str], str]:
-    self.cleanup()
-    if not file_path:
-        return None, "⚠️ يرجى رفع ملف ZIP / DOCX / PDF أولاً"
-    self.dir = tempfile.mkdtemp(prefix='sandbox_')
-    self.password = password
-    self.original_file_path = file_path
-    try:
-        ext = Path(file_path).suffix.lower()
-        if ext == '.zip':
-            with pyzipper.AESZipFile(file_path, 'r') as zf:
-                if password:
-                    zf.setpassword(password.encode('utf-8'))
-                for info in zf.infolist():
-                    safe_name = os.path.normpath(info.filename).lstrip(os.sep)
-                    out_path = os.path.join(self.dir, safe_name)
-                    os.makedirs(os.path.dirname(out_path), exist_ok=True)
-                    zf.extract(info, path=self.dir)
-        elif ext == '.docx':
-            doc = Document(file_path)
-            out_txt = os.path.join(self.dir, 'document.txt')
-            with open(out_txt, 'w', encoding='utf-8') as f:
-                for para in doc.paragraphs:
-                    f.write(para.text + '\n')
-        elif ext == '.pdf':
-            reader = PdfReader(file_path)
-            out_txt = os.path.join(self.dir, 'document.txt')
-            with open(out_txt, 'w', encoding='utf-8') as f:
-                for page in reader.pages:
-                    f.write(page.extract_text() + '\n')
-        else:
-            return None, "❌ صيغة الملف غير مدعومة"
-
-        # قراءة الملفات المسموح بها
+r"""
+Smart Dev Sandbox - تطبيق بسيط وآمن يدعم ZIP / DOCX / PDF مع واجهة Gradio
+متطلبات:
+  pip install gradio huggingface-hub pyzipper python-docx PyPDF2
+
+متغيرات بيئة:
+  HF_TOKEN (اختياري، لتكامل Hugging Face)
+  HF_MODEL_ID (اختياري، معرف النموذج)
+"""
+import os
+import re
+import shutil
+import tempfile
+import traceback
+from pathlib import Path
+from typing import Dict, Tuple, Optional, List
+
+import pyzipper
+import gradio as gr
+
+# استيراد مكتبات لتحليل docx/pdf
+try:
+    from docx import Document
+except Exception:
+    Document = None
+
+try:
+    from PyPDF2 import PdfReader
+except Exception:
+    PdfReader = None
+
+# إعدادات عامة
+ALLOWED_EXTENSIONS = {
+    '.py', '.js', '.html', '.css', '.txt', '.json',
+    '.md', '.php', '.yml', '.yaml', '.docx', '.pdf'
+}
+MODEL_ID = os.environ.get('HF_MODEL_ID', 'moonshotai/Kimi-K2-Instruct')
+HF_TOKEN = os.environ.get('HF_TOKEN')
+
+if not HF_TOKEN:
+    print("ملاحظة: HF_TOKEN غير معطى — سيعمل التطبيق لكن بدون اتصال بنموذج Hugging Face.")
+
+# ---------- Sandbox بسيط وآمن ----------
+class Sandbox:
+    def __init__(self):
+        self.dir: Optional[str] = None
+        self.files_content: Dict[str, str] = {}
+        self.original_file_path: Optional[str] = None
+
+    def cleanup(self):
+        if self.dir and os.path.isdir(self.dir):
+            shutil.rmtree(self.dir, ignore_errors=True)
+        self.dir = None
         self.files_content = {}
-        for path in Path(self.dir).rglob('*'):
-            if path.is_file():
-                self.files_content[str(path.relative_to(self.dir))] = path.read_text(encoding='utf-8', errors='ignore')
+        self.original_file_path = None
 
-        return file_path, f"✅ تم استخراج {len(self.files_content)} ملف/ملفات قابلة للتعديل"
-    except Exception as e:
+    def reset(self, file_path: str, password: Optional[str] = None) -> Tuple[Optional[str], str]:
+        """
+        يدعم: .zip (استخراج)، .docx (استخراج نص)، .pdf (استخراج نص).
+        يعيد (path, message) أو (None, error_message).
+        """
         self.cleanup()
-        return None, f"❌ استثناء: {e}"
-
-def get_context(self) -> str:
-    return '\n'.join([f'--- FILE: {name} ---\n{content}' for name, content in self.files_content.items()])
-
-def update_file(self, name: str, content: str) -> None:
-    if not self.dir:
-        raise RuntimeError("لا توجد بيئة عمل (sandbox) نشطة")
-    safe_name = os.path.normpath(name).lstrip(os.sep)
-    out_path = os.path.join(self.dir, safe_name)
-    os.makedirs(os.path.dirname(out_path), exist_ok=True)
-    with open(out_path, 'w', encoding='utf-8') as f:
-        f.write(content)
-    self.files_content[safe_name] = content
-
-def create_zip(self) -> Tuple[Optional[str], str]:
-    if not self.dir:
-        return None, "❌ لا توجد ملفات لحزمها"
-    out_fd, out_path = tempfile.mkstemp(prefix='modified_', suffix='.zip')
-    os.close(out_fd)
-    base_name = out_path[:-4]
-    shutil.make_archive(base_name, 'zip', self.dir)
-    return base_name + '.zip', "✅ تم إنشاء ZIP المعدل"
+        if not file_path:
+            return None, "⚠️ يرجى رفع ملف ZIP / DOCX / PDF أولاً."
+        self.dir = tempfile.mkdtemp(prefix='sandbox_')
+        self.original_file_path = file_path
+        try:
+            ext = Path(file_path).suffix.lower()
+            if ext == '.zip':
+                with pyzipper.AESZipFile(file_path, 'r') as zf:
+                    # استخراج آمن: منع zip-slip
+                    for info in zf.infolist():
+                        name = info.filename
+                        if name.startswith('__MACOSX') or name.strip() == '':
+                            continue
+                        # منع المسارات المطلقة أو ../../
+                        safe_name = os.path.normpath(name).lstrip(os.sep)
+                        dest_path = os.path.join(self.dir, safe_name)
+                        os.makedirs(os.path.dirname(dest_path), exist_ok=True)
+                        zf.extract(info, path=self.dir)
+            elif ext == '.docx':
+                if Document is None:
+                    raise RuntimeError("مكتبة python-docx غير مثبتة")
+                doc = Document(file_path)
+                out_txt = os.path.join(self.dir, 'document.txt')
+                os.makedirs(os.path.dirname(out_txt), exist_ok=True)
+                with open(out_txt, 'w', encoding='utf-8') as f:
+                    for para in doc.paragraphs:
+                        f.write(para.text + '\n')
+            elif ext == '.pdf':
+                if PdfReader is None:
+                    raise RuntimeError("مكتبة PyPDF2 غير مثبتة")
+                reader = PdfReader(file_path)
+                out_txt = os.path.join(self.dir, 'document.txt')
+                os.makedirs(os.path.dirname(out_txt), exist_ok=True)
+                with open(out_txt, 'w', encoding='utf-8') as f:
+                    for page in reader.pages:
+                        text = page.extract_text() or ''
+                        f.write(text + '\n')
+            else:
+                self.cleanup()
+                return None, "❌ صيغة الملف غير مدعومة. ادعم: .zip, .docx, .pdf"
+
+            # جمع المحتويات للعرض / للسياق
+            self.files_content = {}
+            for p in Path(self.dir).rglob('*'):
+                if p.is_file():
+                    rel = str(p.relative_to(self.dir))
+                    try:
+                        self.files_content[rel] = p.read_text(encoding='utf-8', errors='ignore')
+                    except Exception:
+                        self.files_content[rel] = "/* لا يمكن قراءة الملف */"
+
+            return file_path, f"✅ تم تجهيز {len(self.files_content)} ملف/ملفات داخل الـ sandbox."
+        except Exception as e:
+            self.cleanup()
+            return None, f"❌ استثناء أثناء المعالجة: {e}"
+
+    def get_context(self) -> str:
+        if not self.files_content:
+            return ""
+        return "\n".join(f"--- FILE: {name} ---\n{content}" for name, content in self.files_content.items())
+
+    def update_file(self, name: str, content: str) -> None:
+        if not self.dir:
+            raise RuntimeError("لا توجد بيئة عمل نشطة")
+        safe_name = os.path.normpath(name).lstrip(os.sep)
+        out_path = os.path.join(self.dir, safe_name)
+        os.makedirs(os.path.dirname(out_path), exist_ok=True)
+        with open(out_path, 'w', encoding='utf-8') as f:
+            f.write(content)
+        self.files_content[safe_name] = content
+
+    def create_zip(self) -> Tuple[Optional[str], str]:
+        if not self.dir:
+            return None, "❌ لا توجد ملفات لحزمها"
+        fd, temp_zip = tempfile.mkstemp(prefix='modified_', suffix='.zip')
+        os.close(fd)
+        base = temp_zip[:-4]
+        shutil.make_archive(base, 'zip', self.dir)
+        return base + '.zip', "✅ تم إنشاء ZIP المعدل"
 
 sandbox = Sandbox()
 
-واجهة Gradio
-
-with gr.Blocks(title="Smart Dev Sandbox — المبرمج الذكي PRO") as demo: gr.Markdown("# 🚀 المبرمج الذكي PRO — Sandbox احترافي لتعديل مشاريع") with gr.Row(): with gr.Column(scale=3): chatbot = gr.Chatbot(label="محادثة المساعد") msg = gr.Textbox(placeholder="اطلب تعديلًا", label="طلب التعديل") send_btn = gr.Button("إرسال") with gr.Column(scale=1): file_input = gr.File(label="اختر ملف المشروع (ZIP / DOCX / PDF)") pwd_input = gr.Textbox(label="كلمة مرور (اختياري)", type="password") analyze = gr.Button("🔍 تحليل الملفات") status = gr.Markdown("الحالة: جاهز") gr.HTML("<hr>") save_btn = gr.Button("✨ تطبيق التعديلات وإنشاء ZIP") download_output = gr.File(label="حمل الملف المعدل")
-
-def api_reset(file, password):
-    if file is None:
-        return None, "⚠️ ارفع ملف أولاً"
-    return sandbox.reset(file.name, password)
-
-def api_send_message(user_message, chat_history):
-    chat_history = chat_history or []
-    chat_history.append(("user", user_message))
-    sandbox_ctx = sandbox.get_context()
-    full_response = f"نموذج AI سيجيب هنا على: {user_message}"
-    chat_history.append(("assistant", full_response))
-    return chat_history
-
-def api_apply_and_package(chat_history):
-    out_zip, msg = sandbox.create_zip()
-    return out_zip, msg
-
-analyze.click(api_reset, inputs=[file_input, pwd_input], outputs=[file_input, status])
-send_btn.click(api_send_message, inputs=[msg, chatbot], outputs=[chatbot])
-save_btn.click(api_apply_and_package, inputs=[chatbot], outputs=[download_output, status])
-
-if name == "main": demo.launch()
+# ---------- واجهة Gradio ----------
+with gr.Blocks(title="Smart Dev Sandbox") as demo:
+    gr.Markdown("# Smart Dev Sandbox — دعم ZIP / DOCX / PDF")
+    with gr.Row():
+        with gr.Column(scale=3):
+            chatbot = gr.Chatbot(label="محادثة")
+            msg = gr.Textbox(placeholder="اطرح سؤالاً أو اطلب تعديلًا...", label="الرسالة")
+            send_btn = gr.Button("إرسال")
+        with gr.Column(scale=1):
+            file_input = gr.File(label="اختر ملف (ZIP / DOCX / PDF)", file_types=[".zip", ".docx", ".pdf"])
+            pwd_input = gr.Textbox(label="كلمة مرور ZIP (اختياري)", type="password")
+            analyze_btn = gr.Button("🔍 تجهيز الملف")
+            status_md = gr.Markdown("**الحالة:** جاهز")
+            gr.HTML("<hr>")
+            save_btn = gr.Button("✨ تطبيق التعديلات وإنشاء ZIP")
+            download_file = gr.File(label="تحميل الناتج")
+
+    # دوال الواجهة
+    def api_reset(uploaded_file, password):
+        if uploaded_file is None:
+            return None, "⚠️ ارفع ملفًا أولًا"
+        # gradio يعطي كائن له .name أو .temp_path
+        file_path = getattr(uploaded_file, "name", None) or getattr(uploaded_file, "temp_path", None) or uploaded_file
+        return sandbox.reset(file_path, password)
+
+    def api_send_message(user_message, chat_history):
+        chat_history = chat_history or []
+        # يسمح بالدردشة حتى بدون ملف مرفوع
+        chat_history.append(("user", user_message))
+        if sandbox.files_content:
+            ctx = sandbox.get_context()
+            assistant_reply = f"تم استلام طلب تعديل. (سياق المشروع مرفوع — {len(sandbox.files_content)} ملف)."  # placeholder
+        else:
+            assistant_reply = "يمكنني الإجابة على الأسئلة العامة الآن. لطلب تعديل ملفات، ارفع ZIP/DOCX/PDF ثم اضغط 'تجهيز الملف'."
+        chat_history.append(("assistant", assistant_reply))
+        return chat_history
+
+    def api_apply_and_package(chat_history):
+        # يأخذ آخر رد مساعد ويستخرج كتل التعديل بصيغة [ملف: path]```...```
+        if not sandbox.files_content:
+            return None, "⚠️ لا ملفات داخل الـ sandbox للتعديل."
+        if not chat_history:
+            return None, "⚠️ المحادثة فارغة."
+        # إيجاد آخر رد المساعد
+        last = None
+        for role, content in reversed(chat_history):
+            if role == 'assistant' and content and content.strip():
+                last = content
+                break
+        if not last:
+            return None, "⚠️ لا توجد تعديلات في المحادثة."
+        # استخراج كتل التعديل
+        pattern = re.compile(r"\[\s*(?:ملف|file)\s*:\s*([^\]]+?)\s*\]\s*```(?:[\\w\\-+]+)?\\n(.*?)\\n```", re.IGNORECASE | re.DOTALL)
+        matches = pattern.findall(last)
+        if not matches:
+            return None, "⚠️ لم يعثر على كتل تعديل بصيغة [ملف: ...]```...```"
+        applied = 0
+        for name, code in matches:
+            try:
+                sandbox.update_file(name.strip(), code)
+                applied += 1
+            except Exception as e:
+                print(f"فشل تطبيق {name}: {e}")
+        out_zip, msg = sandbox.create_zip()
+        return out_zip, f"✅ تم تطبيق {applied} ملفات. {msg}"
+
+    analyze_btn.click(api_reset, inputs=[file_input, pwd_input], outputs=[file_input, status_md])
+    send_btn.click(api_send_message, inputs=[msg, chatbot], outputs=[chatbot])
+    save_btn.click(api_apply_and_package, inputs=[chatbot], outputs=[download_file, status_md])
+
+if __name__ == "__main__":
+    demo.launch()