Update app.py

app.py

@@ -13,13 +13,9 @@ import PyPDF2
 
 # --------------------------- إعدادات عامة ---------------------------
 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') 
 
-# --------------------------- منطق Sandbox ---------------------------
 class Sandbox:
     def __init__(self):
         self.dir = None
@@ -27,46 +23,48 @@ class Sandbox:
 
     def cleanup(self):
         if self.dir and os.path.exists(self.dir):
-            try:
-                shutil.rmtree(self.dir)
+            try: shutil.rmtree(self.dir)
             except: pass
         self.dir = None
         self.files_content = {}
 
-    def reset(self, zip_file, password: Optional[str] = None):
-        if zip_file is None:
-            return "⚠️ يرجى رفع ملف ZIP أولاً"
+    def process_input_files(self, file_objects, password: Optional[str] = None):
+        if not file_objects:
+            return "⚠️ لم يتم اختيار أي ملفات."
         
         self.cleanup()
         self.dir = tempfile.mkdtemp(prefix="dev_sandbox_")
         
-        try:
-            # استخدام getattr للتعامل مع كائنات Gradio المختلفة
-            zip_path = zip_file.name if hasattr(zip_file, 'name') else zip_file
-            
-            with pyzipper.AESZipFile(zip_path, 'r') as zf:
-                if password:
-                    zf.setpassword(password.encode('utf-8'))
-                
-                for info in zf.infolist():
-                    if info.filename.startswith('__MACOSX') or info.is_dir():
-                        continue
-                    dest_path = os.path.normpath(os.path.join(self.dir, info.filename))
-                    if not dest_path.startswith(os.path.normpath(self.dir) + os.sep):
-                        continue 
-                    zf.extract(info, path=self.dir)
-            
-            self.files_content = self.read_allowed_files(self.dir)
-            return f"✅ تم تحميل {len(self.files_content)} ملف بنجاح"
-        except Exception as e:
-            return f"❌ خطأ أثناء الاستخراج: {str(e)}"
+        files_count = 0
+        # تحويل المدخل إلى قائمة دائماً
+        if not isinstance(file_objects, list):
+            file_objects = [file_objects]
+
+        for f_obj in file_objects:
+            path = f_obj.name
+            # إذا كان ملف ZIP
+            if path.lower().endswith('.zip'):
+                try:
+                    with pyzipper.AESZipFile(path, 'r') as zf:
+                        if password: zf.setpassword(password.encode('utf-8'))
+                        for info in zf.infolist():
+                            if info.filename.startswith('__MACOSX') or info.is_dir(): continue
+                            zf.extract(info, path=self.dir)
+                            files_count += 1
+                except Exception as e: return f"❌ خطأ في ZIP: {str(e)}"
+            else:
+                # ملفات عادية (DOCX, PDF, TXT...)
+                dest = os.path.join(self.dir, os.path.basename(path))
+                shutil.copy(path, dest)
+                files_count += 1
+        
+        self.files_content = self.read_allowed_files(self.dir)
+        return f"✅ تم تحميل {len(self.files_content)} ملف (بما في ذلك محتويات ZIP)."
 
     def read_allowed_files(self, base_dir: str) -> Dict[str, str]:
         files = {}
-        count = 0
         for path in Path(base_dir).rglob('*'):
             if path.is_file() and path.suffix.lower() in ALLOWED_EXTENSIONS:
-                if count >= MAX_TOTAL_FILES: break
                 rel = str(path.relative_to(base_dir))
                 try:
                     ext = path.suffix.lower()
@@ -79,9 +77,7 @@ class Sandbox:
                             content = "\n".join([page.extract_text() or "" for page in reader.pages])
                     else:
                         content = path.read_text(encoding='utf-8', errors='ignore')
-                    
                     files[rel] = content
-                    count += 1
                 except: files[rel] = "/* تعذر قراءة الملف */"
         return files
 
@@ -102,92 +98,79 @@ class Sandbox:
 sandbox = Sandbox()
 client = InferenceClient(api_key=HF_TOKEN) if HF_TOKEN else None
 
-# --------------------------- دوال الدردشة والمعالجة ---------------------------
-
-def handle_user_msg(message, history):
-    return "", history + [{"role": "user", "content": message}]
+# --------------------------- منطق الدردشة ---------------------------
 
 def handle_assistant_res(history):
     if not HF_TOKEN or not client:
-        history.append({"role": "assistant", "content": "⚠️ خطأ: HF_TOKEN غ��ر مضبوط."})
+        history.append({"role": "assistant", "content": "⚠️ توكن HF_TOKEN غير مضبوط."})
         yield history
         return
 
-    context = "سياق المشروع:\n"
+    context = "الملفات المتاحة حالياً:\n"
     for name, content in sandbox.files_content.items():
-        context += f"\n--- FILE: {name} ---\n{content[:1500]}\n"
+        context += f"\n[اسم الملف: {name}]\n{content[:1200]}\n"
 
     system_instr = (
-        "أنت مساعد برمجي خبير. عند طلب تعديل، أجب دائماً بالتنسيق التالي:\n"
-        "[ملف: اسم_الملف_هنا]\n"
-        "```language\nالمحتوى المحدث\n```"
+        "أنت مساعد برمجي. عندما يطلب المستخدم تعديل أو إنشاء ملف، يجب أن تلتزم بالتنسيق التالي حرفياً:\n"
+        "[ملف: اسم_الملف_كاملاً]\n"
+        "```\n"
+        "محتوى الملف هنا\n"
+        "```\n"
+        "لا تضف أي نص توضيحي داخل كتلة الكود، واجعل الرد يبدأ بـ [ملف: ...] لكل تعديل."
     )
     
     user_query = history[-1]["content"]
-    messages = [
-        {"role": "system", "content": system_instr},
-        {"role": "user", "content": f"{context}\n\nطلب المستخدم: {user_query}"}
-    ]
+    messages = [{"role": "system", "content": system_instr}, {"role": "user", "content": f"{context}\n\nالطلب: {user_query}"}]
 
     response = ""
     history.append({"role": "assistant", "content": ""})
     
     try:
-        # تصحيح منطق استلام الـ Chunks لتجنب خطأ Index out of range
-        for chunk in client.chat_completion(model=MODEL_ID, messages=messages, max_tokens=2048, stream=True):
-            if hasattr(chunk, 'choices') and len(chunk.choices) > 0:
+        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:
                     response += token
                     history[-1]["content"] = response
                     yield history
     except Exception as e:
-        history[-1]["content"] += f"\n\n❌ خطأ أثناء البث: {str(e)}"
+        history[-1]["content"] += f"\n❌ خطأ: {str(e)}"
         yield history
 
-def run_apply_and_zip(history):
-    if not history or history[-1]["role"] != "assistant":
-        return None, "⚠️ لا يوجد رد من الوكيل لتطبيقه."
-    
+def apply_changes(history):
+    if not history: return None, "الدردشة فارغة"
     text = history[-1]["content"]
+    # Regex محسن لالتقاط النمط المطلوب بدقة
     pattern = r"\[\s*(?:ملف|file)\s*:\s*(.*?)\s*\]\s*```.*?\n(.*?)```"
     matches = re.findall(pattern, text, re.DOTALL | re.IGNORECASE)
     
-    if not matches:
-        return None, "⚠️ لم أجد أكواداً بتنسيق [ملف: ...]"
+    if not matches: return None, "⚠️ لم أجد ملفات للتعديل بتنسيق [ملف: ...]"
     
     for name, code in matches:
         sandbox.update_file(name.strip(), code.strip())
     
     return sandbox.create_zip(), f"✅ تم تحديث {len(matches)} ملف."
 
-# --------------------------- واجهة Gradio ---------------------------
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🚀 Smart Dev Sandbox PRO")
+# --------------------------- الواجهة ---------------------------
+with gr.Blocks(theme=gr.themes.Default()) as demo:
+    gr.Markdown("# 🚀 المحرر الذكي (دعم الملفات و ZIP)")
     
     with gr.Row():
         with gr.Column(scale=2):
-            chatbot = gr.Chatbot(type="messages", label="الدردشة البرمجية", height=450)
-            msg_input = gr.Textbox(placeholder="اطلب تعديلاً على الملفات المرفوعة...", show_label=False)
-            send_btn = gr.Button("إرسال الطلب ⚡", variant="primary")
+            chatbot = gr.Chatbot(type="messages", height=500)
+            msg_input = gr.Textbox(placeholder="مثال: أنشئ ملف README.md ترحيبي", show_label=False)
+            send_btn = gr.Button("إرسال", variant="primary")
         
         with gr.Column(scale=1):
-            file_zip = gr.File(label="1. ارفع ملف ZIP")
-            pwd = gr.Textbox(label="كلمة السر (اختياري)", type="password")
-            load_btn = gr.Button("تحليل الملفات")
-            status = gr.Markdown("🟢 **الحالة:** جاهز")
-            apply_btn = gr.Button("2. تطبيق وحفظ ZIP", variant="stop")
-            download = gr.File(label="تحميل المشروع")
-
-    # الأحداث
-    msg_input.submit(handle_user_msg, [msg_input, chatbot], [msg_input, chatbot]).then(
-        handle_assistant_res, [chatbot], [chatbot]
-    )
-    send_btn.click(handle_user_msg, [msg_input, chatbot], [msg_input, chatbot]).then(
-        handle_assistant_res, [chatbot], [chatbot]
-    )
-    load_btn.click(sandbox.reset, [file_zip, pwd], [status])
-    apply_btn.click(run_apply_and_zip, [chatbot], [download, status])
-
-if __name__ == "__main__":
-    demo.launch()
+            file_input = gr.File(label="ارفع ZIP أو عدة ملفات", file_count="multiple")
+            pwd = gr.Textbox(label="كلمة السر (لـ ZIP)", type="password")
+            load_btn = gr.Button("تحليل الملفات المرفوعة")
+            status = gr.Markdown("🟢 جاهز")
+            apply_btn = gr.Button("حفظ وتحميل المشروع المعدل", variant="stop")
+            download = gr.File(label="المشروع النهائي")
+
+    load_btn.click(sandbox.process_input_files, [file_input, pwd], [status])
+    send_btn.click(lambda m, h: ("", h + [{"role": "user", "content": m}]), [msg_input, chatbot], [msg_input, chatbot]).then(handle_assistant_res, [chatbot], [chatbot])
+    apply_btn.click(apply_changes, [chatbot], [download, status])
+
+demo.launch()