Update app.py

app.py

@@ -18,26 +18,38 @@ def generate_texture(stl_file):
             uv /= uv.max(axis=0) + 1e-10  # Avoid division by zero
             mesh.visual.uv = uv
 
-        # Generate a simple texture (e.g., gradient based on UV coordinates)
+        # Create a texture image
         texture_size = (512, 512)  # Texture resolution
-        texture = np.zeros((texture_size[0], texture_size[1], 3), dtype=np.uint8)
+        texture = np.zeros((texture_size[0], texture_size[1], 3), dtype=np.uint8)  # Black background
 
-        # Create a gradient texture (R, G based on UV coordinates)
-        for i in range(texture_size[0]):
-            for j in range(texture_size[1]):
-                u, v = i / texture_size[0], j / texture_size[1]
-                texture[i, j] = [int(u * 255), int(v * 255), 128]  # RGB gradient
+        # Map UV coordinates to texture pixels and highlight in red
+        uv = mesh.visual.uv
+        pixel_coords = (uv * np.array(texture_size)).astype(int)  # Map UVs to pixel space
+        pixel_coords = np.clip(pixel_coords, 0, np.array(texture_size) - 1)  # Ensure within bounds
 
-        # Alternatively, generate a normal map (optional)
-        # normals = mesh.vertex_normals
-        # texture = ((normals + 1) * 127.5).astype(np.uint8)  # Convert normals to RGB
+        # Highlight pixels corresponding to UV coordinates in red
+        for u, v in pixel_coords:
+            texture[v, u] = [255, 0, 0]  # Red for model areas
+
+        # Optional: Fill faces by rasterizing triangles (more accurate coverage)
+        if hasattr(mesh, 'faces'):
+            for face in mesh.faces:
+                # Get UVs for the face's vertices
+                face_uvs = uv[face]
+                # Convert to pixel coordinates
+                face_pixels = (face_uvs * np.array(texture_size)).astype(int)
+                face_pixels = np.clip(face_pixels, 0, np.array(texture_size) - 1)
+                # Approximate triangle fill by setting a small region around vertices
+                for px, py in face_pixels:
+                    texture[max(0, py-2):min(texture_size[1], py+3),
+                           max(0, px-2):min(texture_size[0], px+3)] = [255, 0, 0]
 
         # Save texture as PNG
         texture_img = Image.fromarray(texture)
         output_path = "texture_output.png"
         texture_img.save(output_path)
 
-        return output_path, "Texture generated successfully!"
+        return output_path, "Texture generated with model areas highlighted in red!"
     except Exception as e:
         return None, f"Error: {str(e)}"
 
@@ -49,8 +61,8 @@ iface = gr.Interface(
         gr.File(label="Download Texture (PNG)"),
         gr.Textbox(label="Status")
     ],
-    title="STL Texture Generator",
-    description="Upload an STL file to generate a texture and export it as a PNG."
+    title="STL Texture Generator with Highlight",
+    description="Upload an STL file to generate a texture with model areas highlighted in red, exported as a PNG."
 )
 
 # Launch the app