Restore full MV3DR app with model loading and custom UI

app.py

@@ -1,9 +1,53 @@
-import gradio as gr
+import os
+import sys
+import torch
+import subprocess
+import logging
 
-def greet(name):
-    return f"Hello, {name}!"
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+logger.info("="*50)
+logger.info("Starting MV3DR Application")
+logger.info("="*50)
+
+if not os.path.exists('dust3r/.git'):
+    logger.info("Initializing dust3r submodule...")
+    subprocess.run(['git', 'submodule', 'update', '--init', '--recursive'], check=False)
+
+logger.info("Adding dust3r to Python path...")
+sys.path.append(os.path.join(os.path.dirname(__file__), 'dust3r'))
+
+logger.info("Importing model initialization module...")
+from model import initialize
+
+logger.info("Importing UI builder...")
+from gradio_ui import build_ui
+
+logger.info("Importing configuration...")
+from config import WEIGHTS_PATH, OUTPUT_DIR
+
+
+def main():
+    logger.info("Creating output directory...")
+    os.makedirs(OUTPUT_DIR, exist_ok=True)
+    
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    logger.info(f"Using device: {device}")
+    
+    logger.info("Loading DUSt3R model...")
+    logger.info(f"Model weights path: {WEIGHTS_PATH}")
+    model = initialize(WEIGHTS_PATH, device)
+    logger.info("Model loaded successfully!")
+    
+    logger.info("Building Gradio UI...")
+    app = build_ui(OUTPUT_DIR, model, device)
+    logger.info("UI built successfully!")
+    
+    logger.info("Launching application...")
+    app.queue().launch(show_api=False)
+    logger.info("Application started!")
 
-interface = gr.Interface(fn=greet, inputs="text", outputs="text")
 
 if __name__ == "__main__":
-    interface.launch(server_name="0.0.0.0", server_port=7860)
+    main()

requirements.txt

@@ -1,7 +1,7 @@
 torch
 torchvision
 roma
-gradio==4.44.0
+gradio==5.0.1
 huggingface_hub==0.26.2
 einops
 trimesh