Update app.py

app.py

@@ -4,6 +4,7 @@ import importlib.util
 import os
 import shutil
 from datetime import datetime
+import subprocess
 
 REMOTE_CODE_URL = os.environ.get("api_url")
 
@@ -40,10 +41,11 @@ def execute_py_file(py_file_path):
     spec = importlib.util.spec_from_file_location("dynamic_module", py_file_path)
     dynamic_module = importlib.util.module_from_spec(spec)
     spec.loader.exec_module(dynamic_module)
-    if not hasattr(dynamic_module, "build_app"):
-        raise AttributeError("No function named build_app() found in the downloaded script.")
-    app = dynamic_module.build_app()
-    app.launch()
+    if hasattr(dynamic_module, "build_app"):
+        app = dynamic_module.build_app()
+        app.launch()
+    else:
+        subprocess.run(["python", py_file_path], check=True)
 
 def main():
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")