Update app.py

app.py

@@ -86,58 +86,85 @@ def create_frames_zip(frames_dir):
     
     return zip_path
 
-def create_3d_model(frames, width, height, depth_scale=0.1):
+def create_3d_model(frames, width, height, depth_scale=0.05):
     """
-    تبدیل فریم‌های Canny به مدل 3D (GLB)
+    تبدیل فریم‌های Canny به مدل 3D (GLB) با Mesh واقعی
     """
     try:
         # انتخاب تعدادی از فریم‌ها برای جلوگیری از سنگین شدن مدل
-        step = max(1, len(frames) // 30)  # حداکثر 30 لایه
+        step = max(1, len(frames) // 20)  # حداکثر 20 لایه
         selected_frames = frames[::step]
         
-        vertices = []
-        faces = []
-        vertex_offset = 0
+        all_vertices = []
+        all_faces = []
+        vertex_count = 0
         
         # تبدیل هر فریم به یک لایه 3D
         for idx, frame in enumerate(selected_frames):
             # کاهش وضوح برای سبک‌تر کردن مدل
-            scale = 4
+            scale = 8
             small_frame = cv2.resize(frame, (width // scale, height // scale))
             
             h, w = small_frame.shape
             z = idx * depth_scale  # فاصله بین لایه‌ها
             
-            # ایجاد vertices از پیکسل‌های سفید (لبه‌ها)
+            # ایجاد یک grid از vertices برای این فریم
+            layer_vertices = []
+            layer_indices = {}
+            
             for y in range(h):
                 for x in range(w):
                     if small_frame[y, x] > 128:  # پیکسل‌های روشن
                         # نرمال‌سازی مختصات
                         nx = (x / w - 0.5) * 2
                         ny = (0.5 - y / h) * 2
-                        vertices.append([nx, ny, z])
-            
-            # ایجاد faces بین لایه‌های متوالی (ساده)
-            if idx > 0 and len(vertices) > vertex_offset:
-                # اتصال ساده بین نقاط
-                for i in range(vertex_offset, len(vertices) - 1):
-                    if i + 1 < len(vertices):
-                        faces.append([i, i + 1, vertex_offset])
+                        
+                        vertex_idx = len(all_vertices)
+                        all_vertices.append([nx, ny, z])
+                        layer_vertices.append(vertex_idx)
+                        layer_indices[(x, y)] = vertex_idx
             
-            vertex_offset = len(vertices)
+            # ایجاد faces درون هر لایه (اتصال پیکسل‌های مجاور)
+            for y in range(h - 1):
+                for x in range(w - 1):
+                    # بررسی اگر این پیکسل و همسایگانش سفید باشند
+                    if ((x, y) in layer_indices and 
+                        (x + 1, y) in layer_indices and 
+                        (x, y + 1) in layer_indices):
+                        
+                        v1 = layer_indices[(x, y)]
+                        v2 = layer_indices[(x + 1, y)]
+                        v3 = layer_indices[(x, y + 1)]
+                        
+                        all_faces.append([v1, v2, v3])
+                    
+                    # مثلث دوم
+                    if ((x + 1, y) in layer_indices and 
+                        (x + 1, y + 1) in layer_indices and 
+                        (x, y + 1) in layer_indices):
+                        
+                        v1 = layer_indices[(x + 1, y)]
+                        v2 = layer_indices[(x + 1, y + 1)]
+                        v3 = layer_indices[(x, y + 1)]
+                        
+                        all_faces.append([v1, v2, v3])
         
-        if len(vertices) < 3:
+        if len(all_vertices) < 3 or len(all_faces) < 1:
             return None
         
-        # ساخت Point Cloud (ابر نقطه) به جای Mesh
-        vertices = np.array(vertices)
+        # ساخت Mesh واقعی
+        vertices = np.array(all_vertices)
+        faces = np.array(all_faces)
+        
+        # ایجاد mesh با trimesh
+        mesh = trimesh.Trimesh(vertices=vertices, faces=faces)
         
-        # ایجاد mesh ساده
-        point_cloud = trimesh.points.PointCloud(vertices)
+        # اصلاح نرمال‌ها
+        mesh.fix_normals()
         
         # ذخیره به فرمت GLB
         glb_path = tempfile.mktemp(suffix='.glb')
-        point_cloud.export(glb_path)
+        mesh.export(glb_path, file_type='glb')
         
         return glb_path