Update app.py

app.py

@@ -6,6 +6,12 @@ import zipfile
 from pathlib import Path
 import gradio as gr
 from huggingface_hub import InferenceClient
+from flask import Flask, send_from_directory, render_template_string, request
+from watchdog.observers import Observer
+from watchdog.events import FileSystemEventHandler
+import tornado.ioloop
+import tornado.web
+import tornado.websocket
 
 # --------------------------- الإعدادات ---------------------------
 ALLOWED_EXTENSIONS = {
@@ -40,6 +46,7 @@ class ProjectTank:
                 shutil.copy(f.name, dest)
         
         self._reload_memory()
+        self._start_live_preview()
         return f"🚀 تم تحميل {len(self.files_cache)} ملف بنجاح.", list(self.files_cache.keys())
 
     def _reload_memory(self):
@@ -100,6 +107,50 @@ class ProjectTank:
         out_path = tempfile.mktemp()
         shutil.make_archive(out_path, 'zip', self.work_dir)
         return out_path + ".zip"
+    
+    def _start_live_preview(self):
+        class PreviewHandler(tornado.web.RequestHandler):
+            def get(self):
+                entry = next((f for f in ["index.html", "main.html"] if f in tank.files_cache), 
+                             next((f for f in tank.files_cache if f.endswith('.html')), None))
+                if not entry:
+                    self.write("<div style='text-align:center; padding:50px;'>⚠️ لم يتم العثور على ملف HTML للمعاينة</div>")
+                    return
+                self.write(tank.files_cache[entry])
+
+        class WebSocketHandler(tornado.websocket.WebSocketHandler):
+            clients = set()
+            
+            def open(self):
+                WebSocketHandler.clients.add(self)
+            
+            def on_close(self):
+                WebSocketHandler.clients.remove(self)
+            
+            @classmethod
+            def send_update(cls, message):
+                for client in cls.clients:
+                    client.write_message(message)
+
+        class Watcher(FileSystemEventHandler):
+            def on_modified(self, event):
+                if event.is_directory: return
+                if Path(event.src_path).suffix in ALLOWED_EXTENSIONS:
+                    tank._reload_memory()
+                    message = tank.build_preview()
+                    WebSocketHandler.send_update(message)
+
+        self.watcher = Observer()
+        self.watcher.schedule(Watcher(), self.work_dir, recursive=True)
+        self.watcher.start()
+
+        self.server = tornado.web.Application([
+            (r"/preview", PreviewHandler),
+            (r"/ws", WebSocketHandler),
+        ])
+        self.server.listen(8888)
+        tornado.ioloop.IOLoop.current().start()
+        
 
 tank = ProjectTank()
 client = InferenceClient(api_key=HF_TOKEN) if HF_TOKEN else None
@@ -164,55 +215,131 @@ def apply_and_preview(history):
 
 # --------------------------- الواجهة ---------------------------
 
-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()
+app = Flask(__name__)
 
-    with gr.Row():
-        with gr.Column(scale=1):
-            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=450)
-                msg = gr.Textbox(placeholder="اطلب تعديلاً أو اسأل عن عمل المشروع...", show_label=False)
-                with gr.Row():
-                    send = gr.Button("إرسال", variant="primary")
-                    apply = gr.Button("⚡ تطبيق التعديلات", variant="stop")
-            
-            export_btn = gr.DownloadButton("📥 تصدير المشروع النهائي", variant="secondary")
-
-        with gr.Column(scale=2):
-            with gr.Tabs():
-                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)
-                    save_manual = gr.Button("💾 حفظ التعديل اليدوي", size="sm")
-    
-    # --- ربط الأحداث ---
+@app.route('/upload', methods=['POST'])
+def upload_file():
+    files = request.files.getlist('files[]')
+    log, choices = tank.sync_files(files)
+    return log
+
+@app.route('/files/<path:filename>')
+def get_file(filename):
+    path = os.path.join(tank.work_dir, filename)
+    return send_from_directory(tank.work_dir, filename)
 
-    def on_upload(files):
-        log, choices = tank.sync_files(files)
-        return log, gr.update(choices=choices), tank.build_preview()
+@app.route('/project')
+def get_project_structure():
+    return {"structure": tank.structure}
 
-    upload.upload(on_upload, upload, [status, file_list, preview])
+@app.route('/preview')
+def serve_preview():
+    entry = next((f for f in ["index.html", "main.html"] if f in tank.files_cache), 
+                 next((f for f in tank.files_cache if f.endswith('.html')), None))
+    if not entry:
+        return "⚠️ لم يتم العثور على ملف HTML للمعاينة"
+    return render_template_string(tank.files_cache[entry])
+
+@app.route('/chat', methods=['POST'])
+def chat_endpoint():
+    message = request.form['message']
+    history = request.form.get('history', [])
+    if not client:
+        history.append({"role": "assistant", "content": "⚠️ مفقود HF_TOKEN"})
+        return str(history)
     
-    send.click(chat_and_code, [msg, chat], chat).then(lambda: "", None, msg)
+    context = f"### PROJECT STRUCTURE ###\n{tank.structure}\n\n"
+    context += "### FULL FILE CONTENTS ###\n"
+    for name, content in tank.files_cache.items():
+        context += f"--- FILE: {name} ---\n{content}\n"
+
+    system_instr = (
+        "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."
+    )
+
+    messages = [{"role": "system", "content": system_instr}]
+    for m in history:
+        messages.append({"role": m["role"], "content": str(m)})
     
-    def on_apply(h):
-        log, pre, choices = apply_and_preview(h)
-        return log, pre, gr.update(choices=choices)
+    messages.append({"role": "user", "content": f"{context}\n\nTask: {message}"})
+
+    response = ""
+    history.append({"role": "user", "content": message})
+    history.append({"role": "assistant", "content": ""})
+
+    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
+        return str(history)
+    except Exception as e:
+        history[-1]["content"] = f"❌ خطأ في السيرفر: {str(e)}"
+        return str(history)
+
+@app.route('/apply', methods=['POST'])
+def apply_endpoint():
+    history = request.form.get('history', [])
+    if not history: return "لا توجد بيانات"
     
-    apply.click(on_apply, chat, [status, preview, file_list])
+    content = history[-1]["content"]
+    matches = re.findall(r"\[TARGET:\s*(.*?)\]\s*```.*?\n(.*?)```", content, re.DOTALL | re.IGNORECASE)
     
-    file_list.change(lambda n: (tank.files_cache.get(n, ""), n), file_list, [editor, active_file])
+    if not matches:
+        return "⚠️ لم يتم اكتشاف ملفات للتحديث."
     
-    save_manual.click(lambda n, c: tank.update_file_content(n, c) or "✅ تم الحفظ يدوياً", [active_file, editor], status).then(tank.build_preview, None, preview)
+    for filename, code in matches:
+        tank.update_file_content(filename.strip(), code.strip())
     
-    export_btn.click(tank.export_project, None, export_btn)
+    return "✅ تم تحديث الملفات وحقن المعاينة بنجاح!"
+
+@app.route('/export')
+def export_project():
+    return send_from_directory(Path(tank.export_project()).parent, Path(tank.export_project()).name)
+
+@app.route('/')
+def serve_index():
+    return """
+    <div>
+        <h1>AI Tank IDE Pro</h1>
+        <a href="/preview">معاينة المشروع</a><br/>
+        <input type="file" id="files" name="files[]" multiple />
+        <button onclick="uploadFiles()">ارفع المشروع</button>
+        <div id="status">جاهز للعمل</div>
+        <div id="preview" style="margin-top: 20px;"></div>
+    </div>
+    <script>
+        function uploadFiles() {
+            const formData = new FormData();
+            const fileInput = document.getElementById('files');
+            Array.from(fileInput.files).forEach(file => {
+                formData.append('files[]', file);
+            });
+            
+            fetch('/upload', { method: 'POST', body: formData })
+                .then(response => response.text())
+                .then(log => {
+                    document.getElementById('status').innerText = log;
+                    if (log.includes('تم تحميل')) {
+                        fetch('/preview')
+                            .then(res => res.text())
+                            .then(html => {
+                                document.getElementById('preview').innerHTML = html;
+                            });
+                    }
+                });
+        }
+    </script>
+    """
 
 if __name__ == "__main__":
-    demo.queue().launch()
+    tank = ProjectTank()
+    client = InferenceClient(api_key=HF_TOKEN) if HF_TOKEN else None
+    app.run(host='0.0.0.0', port=5000)