updates

app.py

@@ -1,4 +1,4 @@
-"""SHARP Gradio demo (Modern, Single-Page UI).
+"""SHARP Gradio demo (Full-Width UI).
 
 This Space:
 - Runs Apple's SHARP model to predict a 3D Gaussian scene from a single image.
@@ -8,6 +8,10 @@ This Space:
 
 from __future__ import annotations
 
+import warnings
+# Suppress the internal torch.distributed warning from ZeroGPU wrappers
+warnings.filterwarnings("ignore", category=FutureWarning, module="torch.distributed")
+
 import json
 from pathlib import Path
 from typing import Final
@@ -29,22 +33,25 @@ EXAMPLES_DIR: Final[Path] = ASSETS_DIR / "examples"
 # Valid image extensions for discovery
 IMAGE_EXTS: Final[tuple[str, ...]] = (".png", ".jpg", ".jpeg", ".webp")
 
-# CSS for a responsive, contained layout
+# CSS for a fluid, full-width layout
 CSS: Final[str] = """
 .gradio-container {
-    max-width: 1400px !important;
+    max-width: 95% !important; /* Fill 95% of the screen width */
     margin: 0 auto;
 }
-/* constrain media height so it doesn't take up the whole screen */
+
+/* Constrain media height so it doesn't overflow vertically on huge screens */
 #input-image img, #output-video video {
-    max-height: 500px;
+    max-height: 65vh; /* Use Viewport Height units for better scaling */
     width: 100%;
     object-fit: contain;
 }
-/* Make the generate button pop slightly */
+
+/* Make the generate button prominent */
 #run-btn {
-    font-size: 1.1em;
+    font-size: 1.2rem;
     font-weight: bold;
+    margin-top: 1rem;
 }
 """
 
@@ -59,7 +66,6 @@ def _ensure_dir(path: Path) -> Path:
 def get_example_files() -> list[list[str]]:
     """
     Scans assets/examples for images to populate the gr.Examples component.
-    Returns a list of lists: [['path/to/img1.jpg'], ['path/to/img2.png']]
     """
     _ensure_dir(EXAMPLES_DIR)
     
@@ -102,7 +108,6 @@ def run_sharp(
     if not image_path:
         raise gr.Error("Please upload or select an input image first.")
 
-    # Validate output resolution
     out_long_side_val: int | None = (
         None if int(output_long_side) <= 0 else int(output_long_side)
     )
@@ -110,8 +115,7 @@ def run_sharp(
     try:
         progress(0.1, desc="Initializing model...")
         
-        # Convert string dropdown back to Enum if needed, or pass string if model accepts it
-        # Assuming model_utils handles string conversion or we map it here:
+        # Convert string dropdown back to Enum if needed
         traj_enum = TrajectoryType[trajectory_type.upper()] if hasattr(TrajectoryType, trajectory_type.upper()) else trajectory_type
 
         progress(0.3, desc="Predicting Gaussians...")
@@ -147,10 +151,10 @@ def run_sharp(
 # -----------------------------------------------------------------------------
 
 def build_demo() -> gr.Blocks:
-    # Use the Default theme for a clean, modern look
+    # Use Default theme
     theme = gr.themes.Default()
 
-    with gr.Blocks(theme=theme, css=CSS, title="SHARP 3D") as demo:
+    with gr.Blocks(theme=theme, css=CSS, title="SHARP 3D", fill_width=True) as demo:
         
         # --- Header ---
         with gr.Row():
@@ -163,26 +167,25 @@ def build_demo() -> gr.Blocks:
                 )
         
         # --- Main Interface ---
-        with gr.Row():
+        with gr.Row(equal_height=False):
             
-            # --- Left Column: Input & Controls ---
-            with gr.Column(scale=1):
+            # --- Left Column: Input ---
+            with gr.Column(scale=1, min_width=500):
                 image_in = gr.Image(
                     label="Input Image",
                     type="filepath",
                     sources=["upload", "clipboard"],
                     elem_id="input-image",
-                    height=400
+                    height=None # Handled by CSS
                 )
 
-                # Collapsible Advanced Settings for a cleaner UI
+                # Collapsible Advanced Settings
                 with gr.Accordion("⚙️ Advanced Configuration", open=False):
                     with gr.Row():
                         trajectory = gr.Dropdown(
                             label="Camera Trajectory",
                             choices=["swipe", "shake", "rotate", "rotate_forward"],
                             value="rotate_forward",
-                            info="Camera movement for video preview"
                         )
                         output_res = gr.Dropdown(
                             label="Resolution (Long Side)",
@@ -203,12 +206,12 @@ def build_demo() -> gr.Blocks:
                 run_btn = gr.Button("✨ Generate 3D Scene", variant="primary", elem_id="run-btn")
 
             # --- Right Column: Output ---
-            with gr.Column(scale=1):
+            with gr.Column(scale=1, min_width=500):
                 video_out = gr.Video(
                     label="Preview Trajectory", 
                     elem_id="output-video",
                     autoplay=True,
-                    height=400
+                    height=None # Handled by CSS
                 )
                 with gr.Group():
                     ply_download = gr.DownloadButton(
@@ -218,7 +221,6 @@ def build_demo() -> gr.Blocks:
                     status_md = gr.Markdown("Ready to run.")
 
         # --- Footer: Examples ---
-        # Standard Gradio Examples component
         example_files = get_example_files()
         if example_files:
             gr.Examples(
@@ -243,21 +245,10 @@ def build_demo() -> gr.Blocks:
             concurrency_limit=1
         )
 
-        # --- Citation ---
         with gr.Accordion("About & Citation", open=False):
             gr.Markdown(
                 """
                 **SHARP: Sharp Monocular View Synthesis in Less Than a Second** (Apple, 2025).
-                
-                If you use this model, please cite:
-                ```bibtex
-                @inproceedings{Sharp2025:arxiv,
-                  title = {Sharp Monocular View Synthesis in Less Than a Second},
-                  author = {Mescheder, Dong, Li, Bai, et al.},
-                  year = {2025},
-                  journal = {arXiv preprint arXiv:2512.10685}
-                }
-                ```
                 """
             )
 
@@ -271,4 +262,5 @@ _ensure_dir(OUTPUTS_DIR)
 
 if __name__ == "__main__":
     demo = build_demo()
+    # allowed_paths needed so Gradio can serve files from the assets directory
     demo.queue().launch(allowed_paths=[str(ASSETS_DIR)])