Update app.py

app.py

@@ -4,62 +4,58 @@ import numpy as np
 import tempfile
 import os
 
-def reconstruct_and_download(input_file):
+def reconstruct_perfect_cube(input_file):
     if input_file is None:
         return None, None
     
     try:
-        # 1. Load the uploaded STL file
+        # 1. Load the irregular STL
         mesh = trimesh.load(input_file.name)
         
-        # Handle cases where the STL might be loaded as a Scene
+        # Handle scenes if necessary
         if isinstance(mesh, trimesh.Scene):
             mesh = mesh.dump(concatenate=True)
 
-        # 2. Geometric Reconstruction (Convex Hull)
-        # This fills the irregular cuts by calculating the smallest convex 
-        # shape that encloses all existing points—perfect for restoring a cube.
-        reconstructed_mesh = mesh.convex_hull
+        # 2. Advanced Reconstruction Logic: Bounding Box
+        # Instead of wrapping the broken shape (Convex Hull), 
+        # we calculate the 'Oriented Bounding Box'. 
+        # This identifies the 'Proper' cube that would perfectly 
+        # contain your irregular parts.
+        perfect_cube = mesh.bounding_box_oriented
 
-        # 3. Create a temporary path for the output STL
+        # 3. Create a temporary path for the output
         temp_dir = tempfile.gettempdir()
-        output_filename = "perfect_cube_generated.stl"
+        output_filename = "reconstructed_perfect_cube.stl"
         output_path = os.path.join(temp_dir, output_filename)
         
-        # 4. Export the reconstructed mesh
-        reconstructed_mesh.export(output_path)
+        # 4. Export the perfect solid cube
+        perfect_cube.export(output_path)
         
-        # Return the path twice: once for the 3D viewer, once for the Download component
         return output_path, output_path
     
     except Exception as e:
         print(f"Error: {e}")
         return None, None
 
-# --- Custom Gradio Interface ---
-with gr.Blocks(title="3D Shape Completion AI", theme=gr.themes.Default()) as demo:
-    gr.Markdown("# 🧊 AI 3D Cube Reconstructor")
-    gr.Markdown("Upload an STL cube with irregular cuts. The AI will reconstruct the missing geometry and provide a downloadable 'proper' cube.")
+# --- Gradio UI ---
+with gr.Blocks(title="AI 3D Cube Completion", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🧊 Perfect Cube Generator")
+    gr.Markdown("This tool analyzes irregular STL data and 'hallucinates' the missing sections to restore a mathematically perfect cube.")
     
     with gr.Row():
-        # Input Section
         with gr.Column():
-            file_input = gr.File(label="Upload Irregular STL", file_types=[".stl"])
-            generate_btn = gr.Button("🔨 Reconstruct Perfect Shape", variant="primary")
+            file_input = gr.File(label="Upload Irregular Cube STL", file_types=[".stl"])
+            generate_btn = gr.Button("🚀 Generate Full Constructed Cube", variant="primary")
         
-        # Output Section
         with gr.Column():
-            model_viewer = gr.Model3D(label="3D Preview (Completed Shape)")
-            download_file = gr.File(label="Download Generated STL")
+            model_viewer = gr.Model3D(label="3D Preview of Reconstructed Cube")
+            download_link = gr.File(label="Download Perfect STL File")
 
-    # Action trigger
     generate_btn.click(
-        fn=reconstruct_and_download, 
+        fn=reconstruct_perfect_cube, 
         inputs=file_input, 
-        outputs=[model_viewer, download_file]
+        outputs=[model_viewer, download_link]
     )
 
-    gr.HTML("<br><p style='text-align: center;'>Powered by Trimesh Reconstruction Logic</p>")
-
 if __name__ == "__main__":
     demo.launch()