Update backend_api.py

backend_api.py

@@ -1333,26 +1333,41 @@ async def import_model(path: str, prefer_local: bool = False):
             "metadata": {}
         }
 
+@app.websocket("/ws/generate")
+async def websocket_generate(websocket: WebSocket):
+    """WebSocket endpoint for real-time code generation"""
+    await websocket.accept()
+    import secrets # Đảm bảo đã import secrets để tạo tên space ngẫu nhiên
 
-@app.get("/api/import/github/{owner}/{repo}")
-async def import_github(owner: str, repo: str):
-    """Import a GitHub repository by owner and repo name"""
     try:
-        importer = ProjectImporter()
-        result = importer.import_github_repo(owner, repo)
-        return result
-    except Exception as e:
-        return {
-            "status": "error",
-            "message": f"Failed to import repository: {str(e)}",
-            "code": "",
-            "language": "python",
-            "url": f"https://github.com/{owner}/{repo}",
-            "metadata": {}
-        }
-
-
-            # ... (Phần code cũ của bạn) ...
+        while True:
+            # Receive message from client
+            data = await websocket.receive_json()
+            
+            query = data.get("query")
+            language = data.get("language", "html")
+            model_id = data.get("model_id", "claude-opus-4.5")
+            
+            # Send acknowledgment
+            await websocket.send_json({
+                "type": "status",
+                "message": "Generating code..."
+            })
+            
+            # Mock code generation
+            await asyncio.sleep(0.5)
+            
+            # Code mẫu
+            sample_code = f"\n<h1>Hello from AnyCoder!</h1>"
+            
+            # Stream code
+            for i, char in enumerate(sample_code):
+                await websocket.send_json({
+                    "type": "chunk",
+                    "content": char,
+                    "progress": (i + 1) / len(sample_code) * 100
+                })
+                await asyncio.sleep(0.01)
             
             # Send completion
             await websocket.send_json({
@@ -1360,7 +1375,7 @@ async def import_github(owner: str, repo: str):
                 "code": sample_code
             })
             
-            # --- CHÈN ĐOẠN NÀY VÀO ĐỂ TỰ ĐỘNG DEPLOY ---
+            # Tự động Deploy
             try:
                 from backend_deploy import deploy_to_huggingface_space
                 print(f"DEBUG: [FLOW] Bắt đầu deploy tự động cho {language}...")
@@ -1369,11 +1384,10 @@ async def import_github(owner: str, repo: str):
                 success, msg, url = deploy_to_huggingface_space(
                     code=sample_code, 
                     language=language,
-                    # Tên space tự sinh để tránh trùng lặp
                     space_name=f"anycoder-{secrets.token_hex(4)}" 
                 )
                 
-                # Gửi kết quả deploy về cho frontend hiển thị
+                # Gửi kết quả về frontend
                 await websocket.send_json({
                     "type": "deploy_status",
                     "success": success,
@@ -1383,8 +1397,25 @@ async def import_github(owner: str, repo: str):
                 print(f"DEBUG: [FLOW] Kết quả deploy: {success}, URL: {url}")
             except Exception as e:
                 print(f"DEBUG: [FLOW] Lỗi gọi deploy: {str(e)}")
-            # -------------------------------------------
+                await websocket.send_json({
+                    "type": "deploy_status",
+                    "success": False,
+                    "message": f"Deploy failed: {str(e)}",
+                    "url": None
+                })
+            
+    except WebSocketDisconnect:
+        print("Client disconnected")
+    except Exception as e:
+        print(f"Error: {str(e)}")
+        try:
+            await websocket.send_json({"type": "error", "message": str(e)})
+            await websocket.close()
+        except:
+            pass
+            
 
+            
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run("backend_api:app", host="0.0.0.0", port=8000, reload=True)