Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 import uuid
 import shutil
 import tempfile
+import sys
 
 DESCRIPTION = """# SHARP · 3D from a Single Photo
 Upload any photo and get a **3D Gaussian Splat (.ply)** in seconds.
@@ -15,37 +16,68 @@ Powered by Apple's [SHARP](https://github.com/apple/ml-sharp) monocular view syn
 3. Download the `.ply` file
 4. View it at [SuperSplat](https://playcanvas.com/supersplat/editor)
 
-> Running on CPU - first run downloads the 2.6GB model and may take 5-10 min."""
+> Running on CPU — first run downloads the ~2.6GB model and may take 5–10 min."""
+
+
+def get_sharp_bin():
+    """Locate the sharp CLI binary from the current Python environment."""
+    scripts_dir = os.path.join(os.path.dirname(sys.executable), "sharp")
+    if os.path.isfile(scripts_dir):
+        return scripts_dir
+    # Standard location: same bin dir as python
+    candidate = os.path.join(os.path.dirname(sys.executable), "sharp")
+    if os.path.isfile(candidate):
+        return candidate
+    # Try finding via which
+    result = subprocess.run(["which", "sharp"], capture_output=True, text=True)
+    if result.returncode == 0:
+        return result.stdout.strip()
+    raise FileNotFoundError(
+        "Could not locate the `sharp` binary. "
+        "Make sure `git+https://github.com/apple/ml-sharp.git` is in requirements.txt."
+    )
+
 
 def generate_splat(image_path):
     if image_path is None:
         raise gr.Error("Please upload an image first.")
+
     job_id = str(uuid.uuid4())
     input_dir = os.path.join(tempfile.gettempdir(), f"sharp_in_{job_id}")
     output_dir = os.path.join(tempfile.gettempdir(), f"sharp_out_{job_id}")
     os.makedirs(input_dir, exist_ok=True)
     os.makedirs(output_dir, exist_ok=True)
+
     try:
         ext = os.path.splitext(image_path)[1] or ".jpg"
         shutil.copy(image_path, os.path.join(input_dir, f"input{ext}"))
+
+        sharp_bin = get_sharp_bin()
         result = subprocess.run(
-            ["sharp", "predict", "-i", input_dir, "-o", output_dir],
+            [sharp_bin, "predict", "-i", input_dir, "-o", output_dir],
             capture_output=True, text=True, timeout=600
         )
+
         if result.returncode != 0:
-            raise gr.Error(f"SHARP failed: {result.stderr[-500:]}")
+            raise gr.Error(f"SHARP failed:\n{result.stderr[-800:]}")
+
         ply_files = [f for f in os.listdir(output_dir) if f.endswith(".ply")]
         if not ply_files:
             raise gr.Error("No .ply file generated. Try a different image.")
+
         out_path = os.path.join(tempfile.gettempdir(), f"output_{job_id}.ply")
         shutil.copy(os.path.join(output_dir, ply_files[0]), out_path)
-        return out_path, "Done! Download your .ply file above, then open it in SuperSplat."
+        return out_path, "✅ Done! Download your .ply above, then open it in SuperSplat."
+
     except subprocess.TimeoutExpired:
         raise gr.Error("Timed out after 10 minutes.")
+    except FileNotFoundError as e:
+        raise gr.Error(str(e))
     finally:
         shutil.rmtree(input_dir, ignore_errors=True)
         shutil.rmtree(output_dir, ignore_errors=True)
 
+
 with gr.Blocks(title="SHARP 3D") as demo:
     gr.Markdown(DESCRIPTION)
     with gr.Row():
@@ -55,7 +87,12 @@ with gr.Blocks(title="SHARP 3D") as demo:
         with gr.Column():
             file_output = gr.File(label="Download .ply file")
             status_output = gr.Markdown("")
-    run_btn.click(fn=generate_splat, inputs=image_input, outputs=[file_output, status_output])
+
+    run_btn.click(
+        fn=generate_splat,
+        inputs=image_input,
+        outputs=[file_output, status_output]
+    )
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0")