Update backend_api.py

backend_api.py

@@ -845,14 +845,47 @@ async def generate_code(
                     {"role": "user", "content": user_content}
                 ]
             
-            # Tìm đoạn "try: # Stream the response" trong backend_api.py và thay bằng:
+                        # ... (Các phần khởi tạo trước đó giữ nguyên)
+            
+            # GỌI GEMINI VÀ XỬ LÝ STREAM
             try:
-                # Sử dụng Gemini stream
+                # 1. Gọi Gemini
                 response = client.generate_content(
                     f"Generate a {language} application: {query}",
                     stream=True
                 )
                 
+                # 2. Xử lý stream từ Gemini
+                for chunk in response:
+                    if chunk.text:
+                        chunk_content = chunk.text
+                        generated_code += chunk_content
+                        
+                        # Gửi chunk về frontend
+                        event_data = json.dumps({
+                            "type": "chunk",
+                            "content": chunk_content
+                        })
+                        yield f"data: {event_data}\n\n"
+                
+                # 3. Gửi sự kiện hoàn tất (Completion)
+                # Lưu ý: Không cần biến 'stream' cũ nữa
+                reasoning = extract_reasoning(generated_code, language)
+                generated_code = cleanup_generated_code(generated_code, language)
+                
+                completion_dict = {
+                    "type": "complete",
+                    "code": generated_code
+                }
+                # Thêm logic reasoning nếu cần
+                if selected_model_id == "zai-org/GLM-4.7" and reasoning:
+                    completion_dict["reasoning"] = reasoning
+                    
+                completion_data = json.dumps(completion_dict)
+                yield f"data: {completion_data}\n\n"
+                
+                # ... (Phần logic Auto-Deploy giữ nguyên như cũ) ...
+                
                 for chunk in response:
                     if chunk.text:
                         chunk_content = chunk.text