added trame reqs

app.py

@@ -62,12 +62,23 @@ def download_private_repo(cache_dir, repo_id, hf_token):
 
 def load_private_app(repo_path):
     sys.path.insert(0, str(repo_path))
-    import app as private_app
-    if hasattr(private_app, "demo"):
-        return private_app.demo
-    if hasattr(private_app, "app"):
-        return private_app.app
-    raise RuntimeError("No demo/app found in private repo")
+    try:
+        import app as private_app
+        # Check for app or demo attribute
+        if hasattr(private_app, "app"):
+            return private_app.app
+        elif hasattr(private_app, "demo"):
+            return private_app.demo
+        # Check if PFMDemo class exists and instantiate it
+        elif hasattr(private_app, "PFMDemo"):
+            app_instance = private_app.PFMDemo()
+            app_instance.server.controller.add("decimate_again", app_instance.decimate_again)
+            app_instance.server.controller.add("reset_mesh", app_instance.reset_mesh)
+            return app_instance
+        raise RuntimeError("No demo/app found in private repo")
+    except Exception as e:
+        traceback.print_exc()
+        raise RuntimeError(f"Failed to load app from private repo: {e}")
 
 def main():
     hf_token = os.environ.get("HF_TOKEN")
@@ -83,14 +94,28 @@ def main():
     repo_path = download_private_repo(cache_dir, repo_id, hf_token)    
     demo = load_private_app(repo_path)
 
-    
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False,
-        show_error=True,
-        allowed_paths=[str(repo_path), DATA_DIR]
-    )
+    # Check if it's a Trame app (has server.start method) or Gradio app (has launch method)
+    if hasattr(demo, 'server') and hasattr(demo.server, 'start'):
+        # Trame app
+        port = int(os.environ.get("PORT", "7860"))
+        demo.server.start(
+            port=port,
+            open_browser=False,
+            show_connection_info=True,
+            backend="aiohttp",
+            exec_mode="main",
+        )
+    elif hasattr(demo, 'launch'):
+        # Gradio app
+        demo.launch(
+            server_name="0.0.0.0",
+            server_port=7860,
+            share=False,
+            show_error=True,
+            allowed_paths=[str(repo_path), DATA_DIR]
+        )
+    else:
+        raise RuntimeError("App does not have launch() or server.start() method")
 
 
 if __name__ == "__main__":