Update app.py

app.py

@@ -8,24 +8,23 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 
 # --------------------------- الإعدادات ---------------------------
-# قائمة شاملة لضمان قبول كل ملفات المشروع دون أخطاء
 ALLOWED_EXTENSIONS = {
     '.py', '.js', '.html', '.css', '.txt', '.json', '.md', 
     '.php', '.xml', '.csv', '.zip', '.png', '.jpg', '.svg', '.ico'
 }
-MODEL_ID = os.environ.get('HF_MODEL_ID', 'moonshotai/Kimi-K2-Instruct')
+# نستخدم نموذج Qwen2.5-Coder لقوته الفائقة في فهم المشاريع البرمجية كاملة
+MODEL_ID = os.environ.get('HF_MODEL_ID', 'Qwen/Qwen2.5-Coder-32B-Instruct')
 HF_TOKEN = os.environ.get('HF_TOKEN')
 
 class ProjectTank:
     def __init__(self):
         self.work_dir = tempfile.mkdtemp(prefix="tank_dev_")
-        self.files_cache = {} # لتخزين محتوى الملفات: {الاسم: المحتوى}
+        self.files_cache = {} # {filename: content}
+        self.structure = ""    # خريطة المشروع
 
     def sync_files(self, file_objs):
-        """رفع الملفات وفكها وتجهيزها للمعالجة"""
         if not file_objs: return "⚠️ لم يتم اختيار ملفات.", []
         
-        # تنظيف المجلد القديم
         if os.path.exists(self.work_dir):
             shutil.rmtree(self.work_dir)
         self.work_dir = tempfile.mkdtemp(prefix="tank_dev_")
@@ -41,16 +40,21 @@ class ProjectTank:
                 shutil.copy(f.name, dest)
         
         self._reload_memory()
-        return f"🚀 تم تحميل {len(self.files_cache)} ملف في السيرفر المحلي.", list(self.files_cache.keys())
+        return f"🚀 تم تحميل {len(self.files_cache)} ملف بنجاح.", list(self.files_cache.keys())
 
     def _reload_memory(self):
         self.files_cache = {}
+        structure_list = []
         for p in Path(self.work_dir).rglob('*'):
             if p.is_file():
                 rel = str(p.relative_to(self.work_dir)).replace("\\", "/")
+                structure_list.append(f"- {rel}")
                 try:
-                    self.files_cache[rel] = p.read_text(encoding='utf-8', errors='ignore')
+                    # نقرأ فقط ملفات الكود في الذاكرة لتوفير الـ Tokens
+                    if p.suffix in ['.html', '.css', '.js', '.json', '.txt', '.md', '.py']:
+                        self.files_cache[rel] = p.read_text(encoding='utf-8', errors='ignore')
                 except: pass
+        self.structure = "\n".join(structure_list)
 
     def update_file_content(self, filename, content):
         path = os.path.join(self.work_dir, filename)
@@ -60,118 +64,155 @@ class ProjectTank:
         self.files_cache[filename] = content
 
     def build_preview(self):
-        """محرك المعاينة المحلي: حقن كل التبعيات في ملف واحد"""
-        # البحث عن نقطة الدخول (Entry Point)
-        entry = next((f for f in ["index.html", "main.html"] if f in self.files_cache), None)
-        if not entry:
-            entry = next((f for f in self.files_cache if f.endswith('.html')), None)
+        entry = next((f for f in ["index.html", "main.html"] if f in self.files_cache), 
+                     next((f for f in self.files_cache if f.endswith('.html')), None))
         
         if not entry:
-            return "<div style='text-align:center; padding:50px;'>⚠️ لم يتم العثور على ملف HTML</div>"
+            return "<div style='text-align:center; padding:50px;'>⚠️ لم يتم العثور على ملف HTML للمعاينة</div>"
 
         html = self.files_cache[entry]
 
-        # حقن CSS
+        # 1. حقن CSS (بذكاء: استبدال أو إضافة في الـ Head)
+        css_content = ""
         for name, code in self.files_cache.items():
             if name.endswith('.css'):
-                style = f"<style data-file='{name}'>\n{code}\n</style>"
-                html = re.sub(f'<link.*?href=["\'].*?{name}["\'].*?>', style, html)
+                css_content += f"\n/* --- {name} --- */\n{code}\n"
+        
+        if "</head>" in html:
+            html = html.replace("</head>", f"<style>{css_content}</style>\n</head>")
+        else:
+            html = f"<style>{css_content}</style>" + html
 
-        # حقن JS
+        # 2. حقن JS (في نهاية الـ Body لضمان عمل الـ DOM)
+        js_content = ""
         for name, code in self.files_cache.items():
             if name.endswith('.js'):
-                script = f"<script data-file='{name}'>\n{code}\n</script>"
-                html = re.sub(f'<script.*?src=["\'].*?{name}["\'].*?></script>', script, html)
+                js_content += f"\n// --- {name} ---\n{code}\n"
+        
+        if "</body>" in html:
+            html = html.replace("</body>", f"<script>{js_content}</script>\n</body>")
+        else:
+            html += f"<script>{js_content}</script>"
 
         return html
 
+    def export_project(self):
+        out_path = tempfile.mktemp()
+        shutil.make_archive(out_path, 'zip', self.work_dir)
+        return out_path + ".zip"
+
 tank = ProjectTank()
 client = InferenceClient(api_key=HF_TOKEN) if HF_TOKEN else None
 
 # --------------------------- منطق الذكاء الاصطناعي ---------------------------
 
-def clean_messages(history):
-    """إصلاح خطأ Data Incompatible - تنظيف مصفوفة الرسائل"""
-    return [{"role": m["role"], "content": str(m["content"])} for m in history if "role" in m and "content" in m]
-
 def chat_and_code(message, history):
     if not client:
         yield history + [{"role": "assistant", "content": "⚠️ مفقود HF_TOKEN"}]
         return
 
-    # جمع السياق الكامل (وضع الدبابة)
-    context = "### CURRENT PROJECT STRUCTURE & CONTENT ###\n"
+    # بناء سياق قوي: الهيكل + المحتوى
+    context = f"### PROJECT STRUCTURE ###\n{tank.structure}\n\n"
+    context += "### FULL FILE CONTENTS ###\n"
     for name, content in tank.files_cache.items():
-        context += f"\nFILE: {name}\n{content}\n--- END ---\n"
+        context += f"--- FILE: {name} ---\n{content}\n"
 
     system_instr = (
-        "You are an AI IDE. Provide short explanations then output the FULL modified files using this format:\n"
-        "[TARGET: filename]\n```\nUpdated Code\n```"
+        "You are an expert AI IDE. Your goal is to help the user build and debug their project.\n"
+        "1. Analyze the project structure to understand how files interact.\n"
+        "2. Provide clear explanations.\n"
+        "3. When modifying code, use the format: [TARGET: filename] followed by a code block.\n"
+        "4. Always output the FULL content of the file in the code block."
     )
 
-    clean_hist = clean_messages(history)
-    payload = [{"role": "system", "content": system_instr}]
-    payload.append({"role": "user", "content": f"{context}\n\nUser Request: {message}"})
+    messages = [{"role": "system", "content": system_instr}]
+    for m in history:
+        messages.append({"role": m["role"], "content": str(m["content"])})
+    
+    messages.append({"role": "user", "content": f"{context}\n\nTask: {message}"})
 
     response = ""
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": ""})
 
-    stream = client.chat_completion(model=MODEL_ID, messages=payload, max_tokens=4000, stream=True)
-    for chunk in stream:
-        token = chunk.choices[0].delta.content
-        if token:
-            response += token
-            history[-1]["content"] = response
-            yield history
+    try:
+        stream = client.chat_completion(model=MODEL_ID, messages=messages, max_tokens=6000, stream=True)
+        for chunk in stream:
+            if hasattr(chunk, 'choices') and 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:
+        history[-1]["content"] = f"❌ خطأ في السيرفر: {str(e)}"
+        yield history
 
 def apply_and_preview(history):
-    """استخراج الكود وتحديث السيرفر المحلي"""
-    if not history: return "لا توجد بيانات", None
+    if not history: return "لا توجد بيانات", None, []
     
     content = history[-1]["content"]
-    matches = re.findall(r"\[TARGET:\s*(.*?)\]\s*```.*?\n(.*?)```", content, re.DOTALL)
+    matches = re.findall(r"\[TARGET:\s*(.*?)\]\s*```.*?\n(.*?)```", content, re.DOTALL | re.IGNORECASE)
+    
+    if not matches:
+        return "⚠️ لم يتم اكتشاف ملفات للتحديث.", tank.build_preview(), list(tank.files_cache.keys())
     
     for filename, code in matches:
         tank.update_file_content(filename.strip(), code.strip())
     
-    return "✅ تم التعديل والمعاينة بنجاح", tank.build_preview()
+    return "✅ تم تحديث الملفات وحقن المعاينة بنجاح!", tank.build_preview(), list(tank.files_cache.keys())
 
 # --------------------------- الواجهة ---------------------------
 
-
-
-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.Default(primary_hue="blue", spacing_size="sm"), title="AI Tank IDE Power") as demo:
+    gr.Markdown("# 🛡️ AI Tank IDE Pro (V2)\n**بيئة تطوير ذكية مع رؤية كاملة للهيكل**")
     
+    active_file = gr.State()
+
     with gr.Row():
         with gr.Column(scale=1):
-            upload = gr.UploadButton("📁 ارفع مشروعك (ZIP/Files)", file_count="multiple")
-            file_list = gr.Dropdown(label="ملفات المشروع المكتشفة", choices=[])
+            upload = gr.UploadButton("📁 ارفع المشروع (ZIP/Files)", file_count="multiple", variant="secondary")
+            file_list = gr.Dropdown(label="مستكشف الملفات", choices=[])
             status = gr.Markdown("🟢 جاهز للعمل")
             
             with gr.Group():
-                chat = gr.Chatbot(type="messages", height=400)
-                msg = gr.Textbox(placeholder="اطلب تعديل المشروع...", show_label=False)
+                chat = gr.Chatbot(type="messages", height=450)
+                msg = gr.Textbox(placeholder="اطلب تعديلاً أو اسأل عن عمل المشروع...", show_label=False)
                 with gr.Row():
                     send = gr.Button("إرسال", variant="primary")
-                    apply = gr.Button("⚡ تطبيق المعاينة", variant="stop")
+                    apply = gr.Button("⚡ تطبيق التعديلات", variant="stop")
+            
+            export_btn = gr.DownloadButton("📥 تصدير المشروع النهائي", variant="secondary")
 
         with gr.Column(scale=2):
             with gr.Tabs():
-                with gr.TabItem("🌐 المعاينة الفورية (Local Server)"):
-                    preview = gr.HTML(value="<div style='text-align:center; padding:50px;'>ارفع مشروعك لرؤية النتيجة</div>")
+                with gr.TabItem("🌐 المعاينة المباشرة"):
+                    preview = gr.HTML(value="<div style='text-align:center; padding:50px;'>ارفع مشروعك لبدء المعاينة</div>")
                 with gr.TabItem("📝 محرر الكود"):
-                    editor = gr.Code(label="كود الملف المختار", lines=25, interactive=True)
+                    editor = gr.Code(label="محتوى الملف", lines=25, interactive=True)
+                    save_manual = gr.Button("💾 حفظ التعديل اليدوي", size="sm")
     
-    # ربط الأحداث
-    upload.upload(tank.sync_files, upload, [status, file_list]).then(tank.build_preview, None, preview)
+    # --- ربط الأحداث ---
+
+    def on_upload(files):
+        log, choices = tank.sync_files(files)
+        return log, gr.update(choices=choices), tank.build_preview()
+
+    upload.upload(on_upload, upload, [status, file_list, preview])
     
     send.click(chat_and_code, [msg, chat], chat).then(lambda: "", None, msg)
     
-    apply.click(apply_and_preview, chat, [status, preview])
+    def on_apply(h):
+        log, pre, choices = apply_and_preview(h)
+        return log, pre, gr.update(choices=choices)
+    
+    apply.click(on_apply, chat, [status, preview, file_list])
+    
+    file_list.change(lambda n: (tank.files_cache.get(n, ""), n), file_list, [editor, active_file])
+    
+    save_manual.click(lambda n, c: tank.update_file_content(n, c) or "✅ تم الحفظ يدوياً", [active_file, editor], status).then(tank.build_preview, None, preview)
     
-    file_list.change(lambda n: tank.files_cache.get(n, ""), file_list, editor)
+    export_btn.click(tank.export_project, None, export_btn)
 
 if __name__ == "__main__":
     demo.queue().launch()