Update app.py

app.py

@@ -44,9 +44,14 @@ os.chdir(repo_dir)
 # # Add repo directory to sys.path so Python can find modules inside it
 sys.path.append(repo_dir)
 
-# Ensure the correct path to app.py
-app_path = os.path.join(repo_dir, "app.py")
-if not os.path.exists(app_path):
+# Recursively search for app.py
+app_path = None
+for root, dirs, files in os.walk(repo_dir):
+    if "app.py" in files:
+        app_path = os.path.join(root, "app.py")
+        break
+
+if app_path is None:
     raise FileNotFoundError(f"app.py not found in {repo_dir}")
 
 # Load and execute `app.py`
@@ -57,6 +62,9 @@ if spec_app is None or spec_app.loader is None:
 app_module = importlib.util.module_from_spec(spec_app)
 spec_app.loader.exec_module(app_module)
 
+# Now you can use functions from app_module
+result = app_module.main()
+
 
 
 # Download specific files (if snapshot_download wasn't used)