Update app.py

app.py

@@ -17,7 +17,7 @@ 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') # تأكد من ضبط هذا المتغير
+HF_TOKEN = os.environ.get('HF_TOKEN') 
 
 # --------------------------- منطق Sandbox ---------------------------
 class Sandbox:
@@ -39,20 +39,21 @@ class Sandbox:
         
         self.cleanup()
         self.dir = tempfile.mkdtemp(prefix="dev_sandbox_")
-        zip_path = zip_file.name
         
         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 # تخطي الملفات المشبوهة
+                        continue 
                     zf.extract(info, path=self.dir)
             
             self.files_content = self.read_allowed_files(self.dir)
@@ -94,9 +95,9 @@ class Sandbox:
 
     def create_zip(self):
         if not self.dir: return None
-        out_path = tempfile.mktemp()
-        shutil.make_archive(out_path, 'zip', self.dir)
-        return out_path + ".zip"
+        temp_zip_base = tempfile.mktemp()
+        shutil.make_archive(temp_zip_base, 'zip', self.dir)
+        return temp_zip_base + ".zip"
 
 sandbox = Sandbox()
 client = InferenceClient(api_key=HF_TOKEN) if HF_TOKEN else None
@@ -104,33 +105,24 @@ 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"
-    if not sandbox.files_content:
-        context += "(لا توجد ملفات مرفوعة حالياً)\n"
-    else:
-        for name, content in sandbox.files_content.items():
-            context += f"\n--- FILE: {name} ---\n{content[:2000]}\n" # نرسل أول 2000 حرف لتوفير التوكنز
+    context = "سياق المشروع:\n"
+    for name, content in sandbox.files_content.items():
+        context += f"\n--- FILE: {name} ---\n{content[:1500]}\n"
 
-    # إعداد الرسائل للنموذج
     system_instr = (
-        "أنت مساعد برمجي خبير. عند طلب تعديل، أجب دائماً بالتنسيق التالي لكل ملف:\n"
+        "أنت مساعد برمجي خبير. عند طلب تعديل، أجب دائماً بالتنسيق التالي:\n"
         "[ملف: اسم_الملف_هنا]\n"
-        "```language\n"
-        "المحتوى الكامل للملف بعد التعديل\n"
-        "```"
+        "```language\nالمحتوى المحدث\n```"
     )
     
-    # استخراج آخر طلب للمستخدم
     user_query = history[-1]["content"]
     messages = [
         {"role": "system", "content": system_instr},
@@ -140,85 +132,62 @@ def handle_assistant_res(history):
     response = ""
     history.append({"role": "assistant", "content": ""})
     
-        try:
+    try:
+        # تصحيح منطق استلام الـ Chunks لتجنب خطأ Index out of range
         for chunk in client.chat_completion(model=MODEL_ID, messages=messages, max_tokens=2048, stream=True):
-            # التأكد من أن القائمة ليست فارغة وأن هناك محتوى في الاختيار الأول
-            if chunk.choices and len(chunk.choices) > 0:
+            if hasattr(chunk, 'choices') and len(chunk.choices) > 0:
                 token = chunk.choices[0].delta.content
                 if token:
                     response += token
                     history[-1]["content"] = response
                     yield history
     except Exception as e:
-        # إذا انقطع الات��ال بعد البدء، نحتفظ بما تم كتابته ونضيف الخطأ
-        error_msg = f"\n\n❌ انقطع الاتصال: {str(e)}"
-        history[-1]["content"] += error_msg
+        history[-1]["content"] += f"\n\n❌ خطأ أثناء البث: {str(e)}"
         yield history
 
-
 def run_apply_and_zip(history):
-    if not history: return None, "لا توجد محادثة."
+    if not history or history[-1]["role"] != "assistant":
+        return None, "⚠️ لا يوجد رد من الوكيل لتطبيقه."
     
-    last_bot_msg = history[-1]["content"]
-    # Regex للبحث عن بلوكات الأكواد مع أسماء الملفات
+    text = history[-1]["content"]
     pattern = r"\[\s*(?:ملف|file)\s*:\s*(.*?)\s*\]\s*```.*?\n(.*?)```"
-    matches = re.findall(pattern, last_bot_msg, re.DOTALL | re.IGNORECASE)
+    matches = re.findall(pattern, text, re.DOTALL | re.IGNORECASE)
     
     if not matches:
-        return None, "⚠️ لم يتم العثور على تعديلات بتنسيق [ملف: ...]"
+        return None, "⚠️ لم أجد أكواداً بتنسيق [ملف: ...]"
     
     for name, code in matches:
         sandbox.update_file(name.strip(), code.strip())
     
-    zip_result = sandbox.create_zip()
-    return zip_result, f"✅ تم تحديث {len(matches)} ملف وحزم المشروع."
+    return sandbox.create_zip(), f"✅ تم تحديث {len(matches)} ملف."
 
 # --------------------------- واجهة Gradio ---------------------------
-with gr.Blocks(theme=gr.themes.Soft(), title="Smart Dev Sandbox") as demo:
-    gr.Markdown("# 🚀 Smart Dev Sandbox \n### التعديل الذكي على المشاريع (ZIP/DOCX/PDF)")
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🚀 Smart Dev Sandbox PRO")
     
     with gr.Row():
         with gr.Column(scale=2):
-            chatbot = gr.Chatbot(type="messages", label="المساعد المبرمج", height=500)
-            msg_input = gr.Textbox(
-                label="اطلب تعديلاً", 
-                placeholder="مثال: اجعل لون الخلفية في style.css داكناً...",
-                show_label=False
-            )
-            with gr.Row():
-                send_btn = gr.Button("إرسال الطلب", variant="primary")
-                clear_btn = gr.Button("مسح الدردشة")
+            chatbot = gr.Chatbot(type="messages", label="الدردشة البرمجية", height=450)
+            msg_input = gr.Textbox(placeholder="اطلب تعديلاً على الملفات المرفوعة...", show_label=False)
+            send_btn = gr.Button("إرسال الطلب ⚡", variant="primary")
         
         with gr.Column(scale=1):
-            with gr.Group():
-                file_zip = gr.File(label="1. ارفع مشروعك (ZIP)")
-                pwd = gr.Textbox(label="كلمة السر", type="password", placeholder="اتركه فارغاً إن لم يوجد")
-                load_btn = gr.Button("تحليل الملفات 🔍")
-            
+            file_zip = gr.File(label="1. ارفع ملف ZIP")
+            pwd = gr.Textbox(label="كلمة السر (اختياري)", type="password")
+            load_btn = gr.Button("تحليل الملفات")
             status = gr.Markdown("🟢 **الحالة:** جاهز")
-            
-            with gr.Group():
-                apply_btn = gr.Button("2. تطبيق التعديلات وحفظ ZIP ✨", variant="stop")
-                download = gr.File(label="3. تحميل المشروع المعدل")
-
-    # ربط الأحداث (Logic flow)
-    # عند الضغط على إرسال أو Enter
-    msg_event = send_btn.click(
-        handle_user_msg, [msg_input, chatbot], [msg_input, chatbot], queue=False
-    ).then(
+            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]
     )
-    
-    msg_input.submit(
-        handle_user_msg, [msg_input, chatbot], [msg_input, chatbot], queue=False
-    ).then(
+    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])
-    clear_btn.click(lambda: [], None, chatbot, queue=False)
 
 if __name__ == "__main__":
-    demo.launch(debug=True)
+    demo.launch()